deploy: auto-deploy 17:41:13

app/api/v1/csv_analysis.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import json
 from typing import Annotated, Any, Dict, List, Optional
 
-from fastapi import APIRouter, Depends, File, Form, HTTPException, Request, UploadFile
+from fastapi import APIRouter, Depends, File, HTTPException, Request, UploadFile
 from pydantic import BaseModel, Field, ValidationError
 
 from app.api.deps import require_auth
@@ -34,7 +34,7 @@ class _AIResponse(BaseModel):
     visualization: List[_VisualizationBlock] = []
     message: str = ""
 
-router = APIRouter()
+router = APIRouter(dependencies=[Depends(require_auth)])
 _settings = get_settings()
 _MAX_UPLOAD_BYTES = _settings.max_upload_bytes
 
@@ -46,7 +46,6 @@ _MAX_UPLOAD_BYTES = _settings.max_upload_bytes
 async def get_csv_info(
     files: Annotated[Optional[List[UploadFile]], File(description="CSV files to inspect (max 10 total with URLs)")] = None,
     urls: Annotated[Optional[str], Form(description="JSON array of file URLs (max 10 total with files)")] = None,
-    token: str = Depends(require_auth),
 ):
     parsed_urls: List[str] = []
     if urls:
@@ -177,16 +176,33 @@ async def get_csv_info(
 async def csv_chat(
     request: Request,
     file: Annotated[Optional[UploadFile], File(description="CSV file to analyze")] = None,
-    url: Annotated[Optional[str], Form(description="URL to a CSV file")] = None,
-    query: str = Form(..., description="Natural language query about the CSV data"),
-    token: str = Depends(require_auth),
 ):
-    if not file and not url:
+    content_type = request.headers.get("content-type", "")
+
+    url: Optional[str] = None
+    query: Optional[str] = None
+
+    if "application/json" in content_type:
+        try:
+            body = await request.json()
+            url = body.get("url")
+            query = body.get("query")
+        except Exception as exc:
+            raise HTTPException(status_code=400, detail=f"Invalid JSON body: {exc}")
+    else:
+        form = await request.form()
+        url = form.get("url")
+        query = form.get("query")
+
+    has_file = file is not None
+    has_url = bool(url)
+
+    if not has_file and not has_url:
         raise HTTPException(status_code=400, detail="Provide either a file or a URL")
-    if file and url:
+    if has_file and has_url:
         raise HTTPException(status_code=400, detail="Provide either a file or a URL, not both")
 
-    if file:
+    if has_file:
         data = await file.read()
         if len(data) > _MAX_UPLOAD_BYTES:
             raise HTTPException(status_code=413, detail=f"File exceeds {_settings.max_upload_mb} MB limit")
@@ -196,6 +212,9 @@ async def csv_chat(
     else:
         source = url
 
+    if not query:
+        raise HTTPException(status_code=400, detail="Query is required")
+
     metadata = await get_dataset_info(source)
     system_prompt = get_csv_system_prompt(metadata)