docuyment-query

src/server/main.py

@@ -560,7 +560,94 @@ async def visual_query(
     except ValueError as e:
         logger.error(f"Invalid JSON response: {str(e)}")
         raise HTTPException(status_code=500, detail="Invalid response format from visual query service")
+
+
+from fastapi import FastAPI, File, HTTPException, Request, UploadFile, Form, Query
+from pydantic import BaseModel, Field
+
+class DocumentQueryResponse(BaseModel):
+    answer: str
+
+    class Config:
+        schema_extra = {"example": {"answer": "The image shows a screenshot of a webpage."}}
+
+@app.post("/v1/document_query", 
+          response_model=DocumentQueryResponse,
+          summary="Docuemnt Query with Image",
+          description="Process a Document query with a text query, image, and language codes. Provide the query and image as form data, and source/target languages as query parameters.",
+          tags=["Chat"],
+          responses={
+              200: {"description": "Query response", "model": DocumentQueryResponse},
+              400: {"description": "Invalid query or language codes"},
+              422: {"description": "Validation error in request body"},
+              504: {"description": "Visual query service timeout"}
+          })
+async def document_query(
+    request: Request,
+    query: str = Form(..., description="Text query to describe or analyze the image (e.g., 'describe the image')"),
+    file: UploadFile = File(..., description="Image file to analyze (e.g., PNG, JPEG)"),
+    src_lang: str = Query(..., description="Source language code (e.g., kan_Knda, en)"),
+    tgt_lang: str = Query(..., description="Target language code (e.g., kan_Knda, en)")
+):
+    # Validate query
+    if not query.strip():
+        raise HTTPException(status_code=400, detail="Query cannot be empty")
+    if len(query) > 1000:
+        raise HTTPException(status_code=400, detail="Query cannot exceed 1000 characters")
+    
+    # Validate language codes
+    supported_languages = ["kan_Knda", "hin_Deva", "tam_Taml", "eng_Latn"]  # Add more as needed
+    if src_lang not in supported_languages:
+        raise HTTPException(status_code=400, detail=f"Unsupported source language: {src_lang}. Must be one of {supported_languages}")
+    if tgt_lang not in supported_languages:
+        raise HTTPException(status_code=400, detail=f"Unsupported target language: {tgt_lang}. Must be one of {supported_languages}")
+    
+    logger.info("Processing document query request", extra={
+        "endpoint": "/v1/document_query",
+        "query_length": len(query),
+        "file_name": file.filename,
+        "client_ip": request.client.host,
+        "src_lang": src_lang,
+        "tgt_lang": tgt_lang
+    })
+    
+    external_url = f"{os.getenv('EXTERNAL_API_BASE_URL')}/v1/document_query/?src_lang={src_lang}&tgt_lang={tgt_lang}"
+    
+    try:
+        file_content = await file.read()
+        files = {"file": (file.filename, file_content, file.content_type)}
+        data = {"query": query}
         
+        response = requests.post(
+            external_url,
+            files=files,
+            data=data,
+            headers={"accept": "application/json"},
+            timeout=60
+        )
+        response.raise_for_status()
+        
+        response_data = response.json()
+        answer = response_data.get("answer", "")
+        
+        if not answer:
+            logger.warning(f"Empty answer received from external API: {response_data}")
+            raise HTTPException(status_code=500, detail="No answer provided by visual query service")
+        
+        logger.info(f"document_query query successful: {answer}")
+        return VisualQueryResponse(answer=answer)
+    
+    except requests.Timeout:
+        logger.error("document_query query request timed out")
+        raise HTTPException(status_code=504, detail="document_query query service timeout")
+    except requests.RequestException as e:
+        logger.error(f"Error during document_query query: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"document_query query failed: {str(e)}")
+    except ValueError as e:
+        logger.error(f"Invalid JSON response: {str(e)}")
+        raise HTTPException(status_code=500, detail="Invalid response format from document_query query service")
+
+
 from enum import Enum
 
 class SupportedLanguage(str, Enum):