add- fix

src/server/main.py

@@ -616,18 +616,17 @@ async def visual_query(
         logger.error(f"Invalid JSON response: {str(e)}")
         raise HTTPException(status_code=500, detail="Invalid response format from visual query service")
     
-
-# Add this import at the top with other imports
+# Ensure these imports are at the top with other imports
 from fastapi.responses import StreamingResponse
+from enum import Enum
 
 # Define supported languages for validation
-from enum import Enum
 class SupportedLanguage(str, Enum):
     kannada = "kannada"
     hindi = "hindi"
     tamil = "tamil"
 
-# Add the new endpoint
+# Add the speech-to-speech endpoint
 @app.post("/v1/speech_to_speech",
           summary="Speech-to-Speech Conversion",
           description="Convert input speech to processed speech by calling an external speech-to-speech API. Rate limited to 5 requests per minute per user. Requires authentication.",
@@ -650,7 +649,7 @@ async def speech_to_speech(
     user_id = await get_current_user(credentials)
     logger.info("Processing speech-to-speech request", extra={
         "endpoint": "/v1/speech_to_speech",
-        "filename": file.filename,
+        "audio_filename": file.filename,  # Changed from 'filename' to avoid KeyError
         "language": language,
         "client_ip": get_remote_address(request),
         "user_id": user_id
@@ -683,12 +682,12 @@ async def speech_to_speech(
         )
 
     except requests.Timeout:
-        logger.error("External speech-to-speech API timed out")
+        logger.error("External speech-to-speech API timed out", extra={"user_id": user_id})
         raise HTTPException(status_code=504, detail="External API timeout")
     except requests.RequestException as e:
-        logger.error(f"External speech-to-speech API error: {str(e)}")
+        logger.error(f"External speech-to-speech API error: {str(e)}", extra={"user_id": user_id})
         raise HTTPException(status_code=500, detail=f"External API error: {str(e)}")
-
+    
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Run the FastAPI server.")
     parser.add_argument("--port", type=int, default=settings.port, help="Port to run the server on.")