Update app.py

app.py

@@ -114,11 +114,14 @@ def handle_post_success_actions(generated_function):
 
 def main(initial_prompt, run_mode, num_runs, console_output, command_input):
     console_output = "Enter the initial prompt for the development process: " + initial_prompt + "\n"
+    yield console_output, gr.update(value="")  # Clear the command input
     
     while True:
         console_output += "\nMenu:\n1. Generate and test a function 🎨\n2. Exit 👋\n"
+        yield console_output, gr.update(interactive=True)  # Wait for user input
+        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":
@@ -157,6 +160,7 @@ def main(initial_prompt, run_mode, num_runs, console_output, command_input):
             console_output += "Invalid choice. Please try again. 😅\n"
             yield console_output, gr.update(interactive=True)
 
+
 with gr.Blocks() as demo:
     gr.Markdown("# LLM-Powered Function Generator")