Update app.py

app.py

@@ -8,6 +8,7 @@ from pydantic import BaseModel
 from typing import List, Dict
 import uvicorn
 import json
+import time
 
 # Initialiser l'application FastAPI
 app = FastAPI()
@@ -89,6 +90,9 @@ async def mock_endpoint(payload: MockRequest):
     Endpoint API qui prend un string et un secret, vérifie le secret,
     puis transmet via WebSocket, attend une réponse et la retourne.
     """
+
+    start_time = time.monotonic()
+    
     try:
         input_string = payload.parameter
         selected_model = payload.model
@@ -132,6 +136,11 @@ async def mock_endpoint(payload: MockRequest):
             print(f"Réponse reçue du WebSocket: '{websocket_response}'")
             return {"response_from_client": websocket_response}
 
+
+            end_time = time.monotonic()
+            duration = end_time - start_time
+            print(f"Requête complétée en {duration:.2f} secondes.")
+
         except asyncio.TimeoutError:
             print("Timeout: Aucune réponse du client WebSocket.")
             raise HTTPException(status_code=408, detail="Timeout: Le client n'a pas répondu à temps.")