Update app.py

app.py

@@ -113,10 +113,11 @@ def handle_post_success_actions(generated_function):
             return generated_function
 
 def main(initial_prompt, run_mode, num_runs, console_output, command_input):
-    console_output = ""
+    console_output = "Enter the initial prompt for the development process: " + initial_prompt + "\n"
     
     while True:
         console_output += "\nMenu:\n1. Generate and test a function 🎨\n2. Exit 👋\n"
+        yield console_output
         choice = command_input
         command_input = ""
         
@@ -159,6 +160,9 @@ def main(initial_prompt, run_mode, num_runs, console_output, command_input):
     
     yield console_output
 
+    
+    yield console_output
+
 with gr.Blocks() as demo:
     gr.Markdown("# LLM-Powered Function Generator")