Update app.py

app.py

@@ -2,8 +2,7 @@ import gradio as gr
 import time
 
 # Function to demonstrate a for loop
-def demonstrate_for_loop(n, progress_bar):
-    progress = 0
+def demonstrate_for_loop(n, progress):
     for_output = ""
     for i in range(1, n + 1):
         if i % 2 == 0:
@@ -11,13 +10,12 @@ def demonstrate_for_loop(n, progress_bar):
         else:
             color = "orange"
         for_output += f'<span style="color:{color}; font-size:24px; font-weight:bold;">{i} </span>'
-        progress = i / n
-        progress_bar.update(progress)
-        yield (for_output)
+        progress(i / n)  # Update progress bar
+        time.sleep(0.5)  # Simulate delay
+    return for_output  # Return final output
 
 # Function to demonstrate a while loop
-def demonstrate_while_loop(n, progress_bar):
-    progress = 0
+def demonstrate_while_loop(n, progress):
     while_output = ""
     i = 1
     while i <= n:
@@ -26,21 +24,17 @@ def demonstrate_while_loop(n, progress_bar):
         else:
             color = "orange"
         while_output += f'<span style="color:{color}; font-size:24px; font-weight:bold;">{i} </span>'
-        progress = i / n
-        progress_bar.update(progress)
-        yield (while_output)
+        progress(i / n)  # Update progress bar
+        time.sleep(0.5)  # Simulate delay
         i += 1
+    return while_output  # Return final output
 
 # Gradio Interface
 def run_loop(n, loop_type):
     if loop_type == "For Loop":
-        generator = demonstrate_for_loop(n, progress_bar)
+        return demonstrate_for_loop(n, gr.Progress())  # Pass progress callback
     else:
-        generator = demonstrate_while_loop(n, progress_bar)
-    
-    for output in generator:
-        result_box.update(output)
-        time.sleep(0.5)
+        return demonstrate_while_loop(n, gr.Progress())  # Pass progress callback
 
 with gr.Blocks() as demo:
     gr.Markdown("# Loop Demonstrator App")
@@ -49,9 +43,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()  # Do not add progress_bar as an output
     
-    gr.Button("Run Loop").click(run_loop, inputs=[n_input, loop_type_input])
+    # Update the button to directly return the result
+    gr.Button("Run Loop").click(run_loop, inputs=[n_input, loop_type_input], outputs=result_box)
 
 # Launch the Gradio app
 demo.launch()