Upload 2 files

main.py

@@ -85,6 +85,7 @@ MONGO_URL = os.environ["MONGO_URL"]
 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_WAIFU_URL = os.environ["SOURCE_WAIFU_URL"]
 SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
 SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
@@ -387,6 +388,36 @@ def ryuzaki_ai(
     except Exception:
         return {"status": "false", "message": "Internal server error"}
 
+@app.post("/ryuzaki/opendalle")
+async def open_dalle(
+    item: OpenDalle,
+    api_key: None = Depends(validate_api_key)
+):
+    API_URL = SOURCE_OPENDALLE_URL
+    try:
+        payload = {"inputs": item.query}
+        headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
+        response = requests.post(API_URL, headers=headers, json=payload)
+        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=[{}])
+        )
+    headers = {"Content-Type": "image/jpeg"}
+    return Response(
+        content=encoded_string,
+        media_type="image/jpeg",
+        headers=headers
+    )
+
 @app.get("/ryuzaki/unsplash")
 async def get_image_unsplash(item: GetImageUnsplash):
     url = SOURCE_UNSPLASH_URL

models.py

@@ -121,3 +121,6 @@ class GetMonitorLogs(BaseModel):
 
 class DownloadLink(BaseModel):
     link: str
+
+class OpenDalle(BaseModel):
+    query: str