Upload main.py

main.py

@@ -599,14 +599,9 @@ def dalle_3xl(
     item: Dalle3XL,
     api_key: None = Depends(validate_api_key)
 ):
-    API_URL = SOURCE_DALLE3XL_URL
+    API_URL = f"https://ufoptg-ufop-api.hf.space/dall-e-3/prompt={item.query}"
     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 = requests.get(API_URL, headers=headers)
         response.raise_for_status()
     except requests.exceptions.RequestException:
         raise HTTPException(
@@ -615,15 +610,35 @@ def dalle_3xl(
         )
 
     try:
-        encoded_string = base64.b64encode(response.content).decode("utf-8")
+        data_1 = response.json()[0]
+        data_2 = response.json()[1]
+        data_3 = response.json()[2]
+        data_4 = response.json()[3]
+        data_response = requests.get(data_1)
+        data_response_2 = requests.get(data_2)
+        data_response_3 = requests.get(data_3)
+        data_response_4 = requests.get(data_4)
+        encoded_string = base64.b64encode(data_response.content).decode("utf-8")
+        encoded_string_2 = base64.b64encode(data_response_2.content).decode("utf-8")
+        encoded_string_3 = base64.b64encode(data_response_3.content).decode("utf-8")
+        encoded_string_4 = base64.b64encode(data_response_4.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})
+    if encoded_string and encoded_string_2 and encoded_string_3 and encoded_string_4:
+        return SuccessResponse(
+            status="True",
+            randydev=
+            {
+                "data": encoded_string,
+                "data_2": encoded_string_2,
+                "data_3": encoded_string_3,
+                "data_4": encoded_string_4
+            }
+        )
     else:
         return SuccessResponse(status="False", randydev={"data": contact_support})