Spaces:
Sleeping
Sleeping
| from flask import Flask, request, jsonify | |
| import requests | |
| import random | |
| import string | |
| import time | |
| app = Flask(__name__) | |
| # ------------------------------------------------------------------- | |
| # Global variables to hold workspace and bot IDs requestsacross | |
| # ------------------------------------------------------------------- | |
| GLOBAL_WORKSPACE_ID = "wkspace_01JR858H3H7EVBXV887X7HVFG4" | |
| GLOBAL_BOT_ID = "271d3c6c-ad82-45d0-a26d-e2759b0b844e" | |
| # 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): | |
| return ''.join(random.choices(string.ascii_letters, k=length)) | |
| # ------------------------------------------------------------------- | |
| # Functions to create/delete workspaces and bots | |
| # ------------------------------------------------------------------- | |
| def create_workspace(): | |
| 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') | |
| 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): | |
| 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.") | |
| # Install integration for the new bot | |
| if bot_id: | |
| install_bot_integration(bot_id, workspace_id) | |
| return bot_id | |
| 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""" | |
| 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_01JN7C46RQXD380E6XNH094WHJ": { | |
| "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): | |
| 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) | |
| except Exception as e: | |
| print(f"Error deleting bot: {str(e)}") | |
| return None | |
| def delete_workspace(workspace_id): | |
| 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) | |
| except Exception as e: | |
| print(f"Error deleting workspace: {str(e)}") | |
| return None | |
| # ------------------------------------------------------------------- | |
| # 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 new 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 new API | |
| payload = { | |
| "type": "anthropic:generateContent", | |
| "input": { | |
| "model": { | |
| "id": "claude-3-7-sonnet-20250219" | |
| }, | |
| "systemPrompt": system_prompt, | |
| "messages": messages, | |
| "temperature": 0.2, | |
| "topP": 0.95, | |
| "debug": False, | |
| } | |
| } | |
| botpress_url = "https://api.botpress.cloud/v1/chat/actions" | |
| max_retries = 5 | |
| timeout = 60 # seconds | |
| # Attempt to send the request | |
| for attempt in range(max_retries): | |
| try: | |
| 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', '') | |
| return assistant_content, bot_id, workspace_id | |
| # Handle 403 error specifically | |
| 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 | |
| # Any other error status code | |
| else: | |
| raise Exception("Unexpected error occurred.") | |
| except Exception as e: | |
| # If this is not the last attempt, retry the same request | |
| if attempt < max_retries - 1: | |
| time.sleep(2) # Wait before retrying | |
| continue | |
| # 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 | |
| # 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: | |
| 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 | |
| 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 and 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) |