Update app.py

app.py

@@ -10,13 +10,14 @@ load_dotenv()
 
 # Get API key from environment variable
 # Set your API key in Hugging Face Space Settings > Variables and Secrets
-# Name it: API_KEY
 API_KEY = os.getenv("API_KEY")
-API_URL = os.getenv("API_URL", "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0")
+
+# NEW CORRECT ENDPOINT (updated from api-inference.huggingface.co)
+API_URL = os.getenv("API_URL", "https://router.huggingface.co/hf-inference/models/black-forest-labs/FLUX.1-schnell")
 
 def generate_image(prompt, negative_prompt="", width=1024, height=1024, guidance_scale=7.5, num_inference_steps=50):
     """
-    Generate image using Hugging Face Inference API or similar API
+    Generate image using Hugging Face Inference API (Router endpoint)
     """
     if not API_KEY:
         return None, "Error: API key not configured. Please set the API_KEY in your Space secrets."
@@ -29,10 +30,10 @@ def generate_image(prompt, negative_prompt="", width=1024, height=1024, guidance
         "Content-Type": "application/json"
     }
     
+    # FLUX.1-schnell uses different parameters - adjust payload accordingly
     payload = {
         "inputs": prompt,
         "parameters": {
-            "negative_prompt": negative_prompt,
             "width": width,
             "height": height,
             "guidance_scale": guidance_scale,
@@ -40,6 +41,10 @@ def generate_image(prompt, negative_prompt="", width=1024, height=1024, guidance
         }
     }
     
+    # Only add negative_prompt if provided and model supports it
+    if negative_prompt:
+        payload["parameters"]["negative_prompt"] = negative_prompt
+    
     try:
         response = requests.post(API_URL, headers=headers, json=payload, timeout=300)
         
@@ -79,14 +84,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             with gr.Row():
                 width_slider = gr.Slider(
                     minimum=512,
-                    maximum=2048,
+                    maximum=1024,
                     step=64,
                     value=1024,
                     label="Width"
                 )
                 height_slider = gr.Slider(
                     minimum=512,
-                    maximum=2048,
+                    maximum=1024,
                     step=64,
                     value=1024,
                     label="Height"
@@ -101,11 +106,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             )
             
             steps_slider = gr.Slider(
-                minimum=20,
-                maximum=100,
+                minimum=1,
+                maximum=50,
                 step=1,
-                value=50,
-                label="Inference Steps"
+                value=4,
+                label="Inference Steps (FLUX works well with 4-20)"
             )
             
             generate_btn = gr.Button("✨ Generate Image", variant="primary")
@@ -119,6 +124,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             - Be specific and descriptive in your prompts
             - Include style keywords: "digital art", "photorealistic", "oil painting"
             - Mention lighting: "cinematic lighting", "golden hour", "studio lighting"
+            - FLUX models work great with just 4-8 inference steps!
             """)
     
     # Event handlers
@@ -138,9 +144,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Examples
     gr.Examples(
         examples=[
-            ["A futuristic city at night with neon lights and flying cars, digital art", "", 1024, 1024, 7.5, 50],
-            ["Portrait of a wise wizard with long beard, magical atmosphere, oil painting", "blurry, ugly", 1024, 1024, 8, 50],
-            ["Cute robot reading a book in a cozy library, pixar style", "", 1024, 1024, 7, 50],
+            ["A futuristic city at night with neon lights and flying cars, digital art", "", 1024, 1024, 7.5, 4],
+            ["Portrait of a wise wizard with long beard, magical atmosphere, oil painting", "blurry, ugly", 1024, 1024, 8, 4],
+            ["Cute robot reading a book in a cozy library, pixar style", "", 1024, 1024, 7, 4],
         ],
         inputs=[prompt_input, negative_prompt_input, width_slider, height_slider, guidance_slider, steps_slider],
         label="Example Prompts"