Update main.py

main.py

@@ -43,6 +43,7 @@ from dotenv import load_dotenv
 from fastapi import Depends, FastAPI, Header, HTTPException, Query, Request
 from fastapi.responses import JSONResponse, StreamingResponse
 from gpytranslate import SyncTranslator
+import httpx
 from httpx import AsyncClient
 from pymongo import MongoClient
 from RyuzakiLib.hackertools.chatgpt import RendyDevChat
@@ -523,7 +524,7 @@ def dalle_3xl(item: Dalle3XL, api_key: None = Depends(validate_api_key)):
 
 
 @app.post("/UFoP/proteusv02")
-def proteusv0_2(item: ProteusV02, api_key: None = Depends(validate_api_key)):
+async def proteusv0_2(item: ProteusV02, api_key: None = Depends(validate_api_key)):
     API_URL = SOURCE_PROTEUSV0_2_URL
     try:
         payload = {"inputs": item.query}
@@ -531,11 +532,12 @@ def proteusv0_2(item: ProteusV02, api_key: None = Depends(validate_api_key)):
             "Authorization": f"Bearer {HUGGING_TOKEN}",
             "Content-Type": "application/json",
         }
-        response = requests.post(
-            API_URL, headers=headers, json=payload, timeout=timeout
-        )
+        async with httpx.AsyncClient() as client:
+            response = await client.post(
+                API_URL, headers=headers, json=payload, timeout=timeout
+            )
         response.raise_for_status()
-    except requests.exceptions.RequestException:
+    except httpx.HTTPStatusError:
         raise HTTPException(status_code=500, detail=internal_error)
 
     try: