Update app.py

app.py

@@ -11,10 +11,15 @@ glif_token_id = 1
 glif_tokens_tried = 0
 no_of_accounts = 3
 
+import gradio as gr
+from PIL import Image
+from io import BytesIO
+import requests
+import os
+
 def get_image_from_url(url):
     """
-    Loads and returns an image from a hardcoded URL.
-    The prompt is ignored in this version.
+    Fetches and returns an image from a given URL.
     """
     try:
         response = requests.get(url, stream=True)
@@ -25,40 +30,40 @@ def get_image_from_url(url):
         return f"Error fetching image: {e}"
     except Exception as e:
         return f"Error processing image: {e}"
-        
+
 def generate_image(prompt, aspect_ratio, realism):
     global glif_token_id
     global GLIF_API_TOKEN
     global glif_tokens_tried
     global no_of_accounts
-    
+
     payload = {
         "id": "cm3ugmzv2002gnckiosrwk6xi",  # Replace with appropriate ID if necessary
         "inputs": [prompt, aspect_ratio, str(realism).lower()],
     }
     headers = {"Authorization": f"Bearer {GLIF_API_TOKEN}"}
-    
+
     try:
         response_data = requests.post(GLIF_API_URL, json=payload, headers=headers).json()
         if "error" in response_data:
             if 'error 429' in response_data['error']:
-                if glif_tokens_tried<no_of_accounts:
-                    glif_token_id = (glif_token_id+1)%no_of_accounts
-                    glif_tokens_tried+=1
+                if glif_tokens_tried < no_of_accounts:
+                    glif_token_id = (glif_token_id + 1) % no_of_accounts
+                    glif_tokens_tried += 1
                     GLIF_API_TOKEN = os.getenv(f"GLIF_API_TOKEN{glif_token_id}")
                     response_data = generate_image(prompt, aspect_ratio, realism)
                     glif_tokens_tried = 0
                     return response_data
-                response_data = "No credits available"
+                return "No credits available"
             return response_data
         elif "output" in response_data:
             url = response_data['output']
-            image  = get_image_from_url(url)
-            return image
+            return get_image_from_url(url)  # Return the image directly
         else:
             return "Error: Unexpected response from server"
     except Exception as e:
-        return f"error"
+        return f"Error: {e}"
+
 # Define the Gradio interface
 interface = gr.Interface(
     fn=generate_image,
@@ -71,12 +76,10 @@ interface = gr.Interface(
         ),
         gr.Checkbox(label="Realism", value=True),  # Checkbox for realism (True/False)
     ],
-    outputs=[
-        gr.Textbox(label="Result"),
-    ],
+    outputs=gr.Image(type="pil"),  # Output image
     title="Image Generator",
     description="Provide a prompt, select an aspect ratio, and set realism to generate an image.",
 )
 
 # Launch the interface
-interface.launch()
+interface.launch()