Update app.py

app.py

@@ -172,78 +172,80 @@ def generate_image(prompt):
         ]
 
         if "face" in prompt:
-            output = random.choice(faces)
-            return output
-        elif "landscape" in prompt:
-            output = random.choice(landscapes)
-            return output
-        elif "animal" in prompt:
-            output = random.choice(animal)
-            return output
-        elif "car" in prompt:
-            output = random.choice(car)
-            return output
-        elif "food" in prompt:
-            output = random.choice(food)
-            return output
-        elif "boat" in prompt:
-            output = random.choice(boat)
-            return output
-        elif "logo" in prompt:
-            output = random.choice(logo)
-            return output
-        elif "eye" in prompt:
-            output = random.choice(eye)
-            return output
-        elif "food" in prompt:
-            output = random.choice(food)
-            return output
-        elif "beard" in prompt:
-            output = random.choice(beard)
-            return output
-        elif "avatar" in prompt:
-            output = random.choice(avatar)
-            return output
-        elif "dreadlocks" in prompt:
-            output = random.choice(dreadlocks)
-            return output
-        elif "soap" in prompt:
-            output = random.choice(soap)
-            return output
-        elif "sumsung" in prompt:
-            output = random.choice(sumsung)
-            return output
-        elif "iphone" in prompt:
-            output = random.choice(iphone)
-            return output
-        elif "phone" in prompt:
-            output = random.choice(phone)
-            return output
-        elif "code" in prompt:
-            output = random.choice(code)
-            return output
-        elif "youtube" in prompt:
-            output = random.choice(ytlogo)
-            return output
-        elif "dog" in prompt:
-            output = random.choice(dog)
-            return output
-        elif "doll" in prompt:
-            output = random.choice(doll)
-            return output
-        elif "background" in prompt:
-            output = random.choice(background)
-            return output
-        else:
-            return "No image generated."
+        output = random.choice(faces)
+        return output
+    elif "landscape" in prompt:
+        output = random.choice(landscapes)
+        return output
+    elif "animal" in prompt:
+        output = random.choice(animal)
+        return output
+    elif "car" in prompt:
+        output = random.choice(car)
+        return output
+    elif "food" in prompt:
+        output = random.choice(food)
+        return output
+    elif "boat" in prompt:
+        output = random.choice(boat)
+        return output
+    elif "logo" in prompt:
+        output = random.choice(logo)
+        return output
+    elif "eye" in prompt:
+        output = random.choice(eye)
+        return output
+    elif "beard" in prompt:
+        output = random.choice(beard)
+        return output
+    elif "avatar" in prompt:
+        output = random.choice(avatar)
+        return output
+    elif "dreadlocks" in prompt:
+        output = random.choice(dreadlocks)
+        return output
+    elif "soap" in prompt:
+        output = random.choice(soap)
+        return output
+    elif "sumsung" in prompt:
+        output = random.choice(sumsung)
+        return output
+    elif "iphone" in prompt:
+        output = random.choice(iphone)
+        return output
+    elif "phone" in prompt:
+        output = random.choice(phone)
+        return output
+    elif "code" in prompt:
+        output = random.choice(code)
+        return output
+    elif "youtube" in prompt:
+        output = random.choice(ytlogo)
+        return output
+    elif "dog" in prompt:
+        output = random.choice(dog)
+        return output
+    elif "doll" in prompt:
+        output = random.choice(doll)
+        return output
+    elif "background" in prompt:
+        output = random.choice(background)
+        return output
+    else:
+        return "No image generated."
 
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.inputs.Textbox(label="Enter prompt"),  # Add an input field of type Textbox
+    inputs=gr.inputs.Textbox(label="Enter prompt"),
     outputs="text",
     title="KVIImager 2.0 - Image Generator",
     description="Generate images based on a prompt.",
-    examples=[["Blue man avatar standing in rainforest"], ["Realistic landscape in mountains"]]
+    examples=[
+        ["I want a face"],
+        ["I want a landscape"],
+        ["I want an animal"]
+        # Add more example prompts if desired
+    ]
 )
 
 iface.launch()