Update app.py

app.py

@@ -7,248 +7,95 @@ import json
 import time
 
 # ==================== FREE HUGGINGFACE INFERENCE API ====================
-# No API key required for public models! Free tier has rate limits (roughly 1-2 requests/minute)
 
 HF_INFERENCE_API = "https://api-inference.huggingface.co/models"
 
 def huggingface_generate(model_id, prompt, width=1024, height=1024, seed=0):
-    """
-    Generic Hugging Face Inference API caller
-    Works with Stable Diffusion, FLUX, and other diffusion models
-    """
     try:
         url = f"{HF_INFERENCE_API}/{model_id}"
-        
-        # Prepare payload - different models accept different parameters
         payload = {
             "inputs": prompt,
             "parameters": {
                 "width": width,
                 "height": height,
                 "seed": seed if seed > 0 else int(time.time()),
-                "num_inference_steps": 25,  # Lower = faster, default for most models
+                "num_inference_steps": 25,
                 "guidance_scale": 7.5,
             }
         }
-        
-        # For some models, we need to send just the string in inputs
-        headers = {
-            "Content-Type": "application/json",
-            # No Authorization header needed for public models!
-        }
-        
+        headers = {"Content-Type": "application/json"}
         response = requests.post(url, json=payload, headers=headers, timeout=120)
-        
         if response.status_code == 200:
-            # Response is image bytes directly
-            return Image.open(BytesIO(response.content)), f"Hugging Face ({model_id})"
+            return Image.open(BytesIO(response.content)), f"HF ({model_id})"
         elif response.status_code == 503:
-            # Model is loading
-            return None, f"Model {model_id} is loading... Try again in 30 seconds"
+            return None, f"Model loading... wait 30s"
         else:
-            error_msg = response.text[:200] if response.text else f"HTTP {response.status_code}"
-            return None, f"HF Error: {error_msg}"
-            
+            return None, f"HTTP {response.status_code}: {response.text[:100]}"
     except Exception as e:
-        return None, f"HF Failed: {str(e)}"
+        return None, f"Error: {str(e)}"
 
-# Specific model wrappers
-def stabilityai_generate(prompt, width=1024, height=1024, seed=0):
-    """Stable Diffusion 2.1 - Most reliable free model"""
-    return huggingface_generate(
-        "stabilityai/stable-diffusion-2-1", 
-        prompt, width, height, seed
-    )
+def stabilityai_generate(prompt, width, height, seed):
+    return huggingface_generate("stabilityai/stable-diffusion-2-1", prompt, width, height, seed)
 
-def flux_schnell_generate(prompt, width=1024, height=1024, seed=0):
-    """FLUX.1-schnell - Fast but might be rate limited"""
-    return huggingface_generate(
-        "black-forest-labs/FLUX.1-schnell", 
-        prompt, width, height, seed
-    )
+def flux_schnell_generate(prompt, width, height, seed):
+    return huggingface_generate("black-forest-labs/FLUX.1-schnell", prompt, width, height, seed)
 
-def flux_dev_generate(prompt, width=1024, height=1024, seed=0):
-    """FLUX.1-dev - High quality but slower"""
-    return huggingface_generate(
-        "black-forest-labs/FLUX.1-dev", 
-        prompt, width, height, seed
-    )
+def sd_xl_generate(prompt, width, height, seed):
+    return huggingface_generate("stabilityai/stable-diffusion-xl-base-1.0", prompt, width, height, seed)
 
-def sd_xl_generate(prompt, width=1024, height=1024, seed=0):
-    """Stable Diffusion XL Base 1.0"""
-    return huggingface_generate(
-        "stabilityai/stable-diffusion-xl-base-1.0", 
-        prompt, width, height, seed
-    )
-
-def pollinations_generate(prompt, width=1024, height=1024, seed=0):
-    """Keep as fallback"""
+def pollinations_generate(prompt, width, height, seed):
     try:
         encoded_prompt = urllib.parse.quote(prompt)
         url = f"https://image.pollinations.ai/prompt/{encoded_prompt}?width={width}&height={height}&nologo=true&seed={seed if seed > 0 else int(time.time())}"
         response = requests.get(url, timeout=30)
         if response.status_code == 200:
