Update app.py

app.py

@@ -40,31 +40,34 @@ pipe.enable_vae_slicing()
 pipe.set_progress_bar_config(disable=True)
 
 # ===============================
-# ULTRA MODE (FIXED API CALL)
+# ULTRA MODE (Z-IMAGE-TURBO)
 # ===============================
-def call_ultra_api(prompt, steps, resolution, seed):
+def call_ultra_api(prompt, negative, steps, resolution, seed):
     try:
-        # Connect specifically to the Z-Image-Turbo Space
         client = Client("mrfakename/Z-Image-Turbo")
         
-        # Based on your documentation, the parameters are:
-        # prompt, height, width, num_inference_steps, seed, randomize_seed
+        # The API doesn't support a separate 'negative_prompt' argument, 
+        # so we append it to the main prompt for better adherence.
+        full_prompt = f"{prompt} . {negative}" if negative else prompt
+        
+        # Call the specific endpoint defined in your HTML
         result = client.predict(
-            prompt=prompt,
-            height=int(resolution),
+            prompt=full_prompt,
             width=int(resolution),
+            height=int(resolution),
             num_inference_steps=int(steps),
             seed=int(seed),
-            randomize_seed=False, 
-            api_name="/predict"
+            randomize_seed=False,  # We control the seed from Python
+            api_name="/generate_image"
         )
         
-        # result[0] is usually the image path in this specific Space
-        if isinstance(result, tuple) or isinstance(result, list):
+        # The API returns [image_filepath, seed_used]
+        if isinstance(result, (list, tuple)):
             return Image.open(result[0])
         return Image.open(result)
+        
     except Exception as e:
-        print(f"Detailed API Error: {e}")
+        print(f"Ultra Mode Error: {e}")
         return None
 
 # ===============================
@@ -72,12 +75,14 @@ def call_ultra_api(prompt, steps, resolution, seed):
 # ===============================
 def toggle_ultra(is_ultra):
     if is_ultra:
+        # Hide 768, extend steps to 20, show negative prompt
         return {
             negative_field: gr.update(visible=True),
             resolution: gr.update(choices=[512, 1024], value=512),
-            steps: gr.update(minimum=6, maximum=20, value=6)
+            steps: gr.update(minimum=1, maximum=20, value=9)
         }
     else:
+        # Normal mode settings
         return {
             negative_field: gr.update(visible=False),
             resolution: gr.update(choices=[512, 768, 1024], value=512),
@@ -91,18 +96,19 @@ def generate(prompt, user_neg, res, step_val, is_ultra):
     
     if is_ultra:
         yield (None, "🎨 Generating Image...", gr.update(interactive=False))
-        # Ultra mode uses the direct prompt + user negative (if API supports neg, 
-        # but Z-Turbo usually ignores neg prompt to stay 'Turbo')
-        # We pass the prompt combined with negative for best results if the API is simple
-        full_prompt = f"{prompt} | Negative: {user_neg}" if user_neg else prompt
         
-        image = call_ultra_api(full_prompt, step_val, size, seed)
+        # Use user negative if provided, otherwise default
+        final_neg = user_neg if user_neg.strip() else default_neg
+        
+        image = call_ultra_api(prompt, final_neg, step_val, size, seed)
         
         if image:
-            yield (image, f"✅ Done in Ultra Mode.", gr.update(interactive=True))
+            yield (image, f"✅ Done (Ultra Mode). Seed: {seed}", gr.update(interactive=True))
         else:
-            yield (None, "❌ API Error: The external GPU server is busy.", gr.update(interactive=True))
+            yield (None, "❌ API Busy or Error. Try again.", gr.update(interactive=True))
+            
     else:
+        # Normal CPU Mode
         yield (None, "🧠 Analysing Prompt", gr.update(interactive=False))
         enhanced = enhance_prompt(prompt)
         
@@ -139,13 +145,21 @@ custom_css = """
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="green"), css=custom_css) as demo:
     with gr.Column(elem_id="container"):
-        gr.Markdown("# 👾 Creeper AI - v4.0")
-        gr.Markdown("Generate images using Creeper AI, running on CPU using fast models.")
+        gr.Markdown("# 🎨 Creeper AI - v4.0")
+        gr.Markdown("Generate images using Creeper AI. Use **Ultra-Fast** for GPU speeds.")
         
         with gr.Row():
             with gr.Column(scale=1):
-                prompt_field = gr.Textbox(label="What do you want to see?", placeholder="e.g. A futuristic city", lines=3)
-                negative_field = gr.Textbox(label="Negative Prompt", value="blurry, low quality, distorted, watermark", visible=False)
+                prompt_field = gr.Textbox(
+                    label="What do you want to see?", 
+                    placeholder="e.g. A futuristic city", 
+                    lines=3
+                )
+                negative_field = gr.Textbox(
+                    label="Negative Prompt", 
+                    value="blurry, low quality, distorted", 
+                    visible=False
+                )
                 
                 with gr.Accordion("Settings ⚙️", open=True):
                     resolution = gr.Radio([512, 768, 1024], value=512, label="Resolution")
@@ -158,6 +172,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="green"), css=custom_css) as dem
                 output_img = gr.Image(label="Result", interactive=False)
                 status = gr.Markdown("🟢 Ready", elem_classes="status-box")
 
+    # Connect the UI logic
     ultra_check.change(
         toggle_ultra, 
         inputs=[ultra_check],