Update app.py

app.py

@@ -159,46 +159,61 @@ def generate_negative_prompt(ad_data):
     
     return ", ".join(base_negative)
 
-def call_huggingface_api(prompt, negative_prompt, api_token):
-    """Call Hugging Face Inference API for FLUX.1-schnell"""
-    API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
-    
+def call_huggingface_api(prompt, negative_prompt, api_token, width=1024, height=1024):
+    """Call Hugging Face Inference API (new Router endpoint) for FLUX.1-schnell"""
+
+    API_URL = "https://router.huggingface.co/hf-inference/models/black-forest-labs/FLUX.1-schnell"
+
     headers = {
         "Authorization": f"Bearer {api_token}",
         "Content-Type": "application/json"
     }
-    
+
     payload = {
         "inputs": prompt,
         "parameters": {
             "negative_prompt": negative_prompt,
-            "num_inference_steps": 4,  # Schnell optimized for 1-4 steps
-            "guidance_scale": 0.0,  # Schnell doesn't use guidance
-            "width": 1024,
-            "height": 1024
+            "num_inference_steps": 4,
+            "guidance_scale": 0.0,
+            "width": width,
+            "height": height
         }
     }
-    
+
     try:
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
-        
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=120)
+
+        # Handle image bytes
         if response.status_code == 200:
-            image = Image.open(io.BytesIO(response.content))
-            return image, None
+            try:
+                image = Image.open(io.BytesIO(response.content))
+                return image, None
+            except Exception:
+                return None, "Received invalid image data."
+
         elif response.status_code == 503:
-            return None, "Model is loading. Please wait 20 seconds and try again."
+            return None, "Model is loading (503). Wait 20–40 sec and retry."
+
         elif response.status_code == 401:
-            return None, "Invalid API token. Please check your Hugging Face token."
+            return None, "Invalid or missing HF token."
+
         elif response.status_code == 402:
-            return None, "You've exceeded your free tier limit. Consider upgrading to PRO or wait for next month."
+            return None, "Free-tier quota exceeded."
+
         else:
-            error_msg = response.json().get('error', 'Unknown error')
+            try:
+                error_msg = response.json().get("error", "Unknown error")
+            except:
+                error_msg = response.text
+
             return None, f"API Error ({response.status_code}): {error_msg}"
-    
+
     except requests.Timeout:
-        return None, "Request timed out. The model might be loading. Please try again."
+        return None, "Request timed out. HF router is busy. Try again."
+
     except Exception as e:
-        return None, f"Error: {str(e)}"
+        return None, f"Unexpected error: {str(e)}"
+
 
 def image_to_base64(image):
     """Convert PIL Image to base64 string"""
@@ -207,8 +222,8 @@ def image_to_base64(image):
     return base64.b64encode(buffered.getvalue()).decode()
 
 # Header
-st.markdown('<h1 class="main-header">🎨 AdForge AI</h1>', unsafe_allow_html=True)
-st.markdown('<p class="sub-header">Transform your concept into stunning advertisements with AI</p>', unsafe_allow_html=True)
+st.markdown('<h1 class="main-header">🎨 Addmaker</h1>', unsafe_allow_html=True)
+st.markdown('<p class="sub-header">Transform your concept into stunning advertisements </p>', unsafe_allow_html=True)
 
 # Sidebar Configuration
 with st.sidebar: