Update backend/src/routes/case_routes.py

backend/src/routes/case_routes.py

@@ -627,10 +627,15 @@ def upload_image_route(caseId):
 def get_case_images(caseId):
 
     try:
+        
+        case_data = db.db.cases.find_one({"_id":ObjectId(caseId)})
+        print(case_data)
         # Fetch images with only file_path and case_id
         images = list(db.db.images.find(
-            {"case_id": caseId},
-            {"file_path": 1, "case_id": 1, "_id": 0}  # exclude _id
+            {"case_id": caseId,"user_id":case_data['user_id']},
+            
+            {"file_path": 1, "case_id": 1, "_id": 0,
+             }  # exclude _id
         ))
 
         if not images:
@@ -651,9 +656,6 @@ def get_case_images(caseId):
         print(f"❌ Error fetching images for case {caseId}: {e}")
         return jsonify({"error": str(e)}), 500
 
-
-
-
 @case_bp.route('/getfilee/<string:caseId>',methods=['GET'])
 def filee_route(caseId):
 
@@ -716,7 +718,10 @@ def get_status(caseId):
 @case_bp.route('/fetch_img_analysis/<string:caseId>',methods=['GET'])
 def fetch_img_analysis(caseId):
     try:
-        img=db.db.images.find({'case_id':caseId})
+        case_data = db.db.cases.find_one({"_id":ObjectId(caseId)})
+        if isinstance(case_data['officer'], ObjectId):
+            officer_id = str(case_data['officer'])
+        img=db.db.images.find({'case_id':caseId,"user_id":officer_id})
         analysis_results = []
         for image in img:
             analysis_results.append({
@@ -732,4 +737,3 @@ def fetch_img_analysis(caseId):
         print(f"❌ Error getting image analysis: {e}")
         return jsonify({"images":img}), 404
 
-