Update app.py

app.py

@@ -5,6 +5,7 @@ import string
 import time
 
 app = Flask(__name__)
+
 # Global variables to store workspace and bot IDs
 GLOBAL_WORKSPACE_ID = None
 GLOBAL_BOT_ID = None
@@ -231,8 +232,8 @@ def chat_with_assistant(user_input, chat_history, bot_id, workspace_id):
     max_retries = 3
     timeout = 120  # Increased timeout for long messages
 
-    # Flag to track if we need to create new IDs
-    need_new_ids = False
+    # Flag to track if we need to create new IDs due to quota exceeded
+    quota_exceeded = False
 
     # Attempt to send the request
     for attempt in range(max_retries):
@@ -247,19 +248,26 @@ def chat_with_assistant(user_input, chat_history, bot_id, workspace_id):
                 print(f"Successfully received response from Botpress API")
                 return assistant_content, bot_id, workspace_id
 
-            # Handle 403 error (authentication/authorization issue)
+            # Check for quota exceeded error specifically
             elif response.status_code == 403:
-                print(f"Received 403 error. Need to create new IDs.")
-                need_new_ids = True
-                break
-                
-            elif response.status_code == 404:
-                print(f"Received 403 error. Need to create new IDs.")
-                need_new_ids = True
-                break
+                error_data = response.json()
+                error_message = error_data.get('message', '')
+
+                # Check if this is the specific quota exceeded error
+                if "has reached its usage limit for ai spend" in error_message:
+                    print(f"Quota exceeded error detected: {error_message}")
+                    quota_exceeded = True
+                    break
+                else:
+                    print(f"Received 403 error but not quota exceeded: {error_message}")
+                    if attempt < max_retries - 1:
+                        time.sleep(2)
+                        continue
+                    else:
+                        return f"Unable to get a response from the assistant (Error 403).", bot_id, workspace_id
 
             # Handle network errors or timeouts (just retry)
-            elif response.status_code in [443, 408, 502, 503, 504]:
+            elif response.status_code in [404, 408, 502, 503, 504]:
                 print(f"Received error {response.status_code}. Retrying...")
                 time.sleep(3)  # Wait before retrying
                 continue
@@ -289,23 +297,35 @@ def chat_with_assistant(user_input, chat_history, bot_id, workspace_id):
             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 quota exceeded, we need to create new resources
+    if quota_exceeded:
+        print("Quota exceeded. Creating new workspace and bot...")
 
-        # First, try to clean up old resources
+        # First delete the bot, then the workspace (in that order)
         if bot_id and workspace_id:
-            delete_bot(bot_id, workspace_id)
-            delete_workspace(workspace_id)
+            print(f"Deleting bot {bot_id} first...")
+            delete_success = delete_bot(bot_id, workspace_id)
+            if delete_success:
+                print(f"Successfully deleted bot {bot_id}")
+            else:
+                print(f"Failed to delete bot {bot_id}")
+
+            print(f"Now deleting workspace {workspace_id}...")
+            ws_delete_success = delete_workspace(workspace_id)
+            if ws_delete_success:
+                print(f"Successfully deleted workspace {workspace_id}")
+            else:
+                print(f"Failed to delete workspace {workspace_id}")
 
-        # Create new resources
+        # Create new workspace
         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
+            return "Failed to create a new workspace after quota exceeded. Please try again later.", bot_id, workspace_id
 
+        # Create new bot in the new workspace
         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
+            return "Failed to create a new bot after quota exceeded. Please try again later.", new_workspace_id, workspace_id
 
         # Update headers with new bot ID
         headers["x-bot-id"] = new_bot_id