Update main.py

main.py

@@ -497,7 +497,10 @@ def bing_dalle(item: BingDalle):
 
 
 @app.post("/UFoP/dalle3xl")
-def dalle_3xl(item: Dalle3XL, api_key: None = Depends(validate_api_key)):
+def dalle_3xl(
+    item: Dalle3XL,
+    api_key: None = Depends(validate_api_key)
+):
     API_URL = SOURCE_DALLE3XL_URL
     try:
         payload = {"inputs": item.query}
@@ -505,17 +508,21 @@ def dalle_3xl(item: Dalle3XL, 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
-        )
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
         response.raise_for_status()
     except requests.exceptions.RequestException:
-        raise HTTPException(status_code=500, detail=internal_error)
+        raise HTTPException(
+            status_code=500,
+            detail=internal_error
+        )
 
     try:
         encoded_string = base64.b64encode(response.content).decode("utf-8")
     except Exception:
-        raise HTTPException(status_code=500, detail=contact_support)
+        raise HTTPException(
+            status_code=500,
+            detail=contact_support
+        )
 
     if encoded_string:
         return SuccessResponse(status="True", randydev={"data": encoded_string})
@@ -524,7 +531,10 @@ def dalle_3xl(item: Dalle3XL, api_key: None = Depends(validate_api_key)):
 
 
 @app.post("/UFoP/proteusv02")
-async 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}
@@ -538,12 +548,18 @@ async def proteusv0_2(item: ProteusV02, api_key: None = Depends(validate_api_key
             )
         response.raise_for_status()
     except httpx.HTTPStatusError:
-        raise HTTPException(status_code=500, detail=internal_error)
+        raise HTTPException(
+            status_code=500, 
+            detail=internal_error
+        )
 
     try:
         encoded_string = base64.b64encode(response.content).decode("utf-8")
     except Exception:
-        raise HTTPException(status_code=500, detail=contact_support)
+        raise HTTPException(
+            status_code=500, 
+            detail=contact_support
+        )
 
     if encoded_string:
         return SuccessResponse(status="True", randydev={"data": encoded_string})