Upload 3 files

app.py

@@ -137,138 +137,72 @@ def call_pixazo_api(prompt, num_steps=4, seed=None, height=512, width=512,
             "timestamp": datetime.now().isoformat()
         }
 
-# Generate images with streaming updates
-def generate_images_stream(prompt, num_steps, seed, height, width, style_preset, 
-                          num_images, guidance_scale):
+# Create the Gradio interface with correct Gradio 6.x API
+with gr.Blocks(title="Pixazo Image Generator") as demo:
     
-    images = []
-    all_results = []
-    
-    for i in range(num_images):
-        try:
-            current_seed = seed + i if seed > 0 else 0
-            
-            image, result = call_pixazo_api(
-                prompt=prompt,
-                num_steps=num_steps,
-                seed=current_seed,
-                height=height,
-                width=width,
-                style_preset=style_preset,
-                guidance_scale=guidance_scale
-            )
-            
-            if image:
-                images.append(image)
-            
-            all_results.append(result)
-            
-            # Yield intermediate results for streaming
-            status_text = f"Generated {i+1}/{num_images} images"
-            if "error" in result:
-                status_text = f"Error on image {i+1}: {result['error'][:50]}..."
-            
-            yield {
-                gallery: images if images else None,
-                status: status_text,
-                json_output: result
-            }
-            
-            # Small delay between requests
-            if i < num_images - 1:
-                time.sleep(1)
-                
-        except Exception as e:
-            error_result = {"error": str(e), "image_index": i, "timestamp": datetime.now().isoformat()}
-            all_results.append(error_result)
-            yield {
-                gallery: images if images else None,
-                status: f"Exception on image {i+1}: {str(e)[:50]}",
-                json_output: error_result
-            }
-            break
-    
-    # Final summary
-    success_count = len(images)
-    
-    if success_count == 0:
-        final_status = f"❌ No images generated ({success_count}/{num_images} successful)"
-    elif success_count < num_images:
-        final_status = f"⚠️ Partial success: {success_count}/{num_images} images"
-    else:
-        final_status = f"🎉 Success! All {success_count} images generated"
-    
-    summary = {
-        "summary": {
-            "total_attempts": num_images,
-            "successful": success_count,
-            "failed": num_images - success_count,
-            "timestamp": datetime.now().isoformat()
-        },
-        "parameters": {
-            "prompt": prompt[:100] + "..." if len(prompt) > 100 else prompt,
-            "steps": num_steps,
-            "seed_used": seed,
-            "resolution": f"{width}x{height}",
-            "style": style_preset,
-            "num_images": num_images,
-            "guidance_scale": guidance_scale
-        }
-    }
-    
-    yield {
-        gallery: images if images else None,
-        status: final_status,
-        json_output: summary
-    }
-
-# Create the Gradio interface
-with gr.Blocks(
-    title="Pixazo Image Generator",
-    theme=gr.themes.Soft(primary_hue="purple", secondary_hue="pink"),
-    css="""
+    # Custom CSS
+    demo.css = """
     .success { color: #10b981 !important; }
     .error { color: #ef4444 !important; }
     .warning { color: #f59e0b !important; }
-    .container {
-        max-width: 1200px;
-        margin: 0 auto;
+    .api-status {
+        padding: 12px 16px;
+        border-radius: 8px;
+        margin-bottom: 20px;
+    }
+    .api-status-success {
+        background: #d1fae5;
+        border-left: 4px solid #10b981;
+    }
+    .api-status-error {
+        background: #fee2e2;
+        border-left: 4px solid #ef4444;
+    }
+    .example-prompt {
+        cursor: pointer;
+        padding: 10px;
+        border-radius: 6px;
+        margin: 5px 0;
+        background: #f8fafc;
+        border: 1px solid #e2e8f0;
+    }
+    .example-prompt:hover {
+        background: #e2e8f0;
     }
     """
-) as demo:
     
     # Header
     gr.Markdown("""
-    <div style="text-align: center;">
-        <h1>🎨 Pixazo AI Image Generator</h1>
-        <p>Create stunning images with FLUX-1 Schnell model</p>
-    </div>
+    # 🎨 Pixazo AI Image Generator
+    
+    Create stunning images with FLUX-1 Schnell model
     """)
     
     # API Status
     api_key = get_api_key()
     if api_key:
-        status_html = """
-        <div style="background: #d1fae5; padding: 10px 15px; border-radius: 8px; border-left: 4px solid #10b981; margin-bottom: 20px;">
+        gr.Markdown("""
+        <div class="api-status api-status-success">
             <strong>✅ API Status:</strong> Connected (API key configured)
         </div>
-        """
+        """)
     else:
-        status_html = """
-        <div style="background: #fee2e2; padding: 15px; border-radius: 8px; border-left: 4px solid #ef4444; margin-bottom: 20px;">
+        gr.Markdown("""
+        <div class="api-status api-status-error">
             <strong>❌ API Status:</strong> Not Configured
             <p style="margin: 8px 0 0 0; font-size: 0.9em;">
                 <strong>Setup:</strong> Go to Space Settings → Repository secrets → Add: <code>PIXAZO_API_KEY</code> = your key
             </p>
         </div>
-        """
-    
-    gr.HTML(status_html)
+        """)
     
     with gr.Row():
         # Left column - Inputs
         with gr.Column(scale=1, min_width=400):
-            with gr.Group(label="Generation Settings"):
+            # Generation Settings group
+            with gr.Group():
+                gr.Markdown("### 🎨 Generation Settings")
+                
                 prompt = gr.Textbox(
                     label="Image Prompt",
                     placeholder="Describe what you want to generate...",
@@ -289,7 +223,7 @@ with gr.Blocks(
                     )
                     
                     guidance_scale = gr.Slider(
-                        label="Guidance",
+                        label="Guidance Scale",
                         minimum=1.0,
                         maximum=20.0,
                         value=7.5,
@@ -299,7 +233,7 @@ with gr.Blocks(
                 
                 with gr.Row():
                     num_steps = gr.Slider(
-                        label="Steps",
+                        label="Number of Steps",
                         minimum=1,
                         maximum=50,
                         value=4,
@@ -358,18 +292,19 @@ with gr.Blocks(
                 
                 for example_prompt, example_style in examples:
                     with gr.Row():
-                        gr.Textbox(
+                        prompt_text = gr.Textbox(
                             value=example_prompt,
                             show_label=False,
                             interactive=False,
                             scale=3
                         )
-                        gr.Button(
-                            "Use",
-                            size="sm",
-                            scale=1
-                        ).click(
-                            lambda p=example_prompt, s=example_style: (p, s),
+                        use_btn = gr.Button("Use", size="sm", scale=1)
+                        
+                        def use_example(p=example_prompt, s=example_style):
+                            return p, s
+                        
+                        use_btn.click(
+                            use_example,
                             outputs=[prompt, style_preset]
                         )
         
@@ -406,33 +341,90 @@ with gr.Blocks(
     """)
     
     # Event handlers
-    @generate_btn.click(inputs=[prompt, num_steps, seed, height, width, style_preset, num_images, guidance_scale])
     def on_generate(prompt, num_steps, seed, height, width, style_preset, num_images, guidance_scale):
         if not prompt.strip():
-            return {
-                status: "❌ Please enter a prompt first",
-                gallery: None,
-                json_output: {"error": "No prompt provided"}
-            }
+            yield "❌ Please enter a prompt first", None, {"error": "No prompt provided"}
+            return
+        
+        images = []
+        all_results = []
         
-        # Show initial status
-        yield {
-            status: f"🚀 Starting generation of {num_images} image(s)...",
-            gallery: None,
-            json_output: {"status": "starting", "num_images": num_images}
+        yield "🚀 Starting generation...", None, {"status": "starting"}
+        
+        for i in range(num_images):
+            try:
+                current_seed = seed + i if seed > 0 else 0
+                
+                image, result = call_pixazo_api(
+                    prompt=prompt,
+                    num_steps=num_steps,
+                    seed=current_seed,
+                    height=height,
+                    width=width,
+                    style_preset=style_preset,
+                    guidance_scale=guidance_scale
+                )
+                
+                if image:
+                    images.append(image)
+                
+                all_results.append(result)
+                
+                status_text = f"Generated {i+1}/{num_images} images"
+                if "error" in result:
+                    status_text = f"Image {i+1} error: {result['error'][:50]}..."
+                
+                yield status_text, images if images else None, result
+                
+                if i < num_images - 1:
+                    time.sleep(1)
+                    
+            except Exception as e:
+                error_result = {"error": str(e), "image_index": i}
+                all_results.append(error_result)
+                yield f"Exception on image {i+1}: {str(e)[:50]}", images if images else None, error_result
+                break
+        
+        # Final summary
+        success_count = len(images)
+        
+        if success_count == 0:
+            final_status = f"❌ No images generated ({success_count}/{num_images} successful)"
+        elif success_count < num_images:
+            final_status = f"⚠️ Partial success: {success_count}/{num_images} images"
+        else:
+            final_status = f"🎉 Success! All {success_count} images generated"
+        
+        summary = {
+            "summary": {
+                "total_attempts": num_images,
+                "successful": success_count,
+                "failed": num_images - success_count
+            },
+            "parameters": {
+                "prompt_length": len(prompt),
+                "steps": num_steps,
+                "resolution": f"{width}x{height}",
+                "style": style_preset
+            }
         }
         
-        # Stream generation results
-        for update in generate_images_stream(prompt, num_steps, seed, height, width, style_preset, num_images, guidance_scale):
-            yield update
+        yield final_status, images if images else None, summary
     
-    @clear_btn.click()
     def on_clear():
-        return {
-            status: "Ready to generate images...",
-            gallery: None,
-            json_output: None
-        }
+        return "Ready to generate images...", None, None
+    
+    # Connect buttons
+    generate_btn.click(
+        fn=on_generate,
+        inputs=[prompt, num_steps, seed, height, width, style_preset, num_images, guidance_scale],
+        outputs=[status, gallery, json_output]
+    )
+    
+    clear_btn.click(
+        fn=on_clear,
+        outputs=[status, gallery, json_output]
+    )
 
 # Launch the app
 if __name__ == "__main__":
@@ -446,5 +438,7 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         share=False,
-        show_api=False
+        show_api=False,
+        theme=gr.themes.Soft(primary_hue="purple", secondary_hue="pink"),
+        css=demo.css if hasattr(demo, 'css') else ""
     )