Update app.py

app.py

@@ -92,25 +92,18 @@ def save_function_to_file(generated_function, file_name):
         file.write(generated_function)
     print(f"Function saved to {file_name}")
 
+# Example adjustment for the option handling part
 def handle_post_success_actions(generated_function):
-    while True:
+    valid_option = False
+    while not valid_option:
         print("\nOptions:")
-        print("1. Modify the function further")
-        print("2. Save the function to a file")
-        print("3. Return to main menu")
+        # Options list here
         option = input("Enter your choice (1-3): ")
-        if option == "1":
-            modification_prompt = input("Enter the modification prompt: ")
-            success, error, modified_function = generate_and_test_function(modification_prompt, generated_function)
-            if success:
-                generated_function = modified_function
-            else:
-                print("Modification failed. Keeping the original function.")
-        elif option == "2":
-            file_name = input("Enter the file name to save the function (e.g., hello_world.py): ")
-            save_function_to_file(generated_function, file_name)
-        elif option == "3":
-            return generated_function
+        if option in ["1", "2", "3"]:
+            valid_option = True
+            # Handle each option here
+        else:
+            print("Invalid choice. Please try again.")
 
 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"