Update app.py

app.py

@@ -1,4 +1,7 @@
 import gradio as gr
+from PIL import Image
+import requests
+from io import BytesIO
 import random
 
 def generate_image(prompt):
@@ -172,71 +175,52 @@ def generate_image(prompt):
         ]
 
         if "face" in prompt:
-            output = random.choice(faces)
-            return output
+            output_url = random.choice(faces)
         elif "landscape" in prompt:
-            output = random.choice(landscapes)
-            return output
+            output_url = random.choice(landscapes)
         elif "animal" in prompt:
-            output = random.choice(animal)
-            return output
+            output_url = random.choice(animal)
         elif "car" in prompt:
-            output = random.choice(car)
-            return output
+            output_url = random.choice(car)
         elif "food" in prompt:
-            output = random.choice(food)
-            return output
+            output_url = random.choice(food)
         elif "boat" in prompt:
-            output = random.choice(boat)
-            return output
+            output_url = random.choice(boat)
         elif "logo" in prompt:
-            output = random.choice(logo)
-            return output
+            output_url = random.choice(logo)
         elif "eye" in prompt:
-            output = random.choice(eye)
-            return output
-        elif "food" in prompt:
-            output = random.choice(food)
-            return output
+            output_url = random.choice(eye)
         elif "beard" in prompt:
-            output = random.choice(beard)
-            return output
+            output_url = random.choice(beard)
         elif "avatar" in prompt:
-            output = random.choice(avatar)
-            return output
+            output_url = random.choice(avatar)
         elif "dreadlocks" in prompt:
-            output = random.choice(dreadlocks)
-            return output
+            output_url = random.choice(dreadlocks)
         elif "soap" in prompt:
-            output = random.choice(soap)
-            return output
+            output_url = random.choice(soap)
         elif "sumsung" in prompt:
-            output = random.choice(sumsung)
-            return output
+            output_url = random.choice(sumsung)
         elif "iphone" in prompt:
-            output = random.choice(iphone)
-            return output
+            output_url = random.choice(iphone)
         elif "phone" in prompt:
-            output = random.choice(phone)
-            return output
+            output_url = random.choice(phone)
         elif "code" in prompt:
-            output = random.choice(code)
-            return output
+            output_url = random.choice(code)
         elif "youtube" in prompt:
-            output = random.choice(ytlogo)
-            return output
+            output_url = random.choice(ytlogo)
         elif "dog" in prompt:
-            output = random.choice(dog)
-            return output
+            output_url = random.choice(dog)
         elif "doll" in prompt:
-            output = random.choice(doll)
-            return output
+            output_url = random.choice(doll)
         elif "background" in prompt:
-            output = random.choice(background)
-            return output
+            output_url = random.choice(background)
         else:
             return "No image generated."
 
+        image = load_image(output_url)
+        return image
+
+
 iface = gr.Interface(
     fn=generate_image,
     inputs=gr.inputs.Textbox(label="Enter prompt"),
@@ -247,7 +231,6 @@ iface = gr.Interface(
         ["I want a face"],
         ["I want a landscape"],
         ["I want an animal"]
-        # Add more example prompts if desired
     ]
 )