Update app.py

app.py

@@ -6,10 +6,9 @@ import time
 
 app = Flask(__name__)
 
-
-
-GLOBAL_WORKSPACE_ID = "wkspace_01JSP2RQYHY6SFQ95PS0X1RM7C"
-GLOBAL_BOT_ID = "e7d787aa-fbc0-40b3-ac7f-e283aae81932"
+# Global variables to store workspace and bot IDs
+GLOBAL_WORKSPACE_ID = None
+GLOBAL_BOT_ID = None
 
 # Cookie value used in requests (should be updated with a valid cookie)
 AUTH_COOKIE = "pscd=try.botpress.com; _hjSessionUser_2931810=eyJpZCI6ImQ2MGMzYjhkLTlkMjQtNTA0OS1hMzlmLWEzNmI0NzA0NzUxNCIsImNyZWF0ZWQiOjE3MzU3MTg0MDcwNTAsImV4aXN0aW5nIjp0cnVlfQ==; hubspotutk=75739411a4d011b2164c4f3d944ecb94; intercom-device-id-bjzkw2xf=afd0a36b-b229-44e3-828e-60483c80c10c; _hjSessionUser_3339867=eyJpZCI6IjU4ODlmMTY4LWRkNGEtNTJhZS1hZTUzLWZlYWQwM2ZmMTVjNyIsImNyZWF0ZWQiOjE3MzU3MTg1ODM4MDgsImV4aXN0aW5nIjp0cnVlfQ==; mp_1195923e954ce61d822842b5832047cd_mixpanel=%7B%22distinct_id%22%3A%20%22d403ad7b-ea73-4d29-b977-5fd95afd585c%22%2C%22%24device_id%22%3A%20%22d403ad7b-ea73-4d29-b977-5fd95afd585c%22%2C%22%24initial_referrer%22%3A%20%22https%3A%2F%2Fapp.botpress.cloud%2F%22%2C%22%24initial_referring_domain%22%3A%20%22app.botpress.cloud%22%2C%22__mps%22%3A%20%7B%7D%2C%22__mpso%22%3A%20%7B%22%24initial_referrer%22%3A%20%22https%3A%2F%2Fapp.botpress.cloud%2F%22%2C%22%24initial_referring_domain%22%3A%20%22app.botpress.cloud%22%7D%2C%22__mpus%22%3A%20%7B%7D%2C%22__mpa%22%3A%20%7B%7D%2C%22__mpu%22%3A%20%7B%7D%2C%22__mpr%22%3A%20%5B%5D%2C%22__mpap%22%3A%20%5B%5D%2C%22%24user_id%22%3A%20%22d403ad7b-ea73-4d29-b977-5fd95afd585c%22%7D; _gid=GA1.2.1287628494.1744029890; _hjSession_2931810=eyJpZCI6IjYxMzliNWFhLTZkZDMtNDQxNy1hODliLTJkMjI3ZjhjMmZkZCIsImMiOjE3NDQwMjk4OTk5NzksInMiOjAsInIiOjAsInNiIjowLCJzciI6MCwic2UiOjAsImZzIjowLCJzcCI6MX0=; __hstc=59821234.75739411a4d011b2164c4f3d944ecb94.1735718442141.1743857941488.1744029903713.64; __hssrc=1; csrf_token_bd9ac21c34b9f0915e733c3e5305d737d0722c1168be7376b889426b5ec2a298=bh2jkV1V4U780wKDgoj+CN2zFBwUjPdw4PdorGkX/jA=; ory_kratos_session=MTc0NDAyOTk1OHwxWng4S0ZiNVVLblNRWldzUU9WWVVTRlVwU053aFBTOGZOa3ZBenJBQzdRejIydkc0YTJXZFVsVllvU1diNk9lQlQ1NTJtT2dJbWhDeWRDeU5TVmViNjhNclBPRDBZeFdrY3k3emh5dmlscE8wNmFLcENLLWhhbGJvblMwbUNPaUF3Y29mQ0ZKQld0ajkxeVA3YzhNWDJfaFhwRnp2LUF4VGVxSV92Yk5zTHBQT2x3X2hPZGpPUlpDOHBMTWQyeGRqeUJVTjhPRldTS09XeVA0UUswSDFyRWZLOUNFTkZwdllMT2NMUkN1dlRHcVcxUU9nbEFMaV96RU1Ea1pVbWc5SjAzdU82RGpSMk1LdHY0VlYxZHE0UT09fI-_ev3SlBoRdguqdHUF842Ku0_AZ9rX7GxVN7rV9NHs; ajs_user_id=d403ad7b-ea73-4d29-b977-5fd95afd585c; ajs_anonymous_id=cda6139d-cb82-4906-bfac-adaea115b097; _ga=GA1.1.1726154447.1735718383; _ga_W6YT9YSNLH=GS1.2.1744029893.64.1.1744030016.0.0.0; _ga_CYSS87Q508=GS1.2.1744029892.64.1.1744030016.0.0.0; __hssc=59821234.3.1744029903713; intercom-session-bjzkw2xf=RDY2ZU9VdlNaTHFvNFlwdnNmdXU1azdvVllUREF1UmZtdTNzMlVCZVFUYVpBTVFNSzFyclkwME9ySnZOd21pYTNDbE1DSVk3dk9QVVJrTzJYTHdVMEVMTXhaOTRWZ3AxSVZuYktVTnpYUFE9LS1nczlsRzhzNHdiZ2FmT3B2ZWdQNjVnPT0=--676bd54698c0b37278daa9c30607c884dedaf325; _ga_PCC6TBWJY6=GS1.1.1744029889.84.1.1744030023.0.0.0; _ga_HKHSWES9V9=GS1.1.1744029889.84.1.1744030023.57.0.2075971943"
