Update app.py

app.py

@@ -33,7 +33,7 @@ def generate_image(prompt, seed=42, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT,
     # Only generate the last image in the sequence
     for img in pipe.generate_images(  
             prompt=prompt,
-            guidance_scale=0,
+            guidance_scale=0, # as Flux schnell is guidance free
             num_inference_steps=num_inference_steps,
             width=width,
             height=height,
@@ -63,7 +63,7 @@ with gr.Blocks() as demo:
             with gr.Column(scale=3):
                 result = gr.Image(label="Generated Image", show_label=False, interactive=False)
             with gr.Column(scale=1):
-                prompt = gr.Text(
+                prompt = gr.Textbox(
                     label="Prompt",
                     placeholder="Describe the image you want to generate...",
                     lines=3,
@@ -96,8 +96,12 @@ with gr.Blocks() as demo:
                 cache_examples="lazy" 
             )
 
+    def enhance_image(*args):
+        gr.Info("Enhancing Image") # currently just runs optimized pipeline for 2 steps. Further implementations later.
+        return next(generate_image(*args))
+
     enhanceBtn.click(
-        fn=generate_image,
+        fn=enhance_image,
         inputs=[prompt, seed, width, height],
         outputs=[result, seed, latency],
         show_progress="hidden",
@@ -135,10 +139,10 @@ with gr.Blocks() as demo:
             return next(generate_image(*args[1:]))
 
     prompt.submit(
-        fn=realtime_generation,
-        inputs=[realtime, prompt, seed, width, height, randomize_seed, num_inference_steps],
+        fn=generate_image,
+        inputs=[prompt, seed, width, height, randomize_seed, num_inference_steps],
         outputs=[result, seed, latency],
-        show_progress="hidden",
+        show_progress="full",
         api_name=False,
         queue=False,
         concurrency_limit=None