Update main.py (#4)

- Update main.py (777d10e4b107e8bb607666c5f290ae0966a377f5)


Co-authored-by: True Saiyan <[email protected]>

main.py

@@ -89,6 +89,7 @@ SOURCE_UNSPLASH_URL = os.environ["SOURCE_UNSPLASH_URL"]
 SOURCE_OCR_URL = os.environ["SOURCE_OCR_URL"]
 SOURCE_ALPHA_URL = os.environ["SOURCE_ALPHA_URL"]
 SOURCE_OPENDALLE_URL = os.environ["SOURCE_OPENDALLE_URL"]
+SOURCE_DALLE3XL_URL = os.environ["SOURCE_DALLE3XL_URL"]
 SOURCE_ANIME_STYLED_URL = os.environ["SOURCE_ANIME_STYLED_URL"]
 SOURCE_WAIFU_URL = os.environ["SOURCE_WAIFU_URL"]
 SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
@@ -117,6 +118,8 @@ trans = SyncTranslator()
 
 app = FastAPI(docs_url=None, redoc_url="/")
 
+timeout = 100
+
 contact_support = """
 We are aware that AI is currently offline. This seems to be caused by the API
 We are investigating and doing our best to get things back online as soon as possible. Thank you for your patience
@@ -528,6 +531,39 @@ def ryuzaki_ai(
     except Exception:
         return SuccessResponse(status="False", randydev={"message": contact_support})
 
+@app.post("/ryuzaki/dalle3xl")
+def dalle_3xl(
+    item: Dalle3XL,
+    api_key: None = Depends(validate_api_key)
+):
+    API_URL = SOURCE_DALLE3XL_URL
+    try:
+        payload = {"inputs": item.query}
+        headers = {
+            "Authorization": f"Bearer {HUGGING_TOKEN}",
+            "Content-Type": "application/json",
+        }
+        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=CustomErrorResponseModel(detail=[{}])
+        )
+
+    try:
+        encoded_string = base64.b64encode(response.content).decode("utf-8")
+    except Exception:
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+
+    if encoded_string:
+        return SuccessResponse(status="True", randydev={"data": encoded_string})
+    else:
+        return SuccessResponse(status="False", randydev={"data": contact_support})
+
 @app.post("/ryuzaki/opendalle")
 def open_dalle(
     item: OpenDalle,