Update main.py

main.py

@@ -59,6 +59,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_DALLE3XL_URL = os.environ["SOURCE_DALLE3XL_URL"]
+SOURCE_PROTEUSV0_2_URL = os.environ["SOURCE_PROTEUSV0_2_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"]
@@ -141,12 +142,12 @@ def validate_api_key_only_devs(api_key: str = Header(...)):
 
 RAMDOM_STATUS = [
     "Spammer",
-    "wanted",
-    "scammer",
-    "rogue_agent",
-    "pornbot prolly",
-    "fugitive",
-    "simp",
+    "Wanted",
+    "Scammer",
+    "Rogue_Agent",
+    "PornBot_Prolly",
+    "Fugitive",
+    "SIMP",
 ]
 
 
@@ -468,6 +469,33 @@ def dalle_3xl(item: Dalle3XL, api_key: None = Depends(validate_api_key)):
         return SuccessResponse(status="False", randydev={"data": contact_support})
 
 
+@app.post("/UFoP/proteusv0_2")
+def proteusv0_2(item: ProteusV02, api_key: None = Depends(validate_api_key)):
+    API_URL = SOURCE_PROTEUSV0_2_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=internal_error)
+
+    try:
+        encoded_string = base64.b64encode(response.content).decode("utf-8")
+    except Exception:
+        raise HTTPException(status_code=500, detail=contact_support)
+
+    if encoded_string:
+        return SuccessResponse(status="True", randydev={"data": encoded_string})
+    else:
+        return SuccessResponse(status="False", randydev={"data": contact_support})
+
+
 @app.get("/ryuzaki/unsplash")
 async def get_image_unsplash(query: str, size: str = "500x500"):
     url = SOURCE_UNSPLASH_URL