Update main.py

main.py

@@ -1,11 +1,10 @@
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException, Query
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import Response
-import requests
+import httpx
 
 app = FastAPI()
 
-# Add CORS middleware to allow all origins. Adjust this in production!
+# Add CORS middleware
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -15,17 +14,17 @@ app.add_middleware(
 )
 
 @app.get("/")
-@app.post("/")
-async def proxy(request: Request, url: str):
+async def proxy(url: str = Query(..., description="URL to proxy")):
     headers = {
-        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
     }
-    if request.method == "GET":
-        response = requests.get(url, headers=headers)
-    elif request.method == "POST":
-        data = await request.json() if request.headers.get("Content-Type") == "application/json" else None
-        response = requests.post(url, json=data, headers=headers)
-    return Response(content=response.content, status_code=response.status_code, headers=dict(response.headers))
+
+    async with httpx.AsyncClient() as client:
+        try:
+            response = await client.get(url, headers=headers)
+            return response.content
+        except httpx.RequestError as e:
+            raise HTTPException(status_code=500, detail="Unexpected error")
 
 if __name__ == "__main__":
     import uvicorn