-            return Image.open(BytesIO(response.content)), "Pollinations.ai"
-        return None, f"Pollinations: HTTP {response.status_code}"
+            return Image.open(BytesIO(response.content)), "Pollinations"
+        return None, f"Pollinations HTTP {response.status_code}"
     except Exception as e:
-        return None, f"Pollinations Failed: {str(e)}"
-
-# ==================== MAIN GENERATION LOGIC ====================
+        return None, f"Pollinations: {str(e)}"
 
 def generate_with_fallback(prompt, width=1024, height=1024, seed=0, primary_model="Stable Diffusion 2.1"):
-    """
-    Try Hugging Face first (free, reliable), then Pollinations as backup
-    """
     if not prompt:
-        return None, "⚠️ Please enter a prompt!"
-    
-    # Map friendly names to functions
+        return None, "⚠️ Enter a prompt!"
     models = {
         "Stable Diffusion 2.1": stabilityai_generate,
         "FLUX.1-schnell (Fast)": flux_schnell_generate,
-        "FLUX.1-dev (Quality)": flux_dev_generate,
         "Stable Diffusion XL": sd_xl_generate,
     }
-    
     errors = []
-    
-    # Try primary model first
-    if primary_model in models:
-        func = models[primary_model]
+    for name, func in list(models.items()) + [("Pollinations", pollinations_generate)]:
         try:
-            print(f"🔄 Trying {primary_model}...")
-            image, status = func(prompt, width, height, seed)
-            if image is not None:
-                return image, f"✅ Success! ({status})"
-            else:
-                errors.append(f"{primary_model}: {status}")
+            img, status = func(prompt, width, height, seed)
+            if img: 
+                return img, f"✅ {status}"
+            errors.append(f"{name}: {status}")
         except Exception as e:
-            errors.append(f"{primary_model}: {str(e)}")
-    
-    # Try other models as fallback
-    for model_name, func in models.items():
-        if model_name == primary_model:
-            continue
-        try:
-            print(f"🔄 Trying fallback: {model_name}...")
-            image, status = func(prompt, width, height, seed)
-            if image is not None:
-                return image, f"✅ Success! ({status} - Fallback)"
-            else:
-                errors.append(f"{model_name}: {status}")
-        except Exception as e:
-            errors.append(f"{model_name}: {str(e)}")
-    
-    # Last resort: Pollinations
-    try:
-        print("🔄 Trying Pollinations as last resort...")
-        image, status = pollinations_generate(prompt, width, height, seed)
-        if image is not None:
-            return image, f"✅ Success! ({status} - Last Resort)"
-        else:
-            errors.append(f"Pollinations: {status}")
-    except Exception as e:
-        errors.append(f"Pollinations: {str(e)}")
-    
-    # All failed
-    error_msg = "❌ All services failed:\n" + "\n".join(errors)
-    error_msg += "\n\n💡 Tips:\n- Hugging Face free tier has rate limits (wait 1-2 minutes)\n- Try a different model\n- Check your internet connection"
-    return None, error_msg
+            errors.append(f"{name}: {str(e)}")
+    return None, "❌ Failed:\\n" + "\\n".join(errors)
 
-# ==================== GRADIO UI ====================
+# ==================== GRADIO UI (FIXED) ====================
 
