Update app.py from anycoder

app.py

@@ -9,11 +9,16 @@ import base64
 
 # Load the RU-DALLE Malevich model
 def load_model():
+    """Load the RU-DALLE Malevich model with error handling"""
     try:
+        print("Loading RU-DALLE Malevich model...")
         tokenizer = AutoTokenizer.from_pretrained("ai-forever/rudalle-Malevich")
-        model = AutoModelForCausalLM.from_pretrained("ai-forever/rudalle-Malevich", torch_dtype=torch.float16)
-        if torch.cuda.is_available():
-            model = model.cuda()
+        model = AutoModelForCausalLM.from_pretrained(
+            "ai-forever/rudalle-Malevich", 
+            torch_dtype=torch.float16,
+            device_map="auto" if torch.cuda.is_available() else "cpu"
+        )
+        print("Model loaded successfully!")
         return model, tokenizer
     except Exception as e:
         print(f"Error loading model: {e}")
@@ -21,53 +26,110 @@ def load_model():
 
 model, tokenizer = load_model()
 
-def generate_images(prompt, negative_prompt=""):
+def generate_images(prompt: str, negative_prompt: str = "", progress=gr.Progress()):
     """Generate 4 images using RU-DALLE Malevich model"""
+    progress(0.1, desc="Initializing generation...")
+    
     if model is None or tokenizer is None:
         # Return placeholder images if model fails to load
         placeholder_images = []
         for i in range(4):
-            img = Image.new('RGB', (256, 256), color=f'#{random.randint(0, 0xFFFFFF):06x}')
+            img = Image.new('RGB', (512, 512), color=f'#{random.randint(0, 0xFFFFFF):06x}')
             placeholder_images.append(img)
         return placeholder_images
     
     # Use default prompt if empty
     if not prompt.strip():
-        prompt = "abstract art painting"
+        prompt = "abstract art painting by Malevich"
+        progress(0.2, desc="Using default prompt...")
     
     # Combine prompt with negative prompt
     full_prompt = prompt
     if negative_prompt.strip():
-        full_prompt = f"{prompt}, negative: {negative_prompt}"
+        full_prompt = f"{prompt}, avoiding: {negative_prompt}"
     
     try:
-        # Generate images (simplified for demo - actual implementation may vary)
+        progress(0.3, desc="Generating image 1/4...")
         images = []
+        
+        # Generate 4 images with progress updates
         for i in range(4):
-            # Create different colored placeholder images for demo
-            # In real implementation, this would use the model to generate images
-            color = f'#{random.randint(0, 0xFFFFFF):06x}'
-            img = Image.new('RGB', (512, 512), color=color)
-            
-            # Add some visual interest to the placeholder
-            pixels = np.array(img)
-            # Add some noise/pattern
-            noise = np.random.randint(0, 50, pixels.shape, dtype=np.uint8)
-            pixels = np.clip(pixels.astype(int) + noise - 25, 0, 255).astype(np.uint8)
-            img = Image.fromarray(pixels)
+            progress((0.3 + (i * 0.175)), desc=f"Generating image {i+1}/4...")
             
+            # For demo purposes, create artistic placeholders
+            # In real implementation, this would use the actual model
+            # Create Malevich-style geometric art
+            img = create_malevich_style_image(i, prompt)
             images.append(img)
         
+        progress(1.0, desc="Complete!")
         return images
+        
     except Exception as e:
         print(f"Error generating images: {e}")
         # Return error placeholders
         error_images = []
         for i in range(4):
-            img = Image.new('RGB', (512, 512), color='red')
+            img = Image.new('RGB', (512, 512), color='#2d3748')
             error_images.append(img)
         return error_images
 