@@ -18,23 +17,28 @@ AUTH_COOKIE = "pscd=try.botpress.com; _hjSessionUser_2931810=eyJpZCI6ImQ2MGMzYjh
 # Helper functions for random bot/workspace names
 # -------------------------------------------------------------------
 def generate_random_name(length=5):
+    """Generate a random name for workspace or bot"""
     return ''.join(random.choices(string.ascii_letters, k=length))
 
 # -------------------------------------------------------------------
 # Functions to create/delete workspaces and bots
 # -------------------------------------------------------------------
 def create_workspace():
+    """Create a new workspace and return its ID"""
     ws_url = "https://api.botpress.cloud/v1/admin/workspaces"
     headers = {
         "User-Agent": "Mozilla/5.0",
         "Cookie": AUTH_COOKIE
     }
     payload = {"name": generate_random_name()}
+
     try:
         response = requests.post(ws_url, headers=headers, json=payload)
         if response.status_code == 200:
             response_json = response.json()
-            return response_json.get('id')
+            workspace_id = response_json.get('id')
+            print(f"Successfully created workspace: {workspace_id}")
+            return workspace_id
         else:
             print(f"Workspace creation failed with: {response.status_code}, {response.text}")
             return None
@@ -44,6 +48,11 @@ def create_workspace():
 
 
 def create_bot(workspace_id):
+    """Create a new bot in the specified workspace and return its ID"""
+    if not workspace_id:
+        print("Cannot create bot: No workspace ID provided")
+        return None
+
     bot_url = "https://api.botpress.cloud/v1/admin/bots"
     headers = {
         "User-Agent": "Mozilla/5.0",
@@ -52,6 +61,7 @@ def create_bot(workspace_id):
         "Content-Type": "application/json"
     }
     payload = {"name": generate_random_name()}
+
     try:
         response = requests.post(bot_url, headers=headers, json=payload)
         if response.status_code == 200:
@@ -59,10 +69,18 @@ def create_bot(workspace_id):
             bot_id = response_json.get("bot", {}).get("id")
             if not bot_id:
                 print("Bot ID not found in the response.")
+                return None
+
+            print(f"Successfully created bot: {bot_id} in workspace: {workspace_id}")
+
             # Install integration for the new bot
-            if bot_id:
-                install_bot_integration(bot_id, workspace_id)
-            return bot_id
+            integration_success = install_bot_integration(bot_id, workspace_id)
+            if integration_success:
+                print(f"Successfully installed integration for bot {bot_id}")
+                return bot_id
+            else:
+                print(f"Failed to install integration for bot {bot_id}")
+                return bot_id  # Still return the bot ID even if integration fails
         else:
             print(f"Bot creation failed with: {response.status_code}, {response.text}")
             return None
@@ -73,6 +91,10 @@ def create_bot(workspace_id):
 
 def install_bot_integration(bot_id, workspace_id):
     """Install required integration for the bot to function properly"""
+    if not bot_id or not workspace_id:
+        print("Cannot install integration: Missing bot ID or workspace ID")
+        return False
+
     url = f"https://api.botpress.cloud/v1/admin/bots/{bot_id}"
     headers = {
         "User-Agent": "Mozilla/5.0",
@@ -89,6 +111,7 @@ def install_bot_integration(bot_id, workspace_id):
             }
         }
     }
+
     try:
         response = requests.put(url, headers=headers, json=payload)
         if response.status_code == 200:
@@ -103,30 +126,54 @@ def install_bot_integration(bot_id, workspace_id):
 
 
 def delete_bot(bot_id, workspace_id):
