Update app.py

app.py

@@ -2,41 +2,37 @@ from fastapi import FastAPI, Request, Response
 import httpx
 
 app = FastAPI()
-
-# Базовый URL Google AI Studio
 GOOGLE_API_URL = "https://generativelanguage.googleapis.com"
 
 @app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
 async def proxy(path: str, request: Request):
     async with httpx.AsyncClient() as client:
-        # Собираем URL
         url = f"{GOOGLE_API_URL}/{path}"
         if request.query_params:
             url += f"?{request.query_params}"
             
-        # Читаем тело запроса
         body = await request.body()
         
-        # Копируем заголовки (особенно важен Authorization или x-goog-api-key)
-        headers = dict(request.headers)
-        # Удаляем хост, чтобы Google не ругался
-        headers.pop("host", None)
+        # Фильтруем заголовки, оставляем только нужные для API
+        allowed_headers = ['content-type', 'x-goog-api-key', 'authorization', 'x-goog-api-client']
+        headers = {k: v for k, v in request.headers.items() if k.lower() in allowed_headers}
 
-        # Делаем запрос к Google
-        resp = await client.request(
-            method=request.method,
-            url=url,
-            content=body,
-            headers=headers,
-            timeout=60.0
-        )
-        
-        return Response(
-            content=resp.content,
-            status_code=resp.status_code,
-            headers=dict(resp.headers)
-        )
+        try:
+            resp = await client.request(
+                method=request.method,
+                url=url,
+                content=body,
+                headers=headers,
+                timeout=60.0
+            )
+            return Response(
+                content=resp.content,
+                status_code=resp.status_code,
+                headers={"Content-Type": resp.headers.get("Content-Type", "application/json")}
+            )
+        except Exception as e:
+            return Response(content=str(e), status_code=500)
 
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+@app.get("/")
+async def health():
+    return {"status": "proxy is running"}