Upload 3 files

README.md

@@ -4,7 +4,7 @@ emoji: 🎨
 colorFrom: purple
 colorTo: pink
 sdk: gradio
-sdk_version: 4.12.0
+sdk_version: 6.2.0
 app_file: app.py
 pinned: false
 license: mit
@@ -12,33 +12,44 @@ license: mit
 
 # 🎨 Pixazo Image Generator
 
-Generate stunning AI images using Pixazo's FLUX-1 Schnell model.
+Generate AI images using Pixazo's FLUX-1 Schnell model.
 
 ## ✨ Features
 
 - 🖼️ Generate multiple images at once
-- 🎭 Various style presets
-- ⚡ Adjustable quality vs speed
-- 🔒 Secure API key handling
+- 🎭 Various style presets (cyberpunk, fantasy, anime, etc.)
+- ⚡ Adjustable quality settings
+- 🔒 Secure API key handling via Hugging Face secrets
 - 📱 Responsive design
+- 🔄 Real-time progress updates
 
-## 🔧 Setup
+## 🚀 Quick Start
 
-1. Get your API key from [Pixazo](https://pixazo.ai)
-2. Add it to Hugging Face Space secrets:
+1. **Set your API key:**
    - Go to **Settings → Repository secrets**
    - Add: `PIXAZO_API_KEY = your_key_here`
-3. The app will automatically restart
 
-## 🚀 Usage Tips
+2. **Restart the Space** (from Settings page)
+
+3. **Start generating images!**
+
+## 🎯 Usage Tips
 
 - Be descriptive in your prompts
-- Start with 4 steps for quick testing
+- Start with 4 steps for quick results
 - Use seeds for reproducible results
-- Try different style presets
+- Try different style presets for varied effects
+
+## 🔧 Technical Details
+
+- **API Format:** `{"output": "IMAGE_URL"}`
+- **Framework:** Gradio 6.2.0
+- **Image Processing:** PIL/Pillow
+- **HTTP Client:** Requests
 
-## 📞 API Response Format
+## 📞 Support
 
-The Pixazo API returns:
-```json
-{"output": "https://image-url-here.png"}
+If you encounter issues:
+1. Check your API key is correctly set
+2. Verify the prompt follows guidelines
+3. Check the "Detailed Response" section for API errors

app.py

@@ -5,7 +5,6 @@ import os
 from PIL import Image
 import io
 import time
-import base64
 from datetime import datetime
 
 # Get API key from Hugging Face secrets
@@ -83,14 +82,15 @@ def call_pixazo_api(prompt, num_steps=4, seed=None, height=512, width=512,
                         return image, {
                             "success": True,
                             "status_code": 200,
-                            "image_url": image_url[:100] + "..." if len(image_url) > 100 else image_url,
+                            "image_url_preview": image_url[:80] + "..." if len(image_url) > 80 else image_url,
                             "image_size": f"{image.size[0]}x{image.size[1]}",
+                            "image_format": image.format if hasattr(image, 'format') else "Unknown",
                             "timestamp": datetime.now().isoformat()
                         }
                     else:
                         return None, {
                             "error": f"Failed to download image: HTTP {img_response.status_code}",
-                            "image_url": image_url[:100] + "..." if len(image_url) > 100 else image_url,
+                            "image_url": image_url[:80] + "..." if len(image_url) > 80 else image_url,
                             "download_status": img_response.status_code,
                             "timestamp": datetime.now().isoformat()
                         }
@@ -98,7 +98,7 @@ def call_pixazo_api(prompt, num_steps=4, seed=None, height=512, width=512,
                 except Exception as e:
                     return None, {
                         "error": f"Error downloading image: {str(e)}",
-                        "image_url": image_url[:100] + "..." if len(image_url) > 100 else image_url,
+                        "image_url": image_url[:80] + "..." if len(image_url) > 80 else image_url,
                         "timestamp": datetime.now().isoformat()
                     }
             else:
@@ -137,9 +137,9 @@ def call_pixazo_api(prompt, num_steps=4, seed=None, height=512, width=512,
             "timestamp": datetime.now().isoformat()
         }
 
-# Generate images
-def generate_images(prompt, num_steps, seed, height, width, style_preset, 
-                   num_images, guidance_scale):
+# Generate images with streaming updates
+def generate_images_stream(prompt, num_steps, seed, height, width, style_preset, 
+                          num_images, guidance_scale):
     
     images = []
     all_results = []
@@ -163,12 +163,16 @@ def generate_images(prompt, num_steps, seed, height, width, style_preset,
             
             all_results.append(result)
             
-            # Update status
-            yield [
-                images if images else None,
-                f"Progress: {i+1}/{num_images} images",
-                json.dumps(result, indent=2)
-            ]
+            # 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:
@@ -177,194 +181,270 @@ def generate_images(prompt, num_steps, seed, height, width, style_preset,
         except Exception as e:
             error_result = {"error": str(e), "image_index": i, "timestamp": datetime.now().isoformat()}
             all_results.append(error_result)
-            yield [
-                images if images else None,
-                f"Error on image {i+1}: {str(e)[:50]}",
-                json.dumps(error_result, indent=2)
-            ]
+            yield {
+                gallery: images if images else None,
+                status: f"Exception on image {i+1}: {str(e)[:50]}",
+                json_output: error_result
+            }
             break
     
-    # Prepare final output
+    # Final summary
     success_count = len(images)
     
     if success_count == 0:
-        status = f"❌ No images generated ({success_count}/{num_images} successful)"
-        final_images = None
+        final_status = f"❌ No images generated ({success_count}/{num_images} successful)"
     elif success_count < num_images:
-        status = f"⚠️ Partial success: {success_count}/{num_images} images"
-        final_images = images
+        final_status = f"⚠️ Partial success: {success_count}/{num_images} images"
     else:
-        status = f"🎉 Success! All {success_count} images generated"
-        final_images = images
+        final_status = f"🎉 Success! All {success_count} images generated"
     
     summary = {
-        "total_attempts": num_images,
-        "successful": success_count,
-        "failed": num_images - success_count,
-        "timestamp": datetime.now().isoformat(),
+        "summary": {
+            "total_attempts": num_images,
+            "successful": success_count,
+            "failed": num_images - success_count,
+            "timestamp": datetime.now().isoformat()
+        },
         "parameters": {
-            "prompt_length": len(prompt),
+            "prompt": prompt[:100] + "..." if len(prompt) > 100 else prompt,
             "steps": num_steps,
-            "size": f"{width}x{height}",
+            "seed_used": seed,
+            "resolution": f"{width}x{height}",
             "style": style_preset,
-            "guidance": guidance_scale
+            "num_images": num_images,
+            "guidance_scale": guidance_scale
         }
     }
     
-    yield [final_images, status, json.dumps(summary, indent=2)]
+    yield {
+        gallery: images if images else None,
+        status: final_status,
+        json_output: summary
+    }
 
-# Create Gradio interface
+# Create the Gradio interface
 with gr.Blocks(
     title="Pixazo Image Generator",
-    theme=gr.themes.Soft()
+    theme=gr.themes.Soft(primary_hue="purple", secondary_hue="pink"),
+    css="""
+    .success { color: #10b981 !important; }
+    .error { color: #ef4444 !important; }
+    .warning { color: #f59e0b !important; }
+    .container {
+        max-width: 1200px;
+        margin: 0 auto;
+    }
+    """
 ) as demo:
     
     # Header
     gr.Markdown("""
-    # 🎨 Pixazo Image Generator
-    
-    Generate images using Pixazo's FLUX-1 Schnell AI model
+    <div style="text-align: center;">
+        <h1>🎨 Pixazo AI Image Generator</h1>
+        <p>Create stunning images with FLUX-1 Schnell model</p>
+    </div>
     """)
     
     # API Status
     api_key = get_api_key()
     if api_key:
-        gr.Markdown("### ✅ API Key: Configured")
+        status_html = """
+        <div style="background: #d1fae5; padding: 10px 15px; border-radius: 8px; border-left: 4px solid #10b981; margin-bottom: 20px;">
+            <strong>✅ API Status:</strong> Connected (API key configured)
+        </div>
+        """
     else:
-        gr.Markdown("""
-        ### ❌ API Key: Not Configured
-        
-        **To set up:**
-        1. Go to Space Settings → Repository secrets
-        2. Add: `PIXAZO_API_KEY` = your API key
-        3. Click "Restart this Space"
-        """)
+        status_html = """
+        <div style="background: #fee2e2; padding: 15px; border-radius: 8px; border-left: 4px solid #ef4444; margin-bottom: 20px;">
+            <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():
-        with gr.Column(scale=1):
-            # Input Section
-            with gr.Group():
+        # Left column - Inputs
+        with gr.Column(scale=1, min_width=400):
+            with gr.Group(label="Generation Settings"):
                 prompt = gr.Textbox(
-                    label="Prompt",
-                    placeholder="Describe the image you want to generate...",
+                    label="Image Prompt",
+                    placeholder="Describe what you want to generate...",
                     lines=3,
-                    value="A beautiful sunset over mountains"
-                )
-            
-            with gr.Row():
-                style_preset = gr.Dropdown(
-                    label="Style Preset",
-                    choices=[
-                        "none", "cyberpunk", "fantasy", "anime", "photographic", 
-                        "digital-art", "comic", "3d-model", "pixel-art"
-                    ],
-                    value="none"
+                    max_lines=5
                 )
                 
-                guidance_scale = gr.Slider(
-                    label="Guidance Scale",
-                    minimum=1.0,
-                    maximum=20.0,
-                    value=7.5,
-                    step=0.5
-                )
-            
-            with gr.Row():
-                num_steps = gr.Slider(
-                    label="Number of Steps",
+                with gr.Row():
+                    style_preset = gr.Dropdown(
+                        label="Style Preset",
+                        choices=[
+                            "none", "cyberpunk", "fantasy", "anime", "photographic", 
+                            "digital-art", "comic", "3d-model", "pixel-art", 
+                            "isometric", "watercolor", "oil-painting"
+                        ],
+                        value="none",
+                        scale=2
+                    )
+                    
+                    guidance_scale = gr.Slider(
+                        label="Guidance",
+                        minimum=1.0,
+                        maximum=20.0,
+                        value=7.5,
+                        step=0.5,
+                        scale=1
+                    )
+                
+                with gr.Row():
+                    num_steps = gr.Slider(
+                        label="Steps",
+                        minimum=1,
+                        maximum=50,
+                        value=4,
+                        step=1,
+                        scale=1
+                    )
+                    
+                    seed = gr.Number(
+                        label="Seed",
+                        value=42,
+                        minimum=0,
+                        maximum=999999,
+                        scale=1
+                    )
+                
+                with gr.Row():
+                    width = gr.Slider(
+                        label="Width",
+                        minimum=256,
+                        maximum=1024,
+                        value=512,
+                        step=64,
+                        scale=1
+                    )
+                    
+                    height = gr.Slider(
+                        label="Height",
+                        minimum=256,
+                        maximum=1024,
+                        value=512,
+                        step=64,
+                        scale=1
+                    )
+                
+                num_images = gr.Slider(
+                    label="Number of Images",
                     minimum=1,
-                    maximum=50,
-                    value=4,
+                    maximum=8,
+                    value=1,
                     step=1
                 )
-                
-                seed = gr.Number(
-                    label="Seed",
-                    value=42,
-                    minimum=0,
-                    maximum=999999
-                )
             
             with gr.Row():
-                width = gr.Slider(
-                    label="Width",
-                    minimum=256,
-                    maximum=1024,
-                    value=512,
-                    step=64
-                )
-                
-                height = gr.Slider(
-                    label="Height",
-                    minimum=256,
-                    maximum=1024,
-                    value=512,
-                    step=64
-                )
-            
-            num_images = gr.Slider(
-                label="Number of Images",
-                minimum=1,
-                maximum=4,
-                value=1,
-                step=1
-            )
-            
-            generate_btn = gr.Button("Generate Images", variant="primary")
+                generate_btn = gr.Button("✨ Generate Images", variant="primary", scale=2)
+                clear_btn = gr.Button("🗑️ Clear", variant="secondary", scale=1)
             
-            # Examples
-            gr.Examples(
-                examples=[
-                    ["A cyberpunk cityscape with flying cars and neon lights at night", 4, 123, 512, 512, "cyberpunk", 1, 7.5],
-                    ["A magical dragon in a fantasy landscape, cinematic lighting", 4, 456, 512, 512, "fantasy", 1, 7.5],
-                    ["Cute anime character with magical powers", 4, 789, 512, 512, "anime", 1, 7.5],
-                    ["Photorealistic portrait of an astronaut", 6, 999, 768, 512, "photographic", 1, 8.0]
-                ],
-                inputs=[prompt, num_steps, seed, height, width, style_preset, num_images, guidance_scale],
-                label="Try these examples:"
-            )
+            # Quick examples
+            with gr.Accordion("💡 Example Prompts", open=False):
+                examples = [
+                    ("Cyberpunk city at night with neon lights and flying cars", "cyberpunk"),
+                    ("Majestic dragon on a mountain peak, fantasy art", "fantasy"),
+                    ("Cute anime cat in a magical forest", "anime"),
+                    ("Photorealistic portrait of an astronaut on Mars", "photographic"),
+                    ("Isometric office scene with plants and computers", "isometric"),
+                ]
+                
+                for example_prompt, example_style in examples:
+                    with gr.Row():
+                        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),
+                            outputs=[prompt, style_preset]
+                        )
         
-        with gr.Column(scale=2):
-            # Output Section
+        # Right column - Outputs
+        with gr.Column(scale=2, min_width=600):
             status = gr.Textbox(
                 label="Status",
-                value="Ready to generate..."
+                value="Ready to generate images...",
+                interactive=False
             )
             
             gallery = gr.Gallery(
                 label="Generated Images",
                 columns=2,
-                height="auto"
+                height="auto",
+                object_fit="contain",
+                show_label=True,
+                preview=True
             )
             
-            # Response viewer
-            with gr.Accordion("API Response", open=False):
+            with gr.Accordion("📊 Detailed Response", open=False):
                 json_output = gr.JSON(
-                    label="Response"
+                    label="API Response",
+                    container=True
                 )
     
     # Footer
     gr.Markdown("""
     ---
-    **Note:** The Pixazo API returns `{"output": "IMAGE_URL"}`, and the app downloads the image from that URL.
-    
-    Make sure to set `PIXAZO_API_KEY` in your Space secrets for the app to work.
+    <div style="text-align: center; color: #666; font-size: 0.9em; margin-top: 20px;">
+        <p><strong>API Response Format:</strong> <code>{"output": "IMAGE_URL"}</code></p>
+        <p>Built with Gradio • Powered by Pixazo FLUX-1 Schnell</p>
+    </div>
     """)
     
-    # Connect button to function
-    generate_btn.click(
-        fn=generate_images,
-        inputs=[prompt, num_steps, seed, height, width, style_preset, num_images, guidance_scale],
-        outputs=[gallery, status, json_output]
-    )
+    # 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"}
+            }
+        
+        # Show initial status
+        yield {
+            status: f"🚀 Starting generation of {num_images} image(s)...",
+            gallery: None,
+            json_output: {"status": "starting", "num_images": num_images}
+        }
+        
+        # Stream generation results
+        for update in generate_images_stream(prompt, num_steps, seed, height, width, style_preset, num_images, guidance_scale):
+            yield update
+    
+    @clear_btn.click()
+    def on_clear():
+        return {
+            status: "Ready to generate images...",
+            gallery: None,
+            json_output: None
+        }
 
 # Launch the app
 if __name__ == "__main__":
-    print("Starting Pixazo Image Generator...")
-    print(f"API Key configured: {'Yes' if get_api_key() else 'No'}")
+    print("=" * 60)
+    print("🚀 Pixazo Image Generator")
+    print(f"📦 Gradio Version: {gr.__version__}")
+    print(f"🔑 API Key Configured: {'Yes' if get_api_key() else 'No'}")
+    print("=" * 60)
     
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=False
+        share=False,
+        show_api=False
     )

requirements.txt

@@ -1,3 +1,3 @@
-gradio>=3.41.0,<4.0.0
-pillow>=10.0.0
-requests>=2.31.0
+gradio==6.2.0
+pillow==10.4.0
+requests==2.32.3