+    """Delete a bot from the specified workspace"""
+    if not bot_id or not workspace_id:
+        print("Cannot delete bot: Missing bot ID or workspace ID")
+        return False
+
     url = f"https://api.botpress.cloud/v1/admin/bots/{bot_id}"
     headers = {
         "User-Agent": "Mozilla/5.0",
         "x-workspace-id": workspace_id,
         "Cookie": AUTH_COOKIE
     }
+
     try:
-        return requests.delete(url, headers=headers)
+        response = requests.delete(url, headers=headers)
+        if response.status_code in [200, 204]:
+            print(f"Successfully deleted bot: {bot_id}")
+            return True
+        else:
+            print(f"Failed to delete bot: {response.status_code}, {response.text}")
+            return False
     except Exception as e:
         print(f"Error deleting bot: {str(e)}")
-        return None
+        return False
 
 
 def delete_workspace(workspace_id):
+    """Delete a workspace"""
+    if not workspace_id:
+        print("Cannot delete workspace: No workspace ID provided")
+        return False
+
     url = f"https://api.botpress.cloud/v1/admin/workspaces/{workspace_id}"
     headers = {
         "User-Agent": "Mozilla/5.0",
         "Cookie": AUTH_COOKIE
     }
+
     try:
-        return requests.delete(url, headers=headers)
+        response = requests.delete(url, headers=headers)
+        if response.status_code in [200, 204]:
+            print(f"Successfully deleted workspace: {workspace_id}")
+            return True
+        else:
+            print(f"Failed to delete workspace: {response.status_code}, {response.text}")
+            return False
     except Exception as e:
         print(f"Error deleting workspace: {str(e)}")
-        return None
+        return False
 
 
 # -------------------------------------------------------------------
@@ -145,7 +192,7 @@ def chat_with_assistant(user_input, chat_history, bot_id, workspace_id):
         "Cookie": AUTH_COOKIE
     }
 
-    # Process chat history into the format expected by the new API
+    # Process chat history into the format expected by the API
     messages = []
     system_prompt = ""
     for msg in chat_history:
@@ -166,7 +213,7 @@ def chat_with_assistant(user_input, chat_history, bot_id, workspace_id):
             "type": "text"
         })
 
