Update app.py

app.py

@@ -117,9 +117,8 @@ def main(initial_prompt, run_mode, num_runs, console_output, command_input):
     
     while True:
         console_output += "\nMenu:\n1. Generate and test a function 🎨\n2. Exit 👋\n"
-        yield console_output
-        choice = command_input
-        command_input = ""
+        yield console_output, gr.update(value="")  # Clear the command input
+        choice = yield console_output, gr.update(interactive=True)  # Wait for user input
         
         if choice == "1":
             if run_mode == "1":
@@ -129,18 +128,18 @@ def main(initial_prompt, run_mode, num_runs, console_output, command_input):
                     initial_prompt = f"Continue developing the function:\n{generated_function}"
                 else:
                     console_output += "Function test failed. 😞\n"
-                    yield console_output
+                    yield console_output, gr.update(interactive=True)
             elif run_mode == "2":
                 for i in range(int(num_runs)):
                     console_output += f"\nRun {i+1}:\n"
-                    yield console_output
+                    yield console_output, gr.update(interactive=True)
                     success, error, generated_function = generate_and_test_function(initial_prompt)
                     if success:
                         generated_function = handle_post_success_actions(generated_function)
                         initial_prompt = f"Continue developing the function:\n{generated_function}"
                     else:
                         console_output += "Function test failed. 😞\n"
-                        yield console_output
+                        yield console_output, gr.update(interactive=True)
             elif run_mode == "3":
                 while True:
                     success, error, generated_function = generate_and_test_function(initial_prompt)
@@ -149,19 +148,14 @@ def main(initial_prompt, run_mode, num_runs, console_output, command_input):
                         initial_prompt = f"Continue developing the function:\n{generated_function}"
                     else:
                         console_output += "Function test failed. Retrying...\n"
-                        yield console_output
+                        yield console_output, gr.update(interactive=True)
         elif choice == "2":
             console_output += "Exiting... Goodbye! 👋\n"
-            yield console_output
+            yield console_output, gr.update(interactive=False)
             break
         else:
             console_output += "Invalid choice. Please try again. 😅\n"
-            yield console_output
-    
-    yield console_output
-
-    
-    yield console_output
+            yield console_output, gr.update(interactive=True)
 
 with gr.Blocks() as demo:
     gr.Markdown("# LLM-Powered Function Generator")