Update app.py

app.py

@@ -10,7 +10,7 @@ from huggingface_hub import InferenceClient
 
 # Project by Nymbo
 
-def query_with_auto_routing(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
+def query_with_auto_routing(prompt, model, custom_lora, negative_prompt: str | None = None, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
     """
     Generate images using HF's automatic provider routing
     """
@@ -62,14 +62,39 @@ def query_with_auto_routing(prompt, model, custom_lora, is_negative=False, steps
             num_inference_steps=steps,
             guidance_scale=cfg_scale,
             seed=seed if seed != -1 else None,
+            negative_prompt=negative_prompt if negative_prompt else None,
         )
         
         print(f'Generation {key} completed with automatic routing!')
         return image
         
     except Exception as e:
-        print(f"Error with automatic routing: {e}")
-        raise gr.Error(f"Failed to generate image: {str(e)}")
+        # Sanitize payment/credit-related errors so users don't see sensitive details
+        err_text = str(e)
+        is_payment_err = False
+        try:
+            # Try to detect HTTP status code 402 if available (e.g., httpx.HTTPStatusError)
+            status = getattr(getattr(e, "response", None), "status_code", None)
+            if status == 402:
+                is_payment_err = True
+        except Exception:
+            pass
+        # Fallback string heuristics
+        payment_markers = (
+            "402",
+            "Payment Required",
+            "exceeded your monthly included credits",
+            "credits for Inference Providers",
+        )
+        if any(m.lower() in err_text.lower() for m in payment_markers):
+            is_payment_err = True
+
+        if is_payment_err:
+            print("Generation failed due to billing/credit limits (details hidden from user).")
+            raise gr.Error("Generation temporarily unavailable. Please try again later.")
+        else:
+            print(f"Error with automatic routing: {err_text}")
+            raise gr.Error("Failed to generate image. Please try again.")
 
 def get_model_id_from_name(model_name):
     """
@@ -331,11 +356,11 @@ def apply_model_prompt_enhancements(model_name, prompt):
     
     return prompt
 
-def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
+def query(prompt, model, custom_lora, negative_prompt=None, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
     """
     Main query function - now uses automatic provider routing
     """
-    return query_with_auto_routing(prompt, model, custom_lora, is_negative, steps, cfg_scale, sampler, seed, strength, width, height)
+    return query_with_auto_routing(prompt, model, custom_lora, negative_prompt, steps, cfg_scale, sampler, seed, strength, width, height)
 
 # Custom CSS to hide the footer in the interface
 css = """