Update endpoints.py

endpoints.py

@@ -13,14 +13,14 @@ import io
 from datetime import datetime
 from bson import ObjectId
 import asyncio
-from config import notifications_collection, send_push_notification
+from config import notifications_collection, send_push_notification, SECRET_KEY
 
 def create_router(agent, logger, patients_collection, analysis_collection, users_collection, notifications_collection):
     router = APIRouter()
 
     @router.get("/status")
     async def status(current_user: dict = Depends(get_current_user)):
-        logger.info(f"Status endpoint accessed by {current_user['email']}")
+        logger.info(f"Status endpoint accessed by {current_user['email']} at {datetime.utcnow().isoformat()}")
         return {
             "status": "running",
             "timestamp": datetime.utcnow().isoformat(),
@@ -33,7 +33,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         name: Optional[str] = Query(None),
         current_user: dict = Depends(get_current_user)
     ):
-        logger.info(f"Fetching analysis results by {current_user['email']}")
+        logger.info(f"Fetching analysis results by {current_user['email']} at {datetime.utcnow().isoformat()}")
         try:
             query = {}
             if name:
@@ -57,12 +57,12 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
             return enriched_results
 
         except Exception as e:
-            logger.error(f"Error fetching analysis results: {e}")
+            logger.error(f"Error fetching analysis results: {str(e)} at {datetime.utcnow().isoformat()}")
             raise HTTPException(status_code=500, detail="Failed to retrieve analysis results")
 
     @router.get("/notifications")
     async def get_notifications(current_user: dict = Depends(get_current_user)):
-        logger.info(f"Fetching notifications for {current_user['email']}")
+        logger.info(f"Fetching notifications for {current_user['email']} at {datetime.utcnow().isoformat()}")
         try:
             notifications = await notifications_collection.find(
                 {"recipient_email": current_user["email"]}
@@ -73,7 +73,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
                     notification["timestamp"] = notification["timestamp"].isoformat()
             return notifications
         except Exception as e:
-            logger.error(f"Error fetching notifications: {str(e)}")
+            logger.error(f"Error fetching notifications: {str(e)} at {datetime.utcnow().isoformat()}")
             raise HTTPException(status_code=500, detail="Failed to retrieve notifications")
 
     @router.post("/chat-stream")
@@ -81,7 +81,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         request: ChatRequest,
         current_user: dict = Depends(get_current_user)
     ):
-        logger.info(f"Chat stream initiated by {current_user['email']}")
+        logger.info(f"Chat stream initiated by {current_user['email']} at {datetime.utcnow().isoformat()}")
         async def token_stream():
             try:
                 conversation = [{"role": "system", "content": agent.chat_prompt}]
@@ -107,7 +107,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
                     yield chunk + " "
                     await asyncio.sleep(0.05)
             except Exception as e:
-                logger.error(f"Streaming error: {e}")
+                logger.error(f"Streaming error: {str(e)} at {datetime.utcnow().isoformat()}")
                 yield f"⚠️ Error: {e}"
 
         return StreamingResponse(token_stream(), media_type="text/plain")
@@ -118,7 +118,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         language: str = Query("en-US", description="Language code for speech recognition"),
         current_user: dict = Depends(get_current_user)
     ):
-        logger.info(f"Voice transcription initiated by {current_user['email']}")
+        logger.info(f"Voice transcription initiated by {current_user['email']} at {datetime.utcnow().isoformat()}")
         try:
             audio_data = await audio.read()
             if not audio.filename.lower().endswith(('.wav', '.mp3', '.ogg', '.flac')):
@@ -130,7 +130,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         except HTTPException:
             raise
         except Exception as e:
-            logger.error(f"Error in voice transcription: {e}")
+            logger.error(f"Error in voice transcription: {str(e)} at {datetime.utcnow().isoformat()}")
             raise HTTPException(status_code=500, detail="Error processing voice input")
 
     @router.post("/voice/synthesize")
@@ -138,7 +138,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         request: VoiceOutputRequest,
         current_user: dict = Depends(get_current_user)
     ):
-        logger.info(f"Voice synthesis initiated by {current_user['email']}")
+        logger.info(f"Voice synthesis initiated by {current_user['email']} at {datetime.utcnow().isoformat()}")
         try:
             audio_data = text_to_speech(request.text, request.language, request.slow)
             
