Update main.py

dalle3xl trial add

main.py

@@ -71,11 +71,14 @@ from models import *
 
 logging.basicConfig(level=logging.ERROR)
 
+#MONGO_URL = os.environ["MONGO_URL"]
+
 # I DON'T KNOW LIKE THIS HACKER
 load_dotenv()
 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_DALLE3XL_URL = os.environ["SOURCE_DALLE3XL_URL"]
 SOURCE_WAIFU_URL = "https://api.waifu.pics"
 SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
 SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
@@ -109,6 +112,12 @@ app = FastAPI(
     docs_url="/"
 )
 
+contact_support = """
+We are aware that this service 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
+Contact Support @soulofsukuna
+"""
+
 def validate_api_key(api_key: str = Header(...)):
     USERS_API_KEYS = db.get_all_api_keys()
     if api_key not in USERS_API_KEYS:
@@ -243,6 +252,34 @@ def v1beta3_google_ai(
     except:
         return {"status": "false", "message": "Error response."}
 
+@app.post("/UFoP/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}"}
+        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=[{}])
+        )
+    if encoded_string:
+        return SuccessResponse(status="True", sukuna={"data": encoded_string})
+    else:
+        return SuccessResponse(status="False", sukuna={"data": contact_support})
+
 @app.get("/ryuzaki/unsplash")
 async def get_image_unsplash(query: str, size: str="500x500"):
     url = SOURCE_UNSPLASH_URL