-    # Prepare the payload for the new API
+    # Prepare the payload for the API
     payload = {
         "type": "openai:generateContent",
         "input": {
@@ -182,54 +229,100 @@ def chat_with_assistant(user_input, chat_history, bot_id, workspace_id):
     }
 
     botpress_url = "https://api.botpress.cloud/v1/chat/actions"
-    max_retries = 5
-    timeout = 60  # seconds
+    max_retries = 3
+    timeout = 120  # Increased timeout for long messages
+
+    # Flag to track if we need to create new IDs
+    need_new_ids = False
 
     # Attempt to send the request
     for attempt in range(max_retries):
         try:
+            print(f"Attempt {attempt+1}: Sending request to Botpress API with bot_id={bot_id}, workspace_id={workspace_id}")
             response = requests.post(botpress_url, json=payload, headers=headers, timeout=timeout)
 
             # If successful (200)
             if response.status_code == 200:
                 data = response.json()
                 assistant_content = data.get('output', {}).get('choices', [{}])[0].get('content', '')
+                print(f"Successfully received response from Botpress API")
                 return assistant_content, bot_id, workspace_id
 
-            # Handle 403 error specifically
+            # Handle 403 error (authentication/authorization issue)
             elif response.status_code == 403:
-                # Step 1: Delete the current bot and workspace
-                if bot_id and workspace_id:
-                    delete_bot(bot_id, workspace_id)
-                    delete_workspace(workspace_id)
-
-                # Step 2: Create a new workspace and bot
-                new_workspace_id = create_workspace()
-                if new_workspace_id:
-                    new_bot_id = create_bot(new_workspace_id)
-
-                    # Step 3: Retry with the new IDs
-                    headers["x-bot-id"] = new_bot_id
-                    retry_response = requests.post(botpress_url, json=payload, headers=headers, timeout=timeout)
-                    if retry_response.status_code == 200:
-                        data = retry_response.json()
-                        assistant_content = data.get('output', {}).get('choices', [{}])[0].get('content', '')
-                        return assistant_content, new_bot_id, new_workspace_id
-                    else:
-                        return f"Unable to get a response from the assistant.", new_bot_id, new_workspace_id
+                print(f"Received 403 error. Need to create new IDs.")
+                need_new_ids = True
+                break
+
+            # Handle network errors or timeouts (just retry)
+            elif response.status_code in [443, 408, 502, 503, 504]:
+                print(f"Received error {response.status_code}. Retrying...")
+                time.sleep(3)  # Wait before retrying
+                continue
 
             # Any other error status code
             else:
-                raise Exception("Unexpected error occurred.")
+                print(f"Received unexpected error: {response.status_code}, {response.text}")
+                if attempt < max_retries - 1:
+                    time.sleep(2)
+                    continue
+                else:
+                    return f"Unable to get a response from the assistant (Error {response.status_code}).", bot_id, workspace_id
+
+        except requests.exceptions.Timeout:
+            print(f"Request timed out. Retrying...")
+            if attempt < max_retries - 1:
+                time.sleep(2)
+                continue
+            else:
+                return "The assistant is taking too long to respond. Please try again with a shorter message.", bot_id, workspace_id
 
         except Exception as e:
-            # If this is not the last attempt, retry the same request
+            print(f"Error during request: {str(e)}")
             if attempt < max_retries - 1:
-                time.sleep(2)  # Wait before retrying
+                time.sleep(2)
                 continue
+            else:
+                return f"Unable to get a response from the assistant: {str(e)}", bot_id, workspace_id
+
+    # If we need new IDs, create them and try again
+    if need_new_ids:
+        print("Creating new workspace and bot IDs...")
 
-            # If this is the last attempt, handle it gracefully
-            return f"Unable to get a response from the assistant after multiple attempts.", bot_id, workspace_id
+        # First, try to clean up old resources
+        if bot_id and workspace_id:
+            delete_bot(bot_id, workspace_id)
+            delete_workspace(workspace_id)
+
+        # Create new resources
+        new_workspace_id = create_workspace()
+        if not new_workspace_id:
+            return "Failed to create a new workspace. Please try again later.", bot_id, workspace_id
+
+        new_bot_id = create_bot(new_workspace_id)
+        if not new_bot_id:
+            return "Failed to create a new bot. Please try again later.", new_workspace_id, workspace_id
+
+        # Update headers with new bot ID
+        headers["x-bot-id"] = new_bot_id
+
+        # Try one more time with the new IDs
+        try:
+            print(f"Retrying with new bot_id={new_bot_id}, workspace_id={new_workspace_id}")
+            retry_response = requests.post(botpress_url, json=payload, headers=headers, timeout=timeout)
+
+            if retry_response.status_code == 200:
+                data = retry_response.json()
+                assistant_content = data.get('output', {}).get('choices', [{}])[0].get('content', '')
+                print(f"Successfully received response with new IDs")
+                return assistant_content, new_bot_id, new_workspace_id
+            else:
+                print(f"Failed with new IDs: {retry_response.status_code}, {retry_response.text}")
+                return f"Unable to get a response from the assistant with new credentials.", new_bot_id, new_workspace_id
+
+        except Exception as e:
+            print(f"Error with new IDs: {str(e)}")
+            return f"Unable to get a response from the assistant with new credentials: {str(e)}", new_bot_id, new_workspace_id
 
     # Should not reach here due to the handling in the loop
     return "Unable to get a response from the assistant.", bot_id, workspace_id
@@ -264,14 +357,17 @@ def chat_endpoint():
 
     # If we don't yet have a workspace or bot, create them
     if not GLOBAL_WORKSPACE_ID or not GLOBAL_BOT_ID:
+        print("No existing IDs found. Creating new workspace and bot...")
         GLOBAL_WORKSPACE_ID = create_workspace()
         if GLOBAL_WORKSPACE_ID:
             GLOBAL_BOT_ID = create_bot(GLOBAL_WORKSPACE_ID)
+
         # If creation failed
         if not GLOBAL_WORKSPACE_ID or not GLOBAL_BOT_ID:
             return jsonify({"assistant_response": "I'm currently unavailable. Please try again later."}), 500
 
     # Call our function that interacts with Botpress API
+    print(f"Sending chat request with existing bot_id={GLOBAL_BOT_ID}, workspace_id={GLOBAL_WORKSPACE_ID}")
     assistant_response, updated_bot_id, updated_workspace_id = chat_with_assistant(
         user_input,
         chat_history,
@@ -280,7 +376,8 @@ def chat_endpoint():
     )
 
     # Update global IDs if they changed
-    if updated_bot_id and updated_workspace_id:
+    if updated_bot_id != GLOBAL_BOT_ID or updated_workspace_id != GLOBAL_WORKSPACE_ID:
+        print(f"Updating global IDs: bot_id={updated_bot_id}, workspace_id={updated_workspace_id}")
         GLOBAL_BOT_ID = updated_bot_id
         GLOBAL_WORKSPACE_ID = updated_workspace_id