Update app.py

app.py

@@ -198,7 +198,6 @@ with gr.Blocks() as demo:
         show_progress="full",
         queue=False,
         concurrency_limit=None,
-        fn_kwargs={"is_enhance": True} # Pass the flag to indicate enhance
     )
 
     generateBtn.click(
@@ -216,6 +215,10 @@ with gr.Blocks() as demo:
             generateBtn: gr.update(visible=not realtime_enabled)
         }
 
+    def realtime_generation(*args):
+    if args[0]:  # If realtime is enabled
+        return next(generate_image(*args[1:]))
+
     realtime.change(
         fn=update_ui,
         inputs=[realtime],
@@ -238,7 +241,7 @@ with gr.Blocks() as demo:
 
     for component in [prompt, width, height, num_inference_steps]:
         component.input(
-            fn=handle_realtime_update, # Call the wrapper that checks the toggle
+            fn=realtime_generation,
             inputs=[realtime, prompt, seed, width, height, randomize_seed, num_inference_steps],
             outputs=[result, seed, latency],
             show_progress="hidden",
@@ -246,17 +249,6 @@ with gr.Blocks() as demo:
             queue=False,
             concurrency_limit=None
         )
-        
-    # Also trigger realtime on seed change if randomize is off
-    seed.input(
-        fn=handle_realtime_update,
-        inputs=[realtime, prompt, seed, width, height, randomize_seed, num_inference_steps],
-        outputs=[result, seed, latency],
-        show_progress="hidden",
-        trigger_mode="always_last",
-        queue=False,
-        concurrency_limit=None
-    )
 
 # Launch the app
 demo.launch()