-with gr.Blocks(title="Free Image Generator - HF Edition", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-    # 🎨 Free Image Generator (Hugging Face Edition)
-    **Zero Cost • No API Keys • Powered by Stable Diffusion & FLUX**
-    
-    Made with ❤️ by Srijan Gajurel | Uses Hugging Face Free Inference API
-    """)
+with gr.Blocks(title="Free Image Generator") as demo:  # Removed theme from here
+    gr.Markdown("# 🎨 Free Image Generator (HF Edition)")
+    gr.Markdown("**No API Keys • Multi-Provider Fallback** | By Srijan Gajurel")
     
     with gr.Row():
-        with gr.Column(scale=1):
-            prompt = gr.Textbox(
-                label="📝 Prompt",
-                lines=3,
-                placeholder="Describe your image in detail...",
-                value="A beautiful sunset over mountains, cinematic lighting, 8k quality"
-            )
-            
+        with gr.Column():
+            prompt = gr.Textbox(label="📝 Prompt", lines=3, value="Sunset over mountains, cinematic")
             with gr.Row():
-                width = gr.Slider(
-                    minimum=512, maximum=1024, value=1024, step=64, 
-                    label="📐 Width"
-                )
-                height = gr.Slider(
-                    minimum=512, maximum=1024, value=1024, step=64, 
-                    label="📏 Height"
-                )
-            
-            seed = gr.Number(
-                value=0, label="🎲 Seed (0 for random)", precision=0
-            )
-            
-            model_selector = gr.Radio(
-                choices=[
-                    "Stable Diffusion 2.1",
-                    "FLUX.1-schnell (Fast)", 
-                    "FLUX.1-dev (Quality)",
-                    "Stable Diffusion XL"
-                ],
-                value="Stable Diffusion 2.1",
-                label="🤖 AI Model (Auto-fallback enabled)"
-            )
-            
-            with gr.Accordion("ℹ️ About Free Limits", open=False):
-                gr.Markdown("""
-                **Hugging Face Free Tier:**
-                - ✅ No API key required for public models
-                - ⚠️ Rate limited (~1-2 requests/minute per model)
-                - ⏳ Models may "sleep" after inactivity (first request wakes them up)
-                - 🔄 Auto-fallback tries all models if one fails
-                
-                **Tips:**
-                - If you get "Model is loading", wait 30 seconds and retry
-                - Stable Diffusion 2.1 is the most reliable
-                - FLUX models are higher quality but may be busier
-                """)
-            
-            generate_btn = gr.Button("🚀 Generate Image", variant="primary", size="lg")
-            
-            status_text = gr.Textbox(
-                label="📊 Status",
-                interactive=False,
-                value="Ready - Select a model and click Generate"
-            )
+                width = gr.Slider(512, 1024, 1024, step=64, label="Width")
+                height = gr.Slider(512, 1024, 1024, step=64, label="Height")
+            seed = gr.Number(0, label="Seed (0=random)", precision=0)
+            model = gr.Radio(["Stable Diffusion 2.1", "FLUX.1-schnell (Fast)", "Stable Diffusion XL"], 
+                          value="Stable Diffusion 2.1", label="Model")
+            btn = gr.Button("🚀 Generate", variant="primary")
+            status = gr.Textbox(label="Status", interactive=False, value="Ready")
         
-        with gr.Column(scale=1):
-            output_image = gr.Image(
-                label="🖼️ Generated Image",
-                type="pil",
-                height=600,
-                show_download_button=True
-            )
-    
-    gr.Markdown("### 🎨 Example Prompts")
-    examples = gr.Examples(
-        examples=[
-            ["A futuristic cyberpunk city at night, neon lights, rain, cinematic", 1024, 1024, 42, "Stable Diffusion 2.1"],
-            ["Portrait of a wise old owl wearing glasses, oil painting style, detailed", 1024, 1024, 0, "FLUX.1-schnell (Fast)"],
-            ["Magical forest with glowing mushrooms and fireflies, fantasy art", 1024, 1024, 123, "Stable Diffusion XL"],
-            ["Cute robot baking cookies in a cozy kitchen, 3d render, pixar style", 1024, 1024, 0, "FLUX.1-dev (Quality)"],
-        ],
-        inputs=[prompt, width, height, seed, model_selector]
-    )
+        with gr.Column():
+            # REMOVED show_download_button parameter
+            img = gr.Image(label="Generated Image", type="pil", height=600)
     
-    # Event handler
-    generate_btn.click(
-        fn=generate_with_fallback,
-        inputs=[prompt, width, height, seed, model_selector],
-        outputs=[output_image, status_text]
-    )
+    btn.click(generate_with_fallback, [prompt, width, height, seed, model], [img, status])
 
-if __name__ == "__main__":
-    print("🚀 Starting Free Image Generator...")
-    print("🤖 Models available: Stable Diffusion 2.1, SDXL, FLUX.1-schnell, FLUX.1-dev")
-    print("💡 No API keys required!")
-    demo.launch(share=False)
+# Moved theme to launch() - fixes Gradio 6.0 warning
+demo.launch(theme=gr.themes.Soft())