fix

src/server/main.py

@@ -5,7 +5,7 @@ from typing import List, Optional
 from abc import ABC, abstractmethod
 
 import uvicorn
-from fastapi import Depends, FastAPI, File, HTTPException, Query, Request, UploadFile, Form, Header
+from fastapi import Depends, FastAPI, File, HTTPException, Query, Request, UploadFile, Header
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse, RedirectResponse, StreamingResponse
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
@@ -119,7 +119,7 @@ class AudioProcessingResponse(BaseModel):
 
 class ChatRequest(BaseModel):
     prompt: str = Field(..., description="Base64-encoded encrypted prompt (max 1000 characters after decryption)")
-    src_lang: str = Field(..., description="Base64-encoded encrypted source language code (default: kan_Knda after decryption)")
+    src_lang: str = Field(..., description="Base64-encoded encrypted source language code")
 
     @field_validator("prompt", "src_lang")
     def must_be_valid_base64(cls, v):
@@ -191,6 +191,15 @@ class VisualQueryRequest(BaseModel):
             raise ValueError("Field must be valid base64-encoded data")
         return v
 
+    class Config:
+        schema_extra = {
+            "example": {
+                "query": "base64_encoded_encrypted_describe_image",
+                "src_lang": "base64_encoded_encrypted_kan_Knda",
+                "tgt_lang": "base64_encoded_encrypted_kan_Knda"
+            }
+        }
+
 class VisualQueryResponse(BaseModel):
     answer: str
 
@@ -282,7 +291,7 @@ async def register_user(
 
 @app.post("/v1/app/register",
           response_model=TokenResponse,
-          summary="Register New App User ",
+          summary="Register New App User",
           description="Create a new user account for the mobile app in the `app_users` table using an encrypted email and device token. Returns an access token and refresh token. Rate limited to 5 requests per minute per IP. Requires X-Session-Key header.",
           tags=["Authentication"],
           responses={
@@ -688,7 +697,7 @@ async def translate(
 @app.post("/v1/visual_query", 
           response_model=VisualQueryResponse,
           summary="Visual Query with Image",
-          description="Process a visual query with an encrypted text query, encrypted image, and encrypted language codes. Rate limited to 100 requests per minute per user. Requires authentication and X-Session-Key header.",
+          description="Process a visual query with an encrypted text query, encrypted image, and encrypted language codes provided in a JSON body. Rate limited to 100 requests per minute per user. Requires authentication and X-Session-Key header.",
           tags=["Chat"],
           responses={
               200: {"description": "Query response", "model": VisualQueryResponse},
@@ -700,10 +709,8 @@ async def translate(
 @limiter.limit(settings.chat_rate_limit)
 async def visual_query(
     request: Request,
-    query: str = Form(..., description="Base64-encoded encrypted text query"),
+    visual_query_request: VisualQueryRequest,
     file: UploadFile = File(..., description="Encrypted image file to analyze"),
-    src_lang: str = Form(..., description="Base64-encoded encrypted source language code"),
-    tgt_lang: str = Form(..., description="Base64-encoded encrypted target language code"),
     credentials: HTTPAuthorizationCredentials = Depends(bearer_scheme),
     x_session_key: str = Header(..., alias="X-Session-Key")
 ):
@@ -712,7 +719,7 @@ async def visual_query(
     
     # Decrypt query
     try:
-        encrypted_query = base64.b64decode(query)
+        encrypted_query = base64.b64decode(visual_query_request.query)
         decrypted_query = decrypt_data(encrypted_query, session_key).decode("utf-8")
     except Exception as e:
         logger.error(f"Query decryption failed: {str(e)}")
@@ -720,7 +727,7 @@ async def visual_query(
     
     # Decrypt source language
     try:
-        encrypted_src_lang = base64.b64decode(src_lang)
+        encrypted_src_lang = base64.b64decode(visual_query_request.src_lang)
         decrypted_src_lang = decrypt_data(encrypted_src_lang, session_key).decode("utf-8")
     except Exception as e:
         logger.error(f"Source language decryption failed: {str(e)}")
@@ -728,7 +735,7 @@ async def visual_query(
     
     # Decrypt target language
     try:
-        encrypted_tgt_lang = base64.b64decode(tgt_lang)
+        encrypted_tgt_lang = base64.b64decode(visual_query_request.tgt_lang)
         decrypted_tgt_lang = decrypt_data(encrypted_tgt_lang, session_key).decode("utf-8")
     except Exception as e:
         logger.error(f"Target language decryption failed: {str(e)}")