@@ -154,7 +154,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         except HTTPException:
             raise
         except Exception as e:
-            logger.error(f"Error in voice synthesis: {e}")
+            logger.error(f"Error in voice synthesis: {str(e)} at {datetime.utcnow().isoformat()}")
             raise HTTPException(status_code=500, detail="Error generating voice output")
 
     @router.post("/voice/chat")
@@ -165,7 +165,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         max_new_tokens: int = Query(512, ge=50, le=1024),
         current_user: dict = Depends(get_current_user)
     ):
-        logger.info(f"Voice chat initiated by {current_user['email']}")
+        logger.info(f"Voice chat initiated by {current_user['email']} at {datetime.utcnow().isoformat()}")
         try:
             audio_data = await audio.read()
             user_message = recognize_speech(audio_data, language)
@@ -188,7 +188,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         except HTTPException:
             raise
         except Exception as e:
-            logger.error(f"Error in voice chat: {e}")
+            logger.error(f"Error in voice chat: {str(e)} at {datetime.utcnow().isoformat()}")
             raise HTTPException(status_code=500, detail="Error processing voice chat")
 
     @router.post("/analyze-report")
@@ -199,7 +199,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         max_new_tokens: int = Form(1024),
         current_user: dict = Depends(get_current_user)
     ):
-        logger.info(f"Report analysis initiated by {current_user['email']}")
+        logger.info(f"Report analysis initiated by {current_user['email']} at {datetime.utcnow().isoformat()}")
         try:
             content_type = file.content_type
             allowed_types = [
@@ -260,7 +260,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
                     "read": False
                 }
                 await notifications_collection.insert_one(notification)
-                logger.info(f"Notification created for {current_user['email']}: {notification['message']}")
+                logger.info(f"Notification created for {current_user['email']} at {datetime.utcnow().isoformat()}: {notification['message']}")
                 # Send push notification
                 await send_push_notification(current_user["email"], notification["message"])
 
@@ -275,7 +275,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         except HTTPException:
             raise
         except Exception as e:
-            logger.error(f"Error in report analysis: {str(e)}")
+            logger.error(f"Error in report analysis: {str(e)} at {datetime.utcnow().isoformat()}")
             raise HTTPException(
                 status_code=500,
                 detail=f"Failed to analyze report: {str(e)}"
@@ -286,7 +286,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         patient_id: str,
         current_user: dict = Depends(get_current_user)
     ):
-        logger.info(f"Patient deletion initiated by {current_user['email']} for patient {patient_id}")
+        logger.info(f"Patient deletion initiated by {current_user['email']} at {datetime.utcnow().isoformat()} for patient {patient_id}")
         try:
             patient = await patients_collection.find_one({"fhir_id": patient_id})
             if not patient:
@@ -296,20 +296,20 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
                 raise HTTPException(status_code=403, detail="Not authorized to delete this patient")
 
             await analysis_collection.delete_many({"patient_id": patient_id})
-            logger.info(f"Deleted analyses for patient {patient_id}")
+            logger.info(f"Deleted analyses for patient {patient_id} at {datetime.utcnow().isoformat()}")
 
             await notifications_collection.delete_many({"patient_id": patient_id})
-            logger.info(f"Deleted notifications for patient {patient_id}")
+            logger.info(f"Deleted notifications for patient {patient_id} at {datetime.utcnow().isoformat()}")
 
             await patients_collection.delete_one({"fhir_id": patient_id})
-            logger.info(f"Patient {patient_id} deleted successfully")
+            logger.info(f"Patient {patient_id} deleted successfully at {datetime.utcnow().isoformat()}")
 
             return {"status": "success", "message": f"Patient {patient_id} and associated data deleted"}
 
         except HTTPException:
             raise
         except Exception as e:
-            logger.error(f"Error deleting patient {patient_id}: {str(e)}")
+            logger.error(f"Error deleting patient {patient_id}: {str(e)} at {datetime.utcnow().isoformat()}")
             raise HTTPException(status_code=500, detail=f"Failed to delete patient: {str(e)}")
 
     return router