Update app.py

app.py

@@ -371,7 +371,6 @@ async def upload_document(
 ):
     """Upload and process a legal document"""
     try:
-        print("Working upload")
         # Validate file
         if not file.filename:
             raise HTTPException(status_code=400, detail="No file provided")
@@ -455,8 +454,7 @@ async def get_session_status(session_id: str):
         if not session:
             raise HTTPException(status_code=404, detail="Session not found")
         
-        # --- START FIX ---
-        # Convert all datetime objects to ISO 8601 strings
+        # --- FIX: Convert all datetime objects to ISO 8601 strings ---
         session["_id"] = str(session["_id"])
         if session.get("created_at"):
             session["created_at"] = session["created_at"].isoformat()
@@ -464,7 +462,6 @@ async def get_session_status(session_id: str):
             session["updated_at"] = session["updated_at"].isoformat()
         if session.get("processing_completed_at"):
             session["processing_completed_at"] = session["processing_completed_at"].isoformat()
-        # --- END FIX ---
         
         # Add processing progress info
         if session["status"] == "completed":
@@ -526,12 +523,10 @@ async def get_processing_results(session_id: str):
         for chunk in chunks_metadata:
             chunk["_id"] = str(chunk["_id"])
             
-        # --- START FIX ---
-        # Convert potential datetime object to an ISO 8601 string to make it JSON serializable
+        # --- FIX: Convert potential datetime object to an ISO 8601 string ---
         processing_completed_at = session.get("processing_completed_at")
         if processing_completed_at:
             processing_completed_at = processing_completed_at.isoformat()
-        # --- END FIX ---
             
         return JSONResponse(
             status_code=200,