111

app/main.py

@@ -323,6 +323,36 @@ async def global_exception_handler(request: Request, exc: Exception):
     return JSONResponse(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, content=ErrorResponse(message=str(exc), type="internal_error").dict())
 
 
+@app.post("/v1beta/models/gemini-3-pro-image-preview:generateContent")
+async def proxy_gemini_vision(request: Request, key: str):
+    """
+    Direct proxy for gemini-3-pro-image-preview:generateContent
+    """
+    try:
+        body = await request.json()
+        target_url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-3-pro-image-preview:generateContent?key={key}"
+        
+        headers = {
+            "Content-Type": "application/json"
+        }
+        
+        # Use httpx for async request
+        import httpx
+        async with httpx.AsyncClient() as client:
+            response = await client.post(target_url, json=body, headers=headers, timeout=60)
+            
+            # Forward the status code and content
+            if response.status_code != 200:
+                return JSONResponse(status_code=response.status_code, content=response.json())
+            
+            return response.json()
+            
+    except Exception as e:
+        logger.error(f"Proxy error: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+
 @app.get("/", response_class=HTMLResponse)
 async def root():
     html_content = f"""