Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import time
 
 # Function to demonstrate a for loop
-def demonstrate_for_loop(n):
+def demonstrate_for_loop(n, progress_bar):
     progress = 0
     for_output = ""
     for i in range(1, n + 1):
@@ -12,10 +12,11 @@ def demonstrate_for_loop(n):
             color = "orange"
         for_output += f'<span style="color:{color}; font-size:24px; font-weight:bold;">{i} </span>'
         progress = i / n
-        yield (for_output, progress)
+        progress_bar.update(progress)
+        yield (for_output)
 
 # Function to demonstrate a while loop
-def demonstrate_while_loop(n):
+def demonstrate_while_loop(n, progress_bar):
     progress = 0
     while_output = ""
     i = 1
@@ -26,17 +27,21 @@ def demonstrate_while_loop(n):
             color = "orange"
         while_output += f'<span style="color:{color}; font-size:24px; font-weight:bold;">{i} </span>'
         progress = i / n
-        yield (while_output, progress)
+        progress_bar.update(progress)
+        yield (while_output)
         i += 1
 
-# Define the interface
-def loop_interface(n, loop_type):
+# Gradio Interface
+def run_loop(n, loop_type):
     if loop_type == "For Loop":
-        return demonstrate_for_loop(n)
+        generator = demonstrate_for_loop(n, progress_bar)
     else:
-        return demonstrate_while_loop(n)
+        generator = demonstrate_while_loop(n, progress_bar)
+    
+    for output in generator:
+        result_box.update(output)
+        time.sleep(0.5)
 
-# Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Loop Demonstrator App")
     
@@ -44,20 +49,9 @@ with gr.Blocks() as demo:
     loop_type_input = gr.Dropdown(choices=["For Loop", "While Loop"], label="Loop Type", value="For Loop")
     
     result_box = gr.HTML()
-    progress_bar = gr.Progress()
+    progress_bar = gr.Progress()  # Do not add progress_bar as an output
     
-    def run_loop(n, loop_type):
-        if loop_type == "For Loop":
-            generator = demonstrate_for_loop(n)
-        else:
-            generator = demonstrate_while_loop(n)
-        
-        for output, progress in generator:
-            result_box.update(output)
-            progress_bar.update(progress)
-            time.sleep(0.5)
-
-    gr.Button("Run Loop").click(run_loop, inputs=[n_input, loop_type_input], outputs=[result_box, progress_bar])
+    gr.Button("Run Loop").click(run_loop, inputs=[n_input, loop_type_input])
 
 # Launch the Gradio app
 demo.launch()