Upload main.py

main.py

@@ -594,52 +594,26 @@ def bing_dalle(item: BingDalle, api_key: None = Depends(validate_api_key)):
     except BaseException as e:
         return SuccessResponse(status="False", randydev={"data": f"Error: {e}"})
 
+def get_image_urls(query):
+    response = requests.get("https://ufoptg-ufop-api.hf.space/dall-e-3/prompt=" + query, headers=headers).json()
+    return [response[f"S{i}-Image"] for i in range(1, 5)]
+
 @app.post("/ryuzaki/dalle3xl")
 def dalle_3xl(
     item: Dalle3XL,
     api_key: None = Depends(validate_api_key)
 ):
-    API_URL = f"https://ufoptg-ufop-api.hf.space/dall-e-3/prompt={item.query}"
     try:
-        headers = {"accept":"application/json", "Content-Type": "application/json"}
-        response = requests.get(API_URL, headers=headers)
-        response.raise_for_status()
-    except requests.exceptions.RequestException:
-        raise HTTPException(
-            status_code=500,
-            detail=internal_error
-        )
-
-    try:
-        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")
+        x_image = get_image_urls(item.query)
     except Exception:
         raise HTTPException(
             status_code=500,
             detail=contact_support
         )
-
-    if encoded_string and encoded_string_2 and encoded_string_3 and encoded_string_4:
+    if x_image:
         return SuccessResponse(
             status="True",
-            randydev=
-            {
-                "data": encoded_string,
-                "data_2": encoded_string_2,
-                "data_3": encoded_string_3,
-                "data_4": encoded_string_4
-            }
-        )
+            randydev={"url": x_image})
     else:
         return SuccessResponse(status="False", randydev={"data": contact_support})