From 6dba14cb363095dc6dba20ccebd2f9025f6e5b34 Mon Sep 17 00:00:00 2001
From: Pietro Schirano <pietro.schirano@gmail.com>
Date: Sun, 30 Jun 2024 16:45:59 -0400
Subject: [PATCH] Updated the script to fix a bug when force exiting the
 automode

---
 main.py | 46 ++++++++++++++++++++++++++++------------------
 1 file changed, 28 insertions(+), 18 deletions(-)

diff --git a/main.py b/main.py
index 1b7afab4..f6fd0a44 100644
--- a/main.py
+++ b/main.py
@@ -523,7 +523,7 @@ def chat_with_claude(user_input, image_path=None, current_iteration=None, max_it
     return assistant_response, exit_continuation
 
 def main():
-    global automode
+    global automode, conversation_history
     print_colored("Welcome to the Claude-3.5-Sonnet Engineer Chat with Image Support!", CLAUDE_COLOR)
     print_colored("Type 'exit' to end the conversation.", CLAUDE_COLOR)
     print_colored("Type 'image' to include an image in your message.", CLAUDE_COLOR)
@@ -561,26 +561,36 @@ def main():
                 user_input = input(f"\n{USER_COLOR}You: {Style.RESET_ALL}")
                 
                 iteration_count = 0
-                while automode and iteration_count < max_iterations:
-                    response, exit_continuation = chat_with_claude(user_input, current_iteration=iteration_count+1, max_iterations=max_iterations)
-                    process_and_display_response(response)
-                    
-                    if exit_continuation or CONTINUATION_EXIT_PHRASE in response:
-                        print_colored("Automode completed.", TOOL_COLOR)
-                        automode = False
-                    else:
-                        print_colored(f"Continuation iteration {iteration_count + 1} completed.", TOOL_COLOR)
-                        print_colored("Press Ctrl+C to exit automode.", TOOL_COLOR)
-                        user_input = "Continue with the next step."
-                    
-                    iteration_count += 1
-                    
-                    if iteration_count >= max_iterations:
-                        print_colored("Max iterations reached. Exiting automode.", TOOL_COLOR)
-                        automode = False
+                try:
+                    while automode and iteration_count < max_iterations:
+                        response, exit_continuation = chat_with_claude(user_input, current_iteration=iteration_count+1, max_iterations=max_iterations)
+                        process_and_display_response(response)
+                        
+                        if exit_continuation or CONTINUATION_EXIT_PHRASE in response:
+                            print_colored("Automode completed.", TOOL_COLOR)
+                            automode = False
+                        else:
+                            print_colored(f"Continuation iteration {iteration_count + 1} completed.", TOOL_COLOR)
+                            print_colored("Press Ctrl+C to exit automode.", TOOL_COLOR)
+                            user_input = "Continue with the next step."
+                        
+                        iteration_count += 1
+                        
+                        if iteration_count >= max_iterations:
+                            print_colored("Max iterations reached. Exiting automode.", TOOL_COLOR)
+                            automode = False
+                except KeyboardInterrupt:
+                    print_colored("\nAutomode interrupted by user. Exiting automode.", TOOL_COLOR)
+                    automode = False
+                    # Ensure the conversation history ends with an assistant message
+                    if conversation_history and conversation_history[-1]["role"] == "user":
+                        conversation_history.append({"role": "assistant", "content": "Automode interrupted. How can I assist you further?"})
             except KeyboardInterrupt:
                 print_colored("\nAutomode interrupted by user. Exiting automode.", TOOL_COLOR)
                 automode = False
+                # Ensure the conversation history ends with an assistant message
+                if conversation_history and conversation_history[-1]["role"] == "user":
+                    conversation_history.append({"role": "assistant", "content": "Automode interrupted. How can I assist you further?"})
             
             print_colored("Exited automode. Returning to regular chat.", TOOL_COLOR)
         else: