Spaces:
Paused
Paused
| from flask import Flask, request, jsonify | |
| import requests | |
| import random | |
| import string | |
| import time | |
| app = Flask(__name__) | |
| # Global variables to store workspace and bot IDs | |
| GLOBAL_WORKSPACE_ID = "wkspace_01JSVP5GQ8V4YE1HD4KQCKCW3K" | |
| GLOBAL_BOT_ID = "837a83e-f841-44ea-b978-273312513e24" | |
| # 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" | |
| # ------------------------------------------------------------------- | |
| # 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() | |
| 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 | |
| except Exception as e: | |
| print(f"Error creating workspace: {str(e)}") | |
| return None | |
| 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", | |
| "x-workspace-id": workspace_id, | |
| "Cookie": AUTH_COOKIE, | |
| "Content-Type": "application/json" | |
| } | |
| payload = {"name": generate_random_name()} | |
| try: | |
| response = requests.post(bot_url, headers=headers, json=payload) | |
| if response.status_code == 200: | |
| response_json = response.json() | |
| 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 | |
| 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 | |
| except Exception as e: | |
| print(f"Error creating bot: {str(e)}") | |
| return None | |
| 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", | |
| "Cookie": AUTH_COOKIE, | |
| "Content-Type": "application/json", | |
| "x-bot-id": bot_id, | |
| "x-workspace-id": workspace_id | |
| } | |
| # Integration payload | |
| payload = { | |
| "integrations": { | |
| "intver_01JSERFV00FYHW4SM6TEKZ9RWS": { | |
| "enabled": True | |
| } | |
| } | |
| } | |
| try: | |
| response = requests.put(url, headers=headers, json=payload) | |
| if response.status_code == 200: | |
| print(f"Successfully installed integration for bot {bot_id}") | |
| return True | |
| else: | |
| print(f"Failed to install integration: {response.status_code}, {response.text}") | |
| return False | |
| except Exception as e: | |
| print(f"Error installing integration: {str(e)}") | |
| return False | |
| 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: | |
| 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 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: | |
| 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 False | |
| # ------------------------------------------------------------------- | |
| # Main function that calls the Botpress API endpoint | |
| # ------------------------------------------------------------------- | |
| def chat_with_assistant(user_input, chat_history, bot_id, workspace_id): | |
| """ | |
| Sends the user input and chat history to the Botpress API endpoint, | |
| returns the assistant's response and (possibly updated) bot/workspace IDs. | |
| """ | |
| # Prepare the headers | |
| headers = { | |
| "User-Agent": "Mozilla/5.0", | |
| "x-bot-id": bot_id, | |
| "Content-Type": "application/json", | |
| "Cookie": AUTH_COOKIE | |
| } | |
| # Process chat history into the format expected by the API | |
| messages = [] | |
| system_prompt = "" | |
| for msg in chat_history: | |
| if msg["role"] == "system": | |
| system_prompt = msg["content"] | |
| else: | |
| messages.append({ | |
| "role": msg["role"], | |
| "content": msg["content"], | |
| "type": "text" | |
| }) | |
| # Add the latest user input if not already in chat history | |
| if user_input and (not messages or messages[-1]["role"] != "user" or messages[-1]["content"] != user_input): | |
| messages.append({ | |
| "role": "user", | |
| "content": user_input, | |
| "type": "text" | |
| }) | |
| # Prepare the payload for the API | |
| payload = { | |
| "type": "openai:generateContent", | |
| "input": { | |
| "model": { | |
| "id": "gpt-4.1-2025-04-14" | |
| }, | |
| "systemPrompt": system_prompt, | |
| "messages": messages, | |
| "temperature": 0.5, | |
| "topP": 1, | |
| "debug": False, | |
| } | |
| } | |
| botpress_url = "https://api.botpress.cloud/v1/chat/actions" | |
| 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 (authentication/authorization issue) | |
| elif response.status_code == 403: | |
| 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: | |
| 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: | |
| print(f"Error during request: {str(e)}") | |
| if attempt < max_retries - 1: | |
| 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...") | |
| # 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 | |
| # ------------------------------------------------------------------- | |
| # Flask Endpoint | |
| # ------------------------------------------------------------------- | |
| def chat_endpoint(): | |
| """ | |
| Expects JSON with: | |
| { | |
| "user_input": "string", | |
| "chat_history": [ | |
| {"role": "system", "content": "..."}, | |
| {"role": "user", "content": "..."}, | |
| ... | |
| ] | |
| } | |
| Returns JSON with: | |
| { | |
| "assistant_response": "string" | |
| } | |
| """ | |
| global GLOBAL_WORKSPACE_ID, GLOBAL_BOT_ID | |
| # Parse JSON from request | |
| data = request.get_json(force=True) | |
| user_input = data.get("user_input", "") | |
| chat_history = data.get("chat_history", []) | |
| # 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, | |
| GLOBAL_BOT_ID, | |
| GLOBAL_WORKSPACE_ID | |
| ) | |
| # Update global IDs if they changed | |
| 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 | |
| return jsonify({"assistant_response": assistant_response}) | |
| # ------------------------------------------------------------------- | |
| # Run the Flask app | |
| # ------------------------------------------------------------------- | |
| if __name__ == "__main__": | |
| app.run(host="0.0.0.0", port=7860, debug=True) |