Update app.py

app.py

@@ -1,9 +1,7 @@
 import gradio as gr
 import requests
-import base64
-from PIL import Image
-from io import BytesIO
 
+# Function to call the API and generate the image
 def generate_image(content, style):
     headers = {
         'Authorization': 'Bearer ZZUIQ4OZASNRQ8B8WYHNW',
@@ -14,20 +12,21 @@ def generate_image(content, style):
     }
     response = requests.post('https://api.fliki.ai/v1/generate/text-to-image', headers=headers, json=json_data)
     
-    # Assuming the response contains the image in base64 format
-    image_data = response.json().get('image', '')
-    if image_data:
-        image = Image.open(BytesIO(base64.b64decode(image_data)))
-        return image
-    else:
-        return "Error: No image returned"
+    # Handle the response here based on the API's response format
+    # If the response is a URL:
+    # return response.json()['url']
 
+    # If the response is binary data:
+    return response.content
+
+# Define the Gradio interface
 iface = gr.Interface(
     fn=generate_image,
     inputs=[gr.Textbox(label="Content"), gr.Textbox(label="Style")],
-    outputs="image",
+    outputs=gr.Image(type="auto"),  # 'auto' will automatically handle URLs or binary data
     title="Text to Image Generator",
     description="Enter a description and style to generate an image."
 )
 
+# Launch the interface
 iface.launch()