Update app.py

app.py

@@ -2,37 +2,44 @@ import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
 import random
 
+def draw_rat(draw):
+    # Body
+    draw.ellipse((170, 240, 340, 380), outline="white", width=6)
+    # Head
+    draw.ellipse((120, 260, 210, 330), outline="white", width=6)
+    # Ears
+    draw.ellipse((130, 240, 160, 270), outline="white", width=6)
+    draw.ellipse((180, 240, 210, 270), outline="white", width=6)
+    # Tail
+    draw.line((340, 320, 450, 260), fill="white", width=6)
+    # Eyes
+    draw.ellipse((155, 285, 165, 295), fill="white")
+    draw.ellipse((185, 285, 195, 295), fill="white")
+
 def generate_image(prompt):
-    # Create blank canvas
     img = Image.new("RGB", (512, 512), (20, 20, 30))
     draw = ImageDraw.Draw(img)
 
-    # Add background gradient
+    # Background
     for i in range(512):
         color = (20 + i // 20, 20 + i // 25, 30 + i // 15)
         draw.line([(0, i), (512, i)], fill=color)
 
-    # Simple “cartoon” shapes based on prompt
-    shapes = ["circle", "square", "triangle", "star"]
-    shape = random.choice(shapes)
-
-    # Draw shape
-    if shape == "circle":
-        draw.ellipse([100, 120, 412, 420], outline="white", width=6)
-    elif shape == "square":
-        draw.rectangle([120, 120, 392, 392], outline="white", width=6)
-    elif shape == "triangle":
-        draw.polygon([(256, 120), (120, 420), (392, 420)], outline="white", width=6)
+    # Basic prompt detection
+    prompt_lower = prompt.lower()
+    if "rat" in prompt_lower:
+        draw_rat(draw)
+    elif "cat" in prompt_lower:
+        draw.ellipse((170, 240, 340, 380), outline="white", width=6)
+        draw.ellipse((120, 260, 210, 330), outline="white", width=6)
+        draw.ellipse((130, 240, 160, 270), outline="white", width=6)
+        draw.ellipse((180, 240, 210, 270), outline="white", width=6)
+        draw.line((340, 320, 450, 260), fill="white", width=6)
+        draw.ellipse((155, 285, 165, 295), fill="white")
+        draw.ellipse((185, 285, 195, 295), fill="white")
     else:
-        # star
-        draw.polygon(
-            [(256, 120), (290, 250), (420, 260), (310, 330),
-             (340, 460), (256, 370), (172, 460), (202, 330),
-             (92, 260), (222, 250)],
-            outline="white", width=6
-        )
-
-    # Add prompt text at bottom
+        draw.text((50, 250), "I can draw only cats & rats", fill="white")
+
     draw.rectangle([0, 470, 512, 512], fill=(0, 0, 0))
     draw.text((10, 472), f"Prompt: {prompt[:30]}", fill="white")