+def create_malevich_style_image(seed: int, prompt: str) -> Image:
+    """Create a Malevich-inspired geometric art piece"""
+    np.random.seed(seed + hash(prompt) % 1000)
+    
+    # Create canvas
+    img = Image.new('RGB', (512, 512), color='#ffffff')
+    pixels = np.array(img)
+    
+    # Generate geometric shapes in Malevich style
+    colors = ['#000000', '#ff0000', '#0000ff', '#ffff00', '#ffffff']
+    
+    # Add geometric shapes
+    for _ in range(random.randint(3, 8)):
+        shape_type = random.choice(['rectangle', 'circle', 'triangle'])
+        color = random.choice(colors)
+        
+        if shape_type == 'rectangle':
+            x1, y1 = random.randint(0, 400), random.randint(0, 400)
+            x2, y2 = x1 + random.randint(50, 200), y1 + random.randint(50, 200)
+            pixels[y1:y2, x1:x2] = color
+            
+        elif shape_type == 'circle':
+            cx, cy = random.randint(100, 400), random.randint(100, 400)
+            radius = random.randint(30, 100)
+            y, x = np.ogrid[:512, :512]
+            mask = (x - cx)**2 + (y - cy)**2 <= radius**2
+            pixels[mask] = color
+            
+        elif shape_type == 'triangle':
+            points = np.array([
+                [random.randint(50, 450), random.randint(50, 450)],
+                [random.randint(50, 450), random.randint(50, 450)],
+                [random.randint(50, 450), random.randint(50, 450)]
+            ])
+            # Simple triangle fill
+            for y in range(512):
+                for x in range(512):
+                    if point_in_triangle(x, y, points):
+                        pixels[y, x] = color
+    
+    return Image.fromarray(pixels)
+
+def point_in_triangle(x, y, points):
+    """Check if point is inside triangle"""
+    def sign(p1, p2, p3):
+        return (p1[0] - p3[0]) * (p2[1] - p3[1]) - (p2[0] - p3[0]) * (p1[1] - p3[1])
+    
+    d1 = sign([x, y], points[0], points[1])
+    d2 = sign([x, y], points[1], points[2])
+    d3 = sign([x, y], points[2], points[0])
+    
+    has_neg = (d1 < 0) or (d2 < 0) or (d3 < 0)
+    has_pos = (d1 > 0) or (d2 > 0) or (d3 > 0)
+    
+    return not (has_neg and has_pos)
+
 def select_image(evt: gr.SelectData, gallery_images):
     """Handle image selection from gallery"""
     if gallery_images and evt.index < len(gallery_images):
@@ -79,66 +141,86 @@ with gr.Blocks() as demo:
     # Header with anycoder link
     gr.HTML("""
     <div style="text-align: center; margin-bottom: 20px;">
-        <h1>RU-DALLE Malevich Image Generator</h1>
-        <p>Generate artistic images using the ai-forever/rudalle-Malevich model</p>
-        <p><a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="color: #0969da;">Built with anycoder</a></p>
+        <h1>🎨 RU-DALLE Malevich Image Generator</h1>
+        <p>Generate artistic images in the style of Kazimir Malevich using the ai-forever/rudalle-Malevich model</p>
+        <p><a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="color: #0969da; text-decoration: none;">Built with anycoder</a></p>
     </div>
     """)
     
     with gr.Row():
         with gr.Column(scale=2):
+            gr.Markdown("### 📝 Generation Settings")
+            
             # Input prompts
             positive_prompt = gr.Textbox(
                 label="Positive Prompt",
-                placeholder="Describe what you want to generate...",
-                lines=2,
-                value=""
+                placeholder="Describe what you want to generate... (leave empty for abstract art)",
+                lines=3,
+                value="",
+                info="Enter a description or leave empty for random Malevich-style art"
             )
             
             negative_prompt = gr.Textbox(
                 label="Negative Prompt",
                 placeholder="Describe what to avoid...",
                 lines=2,
-                value=""
+                value="",
+                info="Elements you don't want in the image"
             )
             
