Update app.py

app.py

@@ -3,30 +3,47 @@ from PIL import Image, ImageDraw, ImageFont
 import random
 
 def generate_image(prompt):
-    # Create blank image
-    img = Image.new("RGB", (512, 512), color=(245, 245, 245))
+    # Create blank canvas
+    img = Image.new("RGB", (512, 512), (20, 20, 30))
     draw = ImageDraw.Draw(img)
 
-    # Random sketch lines (fake AI vibe)
-    for _ in range(200):
-        x1 = random.randint(0, 512)
-        y1 = random.randint(0, 512)
-        x2 = random.randint(0, 512)
-        y2 = random.randint(0, 512)
-        draw.line((x1, y1, x2, y2), fill=(0, 0, 0), width=1)
+    # Add background gradient
+    for i in range(512):
+        color = (20 + i // 20, 20 + i // 25, 30 + i // 15)
+        draw.line([(0, i), (512, i)], fill=color)
 
-    # Write prompt text
-    draw.rectangle((0, 460, 512, 512), fill=(0, 0, 0))
-    draw.text((10, 470), f"Prompt: {prompt}", fill=(255, 255, 255))
+    # 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)
+    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.rectangle([0, 470, 512, 512], fill=(0, 0, 0))
+    draw.text((10, 472), f"Prompt: {prompt[:30]}", fill="white")
 
     return img
 
-app = gr.Interface(
+demo = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Describe the image"),
     outputs=gr.Image(label="Generated Image"),
     title="⚡ Dave AI — Instant Image Generator",
-    description="Ultra-fast • CPU-only • No tokens • No GPU"
+    description="Fast CPU-only generator. No tokens. No GPU.",
 )
 
-app.launch()
+demo.launch()