1

app.py

@@ -8,6 +8,7 @@ import os
 import warnings
 from pydub import AudioSegment
 from datetime import datetime
+import time
 
 warnings.filterwarnings("ignore")
 
@@ -72,8 +73,8 @@ class StressResponse(BaseModel):
     stress_level: float
     category: str
     gender: str = None  # Optional, only for audio analysis
-    status: str
-    time: str
+    status_code: int
+    processing_time_ms: int
     size: str
 
 @app.post("/analyze-stress/", response_model=StressResponse)
@@ -85,6 +86,8 @@ async def analyze_stress(
     if file is None and file_path is None and text is None:
         raise HTTPException(status_code=400, detail="Either a file, file path, or text input is required.")
     
+    start_time = time.time()
+
     # Handle audio file analysis
     if file or file_path:
         if file:
@@ -108,12 +111,13 @@ async def analyze_stress(
 
         try:
             result = analyze_voice_stress(temp_audio_path)
+            processing_time_ms = int((time.time() - start_time) * 1000)  # Calculate time in ms
             result.update({
-                "status": "Success",
-                "time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+                "status_code": 200,
+                "processing_time_ms": processing_time_ms,
                 "size": f"{round(file_size / 1024, 2)} KB"  # Convert size to KB and round to 2 decimal places
             })
-            return JSONResponse(content=result)
+            return JSONResponse(content=result, status_code=200)
         except Exception as e:
             raise HTTPException(status_code=500, detail=str(e))
         finally:
@@ -124,12 +128,13 @@ async def analyze_stress(
     # Handle text analysis
     elif text:
         result = analyze_text_stress(text)
+        processing_time_ms = int((time.time() - start_time) * 1000)  # Calculate time in ms
         result.update({
-            "status": "Success",
-            "time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "status_code": 200,
+            "processing_time_ms": processing_time_ms,
             "size": "N/A"  # Size is not applicable for text input
         })
-        return JSONResponse(content=result)
+        return JSONResponse(content=result, status_code=200)
 
 if __name__ == "__main__":
     import uvicorn