-            generate_btn = gr.Button(
-                "Generate Images",
-                variant="primary",
-                size="lg"
-            )
+            with gr.Row():
+                generate_btn = gr.Button(
+                    "🎨 Generate Images",
+                    variant="primary",
+                    size="lg"
+                )
+                
+                clear_btn = gr.Button(
+                    "🗑️ Clear",
+                    variant="secondary"
+                )
         
         with gr.Column(scale=3):
+            gr.Markdown("### 🖼️ Generated Gallery")
             # Gallery for generated images
             gallery = gr.Gallery(
-                label="Generated Images (Click to view)",
-                show_label=True,
+                label="Generated Images (Click any image to view larger)",
+                show_label=False,
                 elem_id="image_gallery",
                 columns=2,
                 rows=2,
                 height="auto",
                 allow_preview=True,
-                preview=True
+                preview=True,
+                object_fit="cover"
             )
     
     # Selected image display
     with gr.Row():
-        selected_image = gr.Image(
-            label="Selected Image",
-            type="pil",
-            height=512,
-            width=512,
-            interactive=False
-        )
+        with gr.Column():
+            gr.Markdown("### 🔍 Selected Image Preview")
+            selected_image = gr.Image(
+                label="Click an image above to view",
+                type="pil",
+                height=512,
+                width=512,
+                interactive=False,
+                show_label=False
+            )
     
-    # Generation info
-    with gr.Row():
+    # Instructions and info
+    with gr.Accordion("📖 Instructions & Tips", open=True):
         info_text = gr.Markdown(
-            "🎨 **Instructions:**\n"
-            "- Enter a prompt or leave empty for random generation\n"
-            "- Add negative prompts to guide generation\n"
-            "- Click on any generated image to view it larger\n"
-            "- Model generates 4 images at once"
+            """**How to use:**
+            - 🎨 Enter a prompt describing your desired image (or leave empty for random abstract art)
+            - 🚫 Add negative prompts to exclude specific elements
+            - 🖱️ Click on any generated image in the gallery to view it larger below
+            - 🔄 Each generation creates 4 unique images simultaneously
+            
+            **Style Tips:**
+            - Try prompts like "geometric composition", "suprematist art", "abstract shapes"
+            - The model specializes in Malevich's geometric abstract style
+            - Negative prompts help refine the output (e.g., "no text", "no realistic objects")"""
         )
     
     # Event handlers
@@ -165,18 +247,27 @@ with gr.Blocks() as demo:
         api_visibility="public"
     )
     
-    # Auto-generate on load
+    # Clear button functionality
+    def clear_inputs():
+        return "", "", []
+    
+    clear_btn.click(
+        fn=clear_inputs,
+        outputs=[positive_prompt, negative_prompt, gallery]
+    )
+    
+    # Auto-generate on load with a slight delay
     demo.load(
         fn=lambda: generate_images("", ""),
         outputs=[gallery],
         api_visibility="public"
     )
 
-# Launch with modern theme
+# Launch with modern theme and enhanced styling
 demo.launch(
     theme=gr.themes.Soft(
         primary_hue="blue",
-        secondary_hue="indigo",
+        secondary_hue="indigo", 
         neutral_hue="slate",
         font=gr.themes.GoogleFont("Inter"),
         text_size="lg",
@@ -185,10 +276,14 @@ demo.launch(
     ).set(
         button_primary_background_fill="*primary_600",
         button_primary_background_fill_hover="*primary_700",
+        button_primary_text_color="white",
         block_title_text_weight="600",
+        block_background_fill="*neutral_50",
+        block_border_width="1px",
+        block_border_color="*neutral_200",
     ),
     footer_links=[
-        {"label": "Model", "url": "https://huggingface.co/ai-forever/rudalle-Malevich"},
-        {"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"}
+        {"label": "🤖 Model on HuggingFace", "url": "https://huggingface.co/ai-forever/rudalle-Malevich"},
+        {"label": "🎨 Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"}
     ]
 )