intrgrating suggestions after betta 1

transportmanager/server/routes/report.py

@@ -2,6 +2,7 @@ from flask import Blueprint, request, jsonify
 from flask_jwt_extended import jwt_required, get_jwt_identity
 from models.order_out import OrdersOut  # Your plain SQL model
 from datetime import datetime
+from models.user import Users
 
 report_bp = Blueprint("report", __name__, url_prefix="/report")
 
@@ -10,6 +11,10 @@ report_bp = Blueprint("report", __name__, url_prefix="/report")
 def get_profit_report():
     try:
         user_id = get_jwt_identity()
+        users = Users()
+        user = users.get_user_by_id(user_id)
+        if user['user_role'] == 'company_user':
+            user_id = user['company_id']
         # Get filters from query params
         date_from = request.args.get("date_from")
         date_to = request.args.get("date_to")