Upload chat_main.py

chatbot/chat_main.py

@@ -2963,6 +2963,9 @@ Please provide a helpful response:
             except subprocess.TimeoutExpired:
                 mcp_process.kill()
         
+        # CRITICAL: Return session state after successful completion
+        return current_session_url, agent_context
+        
     except Exception as e:
         output_messages.append(f"❌ Error: {str(e)}")
         yield "\n".join(output_messages)