[Feature Updated] > Updated response for files such that, the video files only get the confidence and ai_explanation added, while image files do not.

backend/app/api/file_routes.py

@@ -12,6 +12,22 @@ from app.core.limiter import limiter
 
 router = APIRouter(prefix="/files", tags=["files"])
 
+def format_file_response(f):
+    base_dict = {
+        "id": f.id,
+        "filename": f.filename,
+        "type": f.filetype,
+        "size": f.filesize,
+        "status": f.status,
+        "result": f.result,
+        "uploaded_at": f.created_at
+    }
+    # Only include confidence and explanation for videos
+    if f.filetype and f.filetype.startswith("video/"):
+        base_dict["confidence"] = f.confidence
+        base_dict["ai_explanation"] = f.ai_explanation
+    return base_dict
+
 @router.get("/my-files")
 def list_user_files(
     db: Session = Depends(get_db),
@@ -19,24 +35,10 @@ def list_user_files(
     ):
     files = get_user_files(db, current_user.id)
     
-    
     if len(files) != 0:
         return {
             "total": len(files),
-            "files": [
-                {
-                    "id": f.id,
-                    "filename": f.filename,
-                    "type": f.filetype,
-                    "size": f.filesize,
-                    "status": f.status,
-                    "result": f.result,
-                    "confidence": f.confidence,
-                    "ai_explanation": f.ai_explanation,
-                    "uploaded_at": f.created_at
-                }
-                for f in files
-            ]
+            "files": [format_file_response(f) for f in files]
         }
     
     else:
@@ -52,17 +54,7 @@ def get_file(
 ):
     file = get_file_detail(db, file_id, current_user.id)
     
-    return {
-        "id": file.id,
-        "filename": file.filename,
-        "type": file.filetype,
-        "size": file.filesize,
-        "status": file.status,
-        "result": file.result,
-        "confidence": file.confidence,
-        "ai_explanation": file.ai_explanation,
-        "uploaded_at": file.created_at
-    }
+    return format_file_response(file)
 
 @router.post("/upload")
 @limiter.limit("5/minute")