Spaces:

CORVO-AI
/

transcript

Sleeping

File size: 21,014 Bytes

from flask import Flask, request, jsonify
import requests
import random
import string

app = Flask(__name__)

# -------------------------------------------------------------------
# Global variables to hold workspace and bot IDs across requests
# -------------------------------------------------------------------
GLOBAL_WORKSPACE_ID = "wkspace_01JGP4VPC08MHPJ3HNM51W6SPC"
GLOBAL_BOT_ID = "80a2fefa-1ec9-418c-852e-ca0347ad38c6"


# -------------------------------------------------------------------
# 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",
        # Replace with your valid cookie below
        "cookie": "pscd=try.botpress.com; _gcl_au=1.1.2059955793.1735718381; csrf_token_bd9ac21c34b9f0915e733c3e5305d737d0722c1168be7376b889426b5ec2a298=jodj27LFWsi2+szyWXyyeB9kSLWs8KK+JXM6e0HYjk4=; ory_kratos_session=MTczNTY3NTI2MHx5WlZxQnRHZzgwTHlWYzdDemF6aWFuSk4tOXA2eThZS0FZXy1XSnZyZzBWX3EwRVRBUU1hM2E1Mi1RdG9Da0RhYXVWODctQUk0dUlIVFhaZkl1OUpYdmViOGc0RmVRcEtacFhNeG5HYW9kSDNnMlVsbFVQZjdnQmJrUjV2MzJmLWs2Vmh2YnZ4bGl5ZmhGcF9lVkNxSFBIOWN0dS1Pd3MwR2NidzdLOGpwQVZZS3kzRlF4cU1uOHVVSXNoVnhQaXRCRzlkMHlzaWJNRnlVWkp6MnNrMWwwSWtaZGxNc2JONlRMZ3RTaW5XbzI3X1VGb0daNFhVWC1FakR0aWhNTG5EZDc0cGlCc3NtMUVpdGJPZ2hCV3RIQT09fI4khQhSXNtG4vpJU2FiOKNw13-i91YryQW4fmXKuaRU; _hjSessionUser_2931810=eyJpZCI6ImQ2MGMzYjhkLTlkMjQtNTA0OS1hMzlmLWEzNmI0NzA0NzUxNCIsImNyZWF0ZWQiOjE3MzU3MTg0MDcwNTAsImV4aXN0aW5nIjp0cnVlfQ==; hubspotutk=75739411a4d011b2164c4f3d944ecb94; intercom-device-id-bjzkw2xf=afd0a36b-b229-44e3-828e-60483c80c10c; _hjSessionUser_3339867=eyJpZCI6IjU4ODlmMTY4LWRkNGEtNTJhZS1hZTUzLWZlYWQwM2ZmMTVjNyIsImNyZWF0ZWQiOjE3MzU3MTg1ODM4MDgsImV4aXN0aW5nIjp0cnVlfQ==; _gid=GA1.2.2032466231.1735876345; mp_1195923e954ce61d822842b5832047cd_mixpanel=%7B%22distinct_id%22%3A%20%22anonymous%22%2C%22%24device_id%22%3A%20%22anonymous%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%7D; ajs_user_id=d403ad7b-ea73-4d29-b977-5fd95afd585c; ajs_anonymous_id=cda6139d-cb82-4906-bfac-adaea115b097; _hjSession_2931810=eyJpZCI6IjM1NDk2MWNkLTg5MDgtNDYzNC1iODg1LTYwMDFhNThhZTZmZSIsImMiOjE3MzU5MzQ0MDEwMDcsInMiOjAsInIiOjAsInNiIjowLCJzciI6MCwic2UiOjAsImZzIjowLCJzcCI6MH0=; __hstc=59821234.75739411a4d011b2164c4f3d944ecb94.1735718442141.1735876348035.1735934407019.3; __hssrc=1; _ga_PCC6TBWJY6=GS1.1.1735934403.5.1.1735935221.0.0.0; _gat_UA-226900660-1=1; _gat_UA-226900660-2=1; _ga=GA1.1.1726154447.1735718383; _ga_HKHSWES9V9=GS1.1.1735934403.5.1.1735935221.52.0.515702806; __hssc=59821234.3.1735934407019; _ga_W6YT9YSNLH=GS1.2.1735934407.4.1.1735935225.0.0.0; _ga_CYSS87Q508=GS1.2.1735934408.4.1.1735935225.0.0.0; intercom-session-bjzkw2xf=Q05WOFFxaklPdWhSKzhmKy8ySHNlSEpZYmlEcWNodElvUnFma3dEY0g2dWFoM2g4Rk9MUlRQTXFXZXRESE5uRi0tbU5BejBPN2hnODVTQ1Q2V21ZK0hqdz09--5dd3015c5505a16c03758693d6baa30a8045cc82"
    }
    payload = {"name": generate_random_name()}
    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


def create_bot(workspace_id):
    bot_url = "https://api.botpress.cloud/v1/admin/bots"
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
        "x-workspace-id": workspace_id,
        # Replace with your valid cookie below
        "cookie": "pscd=try.botpress.com; _gcl_au=1.1.2059955793.1735718381; csrf_token_bd9ac21c34b9f0915e733c3e5305d737d0722c1168be7376b889426b5ec2a298=jodj27LFWsi2+szyWXyyeB9kSLWs8KK+JXM6e0HYjk4=; ory_kratos_session=MTczNTY3NTI2MHx5WlZxQnRHZzgwTHlWYzdDemF6aWFuSk4tOXA2eThZS0FZXy1XSnZyZzBWX3EwRVRBUU1hM2E1Mi1RdG9Da0RhYXVWODctQUk0dUlIVFhaZkl1OUpYdmViOGc0RmVRcEtacFhNeG5HYW9kSDNnMlVsbFVQZjdnQmJrUjV2MzJmLWs2Vmh2YnZ4bGl5ZmhGcF9lVkNxSFBIOWN0dS1Pd3MwR2NidzdLOGpwQVZZS3kzRlF4cU1uOHVVSXNoVnhQaXRCRzlkMHlzaWJNRnlVWkp6MnNrMWwwSWtaZGxNc2JONlRMZ3RTaW5XbzI3X1VGb0daNFhVWC1FakR0aWhNTG5EZDc0cGlCc3NtMUVpdGJPZ2hCV3RIQT09fI4khQhSXNtG4vpJU2FiOKNw13-i91YryQW4fmXKuaRU; _hjSessionUser_2931810=eyJpZCI6ImQ2MGMzYjhkLTlkMjQtNTA0OS1hMzlmLWEzNmI0NzA0NzUxNCIsImNyZWF0ZWQiOjE3MzU3MTg0MDcwNTAsImV4aXN0aW5nIjp0cnVlfQ==; hubspotutk=75739411a4d011b2164c4f3d944ecb94; intercom-device-id-bjzkw2xf=afd0a36b-b229-44e3-828e-60483c80c10c; _hjSessionUser_3339867=eyJpZCI6IjU4ODlmMTY4LWRkNGEtNTJhZS1hZTUzLWZlYWQwM2ZmMTVjNyIsImNyZWF0ZWQiOjE3MzU3MTg1ODM4MDgsImV4aXN0aW5nIjp0cnVlfQ==; _gid=GA1.2.2032466231.1735876345; mp_1195923e954ce61d822842b5832047cd_mixpanel=%7B%22distinct_id%22%3A%20%22anonymous%22%2C%22%24device_id%22%3A%20%22anonymous%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%7D; ajs_user_id=d403ad7b-ea73-4d29-b977-5fd95afd585c; ajs_anonymous_id=cda6139d-cb82-4906-bfac-adaea115b097; _hjSession_2931810=eyJpZCI6IjM1NDk2MWNkLTg5MDgtNDYzNC1iODg1LTYwMDFhNThhZTZmZSIsImMiOjE3MzU5MzQ0MDEwMDcsInMiOjAsInIiOjAsInNiIjowLCJzciI6MCwic2UiOjAsImZzIjowLCJzcCI6MH0=; __hstc=59821234.75739411a4d011b2164c4f3d944ecb94.1735718442141.1735876348035.1735934407019.3; __hssrc=1; _ga_PCC6TBWJY6=GS1.1.1735934403.5.1.1735935221.0.0.0; _gat_UA-226900660-1=1; _gat_UA-226900660-2=1; _ga=GA1.1.1726154447.1735718383; _ga_HKHSWES9V9=GS1.1.1735934403.5.1.1735935221.52.0.515702806; __hssc=59821234.3.1735934407019; _ga_W6YT9YSNLH=GS1.2.1735934407.4.1.1735935225.0.0.0; _ga_CYSS87Q508=GS1.2.1735934408.4.1.1735935225.0.0.0; intercom-session-bjzkw2xf=Q05WOFFxaklPdWhSKzhmKy8ySHNlSEpZYmlEcWNodElvUnFma3dEY0g2dWFoM2g4Rk9MUlRQTXFXZXRESE5uRi0tbU5BejBPN2hnODVTQ1Q2V21ZK0hqdz09--5dd3015c5505a16c03758693d6baa30a8045cc82"
    }
    payload = {"name": generate_random_name()}
    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 bot_id
    else:
        print(f"Bot creation failed with: {response.status_code}, {response.text}")
        return None


def delete_bot(bot_id, workspace_id):
    url = f"https://api.botpress.cloud/v1/admin/bots/{bot_id}"
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
        "x-workspace-id": workspace_id,
        # Replace with your valid cookie below
        "cookie": "pscd=try.botpress.com; _gcl_au=1.1.2059955793.1735718381; csrf_token_bd9ac21c34b9f0915e733c3e5305d737d0722c1168be7376b889426b5ec2a298=jodj27LFWsi2+szyWXyyeB9kSLWs8KK+JXM6e0HYjk4=; ory_kratos_session=MTczNTY3NTI2MHx5WlZxQnRHZzgwTHlWYzdDemF6aWFuSk4tOXA2eThZS0FZXy1XSnZyZzBWX3EwRVRBUU1hM2E1Mi1RdG9Da0RhYXVWODctQUk0dUlIVFhaZkl1OUpYdmViOGc0RmVRcEtacFhNeG5HYW9kSDNnMlVsbFVQZjdnQmJrUjV2MzJmLWs2Vmh2YnZ4bGl5ZmhGcF9lVkNxSFBIOWN0dS1Pd3MwR2NidzdLOGpwQVZZS3kzRlF4cU1uOHVVSXNoVnhQaXRCRzlkMHlzaWJNRnlVWkp6MnNrMWwwSWtaZGxNc2JONlRMZ3RTaW5XbzI3X1VGb0daNFhVWC1FakR0aWhNTG5EZDc0cGlCc3NtMUVpdGJPZ2hCV3RIQT09fI4khQhSXNtG4vpJU2FiOKNw13-i91YryQW4fmXKuaRU; _hjSessionUser_2931810=eyJpZCI6ImQ2MGMzYjhkLTlkMjQtNTA0OS1hMzlmLWEzNmI0NzA0NzUxNCIsImNyZWF0ZWQiOjE3MzU3MTg0MDcwNTAsImV4aXN0aW5nIjp0cnVlfQ==; hubspotutk=75739411a4d011b2164c4f3d944ecb94; intercom-device-id-bjzkw2xf=afd0a36b-b229-44e3-828e-60483c80c10c; _hjSessionUser_3339867=eyJpZCI6IjU4ODlmMTY4LWRkNGEtNTJhZS1hZTUzLWZlYWQwM2ZmMTVjNyIsImNyZWF0ZWQiOjE3MzU3MTg1ODM4MDgsImV4aXN0aW5nIjp0cnVlfQ==; _gid=GA1.2.2032466231.1735876345; mp_1195923e954ce61d822842b5832047cd_mixpanel=%7B%22distinct_id%22%3A%20%22anonymous%22%2C%22%24device_id%22%3A%20%22anonymous%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%7D; ajs_user_id=d403ad7b-ea73-4d29-b977-5fd95afd585c; ajs_anonymous_id=cda6139d-cb82-4906-bfac-adaea115b097; _hjSession_2931810=eyJpZCI6IjM1NDk2MWNkLTg5MDgtNDYzNC1iODg1LTYwMDFhNThhZTZmZSIsImMiOjE3MzU5MzQ0MDEwMDcsInMiOjAsInIiOjAsInNiIjowLCJzciI6MCwic2UiOjAsImZzIjowLCJzcCI6MH0=; __hstc=59821234.75739411a4d011b2164c4f3d944ecb94.1735718442141.1735876348035.1735934407019.3; __hssrc=1; _ga_PCC6TBWJY6=GS1.1.1735934403.5.1.1735935221.0.0.0; _gat_UA-226900660-1=1; _gat_UA-226900660-2=1; _ga=GA1.1.1726154447.1735718383; _ga_HKHSWES9V9=GS1.1.1735934403.5.1.1735935221.52.0.515702806; __hssc=59821234.3.1735934407019; _ga_W6YT9YSNLH=GS1.2.1735934407.4.1.1735935225.0.0.0; _ga_CYSS87Q508=GS1.2.1735934408.4.1.1735935225.0.0.0; intercom-session-bjzkw2xf=Q05WOFFxaklPdWhSKzhmKy8ySHNlSEpZYmlEcWNodElvUnFma3dEY0g2dWFoM2g4Rk9MUlRQTXFXZXRESE5uRi0tbU5BejBPN2hnODVTQ1Q2V21ZK0hqdz09--5dd3015c5505a16c03758693d6baa30a8045cc82"
    }
    return requests.delete(url, headers=headers)


def delete_workspace(workspace_id):
    url = f"https://api.botpress.cloud/v1/admin/workspaces/{workspace_id}"
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
        # Replace with your valid cookie below
        "cookie": "pscd=try.botpress.com; _gcl_au=1.1.2059955793.1735718381; csrf_token_bd9ac21c34b9f0915e733c3e5305d737d0722c1168be7376b889426b5ec2a298=jodj27LFWsi2+szyWXyyeB9kSLWs8KK+JXM6e0HYjk4=; ory_kratos_session=MTczNTY3NTI2MHx5WlZxQnRHZzgwTHlWYzdDemF6aWFuSk4tOXA2eThZS0FZXy1XSnZyZzBWX3EwRVRBUU1hM2E1Mi1RdG9Da0RhYXVWODctQUk0dUlIVFhaZkl1OUpYdmViOGc0RmVRcEtacFhNeG5HYW9kSDNnMlVsbFVQZjdnQmJrUjV2MzJmLWs2Vmh2YnZ4bGl5ZmhGcF9lVkNxSFBIOWN0dS1Pd3MwR2NidzdLOGpwQVZZS3kzRlF4cU1uOHVVSXNoVnhQaXRCRzlkMHlzaWJNRnlVWkp6MnNrMWwwSWtaZGxNc2JONlRMZ3RTaW5XbzI3X1VGb0daNFhVWC1FakR0aWhNTG5EZDc0cGlCc3NtMUVpdGJPZ2hCV3RIQT09fI4khQhSXNtG4vpJU2FiOKNw13-i91YryQW4fmXKuaRU; _hjSessionUser_2931810=eyJpZCI6ImQ2MGMzYjhkLTlkMjQtNTA0OS1hMzlmLWEzNmI0NzA0NzUxNCIsImNyZWF0ZWQiOjE3MzU3MTg0MDcwNTAsImV4aXN0aW5nIjp0cnVlfQ==; hubspotutk=75739411a4d011b2164c4f3d944ecb94; intercom-device-id-bjzkw2xf=afd0a36b-b229-44e3-828e-60483c80c10c; _hjSessionUser_3339867=eyJpZCI6IjU4ODlmMTY4LWRkNGEtNTJhZS1hZTUzLWZlYWQwM2ZmMTVjNyIsImNyZWF0ZWQiOjE3MzU3MTg1ODM4MDgsImV4aXN0aW5nIjp0cnVlfQ==; _gid=GA1.2.2032466231.1735876345; mp_1195923e954ce61d822842b5832047cd_mixpanel=%7B%22distinct_id%22%3A%20%22anonymous%22%2C%22%24device_id%22%3A%20%22anonymous%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%7D; ajs_user_id=d403ad7b-ea73-4d29-b977-5fd95afd585c; ajs_anonymous_id=cda6139d-cb82-4906-bfac-adaea115b097; _hjSession_2931810=eyJpZCI6IjM1NDk2MWNkLTg5MDgtNDYzNC1iODg1LTYwMDFhNThhZTZmZSIsImMiOjE3MzU5MzQ0MDEwMDcsInMiOjAsInIiOjAsInNiIjowLCJzciI6MCwic2UiOjAsImZzIjowLCJzcCI6MH0=; __hstc=59821234.75739411a4d011b2164c4f3d944ecb94.1735718442141.1735876348035.1735934407019.3; __hssrc=1; _ga_PCC6TBWJY6=GS1.1.1735934403.5.1.1735935221.0.0.0; _gat_UA-226900660-1=1; _gat_UA-226900660-2=1; _ga=GA1.1.1726154447.1735718383; _ga_HKHSWES9V9=GS1.1.1735934403.5.1.1735935221.52.0.515702806; __hssc=59821234.3.1735934407019; _ga_W6YT9YSNLH=GS1.2.1735934407.4.1.1735935225.0.0.0; _ga_CYSS87Q508=GS1.2.1735934408.4.1.1735935225.0.0.0; intercom-session-bjzkw2xf=Q05WOFFxaklPdWhSKzhmKy8ySHNlSEpZYmlEcWNodElvUnFma3dEY0g2dWFoM2g4Rk9MUlRQTXFXZXRESE5uRi0tbU5BejBPN2hnODVTQ1Q2V21ZK0hqdz09--5dd3015c5505a16c03758693d6baa30a8045cc82"
    }
    return requests.delete(url, headers=headers)


# -------------------------------------------------------------------
# Main function that calls the Botpress GPT-4 endpoint
# -------------------------------------------------------------------
def chat_with_assistant(user_input, chat_history, bot_id, workspace_id):
    """
    Sends the user input and chat history to the Botpress GPT-4 endpoint,
    returns the assistant's response and (possibly updated) bot/workspace IDs.
    """

    # Prepare the headers
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
        "x-bot-id": bot_id,  # existing bot ID (could be None on first try)
        "Content-Type": "application/json",
        # Replace with your valid cookie below
        "Cookie": "pscd=try.botpress.com; _gcl_au=1.1.2059955793.1735718381; csrf_token_bd9ac21c34b9f0915e733c3e5305d737d0722c1168be7376b889426b5ec2a298=jodj27LFWsi2+szyWXyyeB9kSLWs8KK+JXM6e0HYjk4=; ory_kratos_session=MTczNTY3NTI2MHx5WlZxQnRHZzgwTHlWYzdDemF6aWFuSk4tOXA2eThZS0FZXy1XSnZyZzBWX3EwRVRBUU1hM2E1Mi1RdG9Da0RhYXVWODctQUk0dUlIVFhaZkl1OUpYdmViOGc0RmVRcEtacFhNeG5HYW9kSDNnMlVsbFVQZjdnQmJrUjV2MzJmLWs2Vmh2YnZ4bGl5ZmhGcF9lVkNxSFBIOWN0dS1Pd3MwR2NidzdLOGpwQVZZS3kzRlF4cU1uOHVVSXNoVnhQaXRCRzlkMHlzaWJNRnlVWkp6MnNrMWwwSWtaZGxNc2JONlRMZ3RTaW5XbzI3X1VGb0daNFhVWC1FakR0aWhNTG5EZDc0cGlCc3NtMUVpdGJPZ2hCV3RIQT09fI4khQhSXNtG4vpJU2FiOKNw13-i91YryQW4fmXKuaRU; _hjSessionUser_2931810=eyJpZCI6ImQ2MGMzYjhkLTlkMjQtNTA0OS1hMzlmLWEzNmI0NzA0NzUxNCIsImNyZWF0ZWQiOjE3MzU3MTg0MDcwNTAsImV4aXN0aW5nIjp0cnVlfQ==; hubspotutk=75739411a4d011b2164c4f3d944ecb94; intercom-device-id-bjzkw2xf=afd0a36b-b229-44e3-828e-60483c80c10c; _hjSessionUser_3339867=eyJpZCI6IjU4ODlmMTY4LWRkNGEtNTJhZS1hZTUzLWZlYWQwM2ZmMTVjNyIsImNyZWF0ZWQiOjE3MzU3MTg1ODM4MDgsImV4aXN0aW5nIjp0cnVlfQ==; _gid=GA1.2.2032466231.1735876345; mp_1195923e954ce61d822842b5832047cd_mixpanel=%7B%22distinct_id%22%3A%20%22anonymous%22%2C%22%24device_id%22%3A%20%22anonymous%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%7D; ajs_user_id=d403ad7b-ea73-4d29-b977-5fd95afd585c; ajs_anonymous_id=cda6139d-cb82-4906-bfac-adaea115b097; _hjSession_2931810=eyJpZCI6IjM1NDk2MWNkLTg5MDgtNDYzNC1iODg1LTYwMDFhNThhZTZmZSIsImMiOjE3MzU5MzQ0MDEwMDcsInMiOjAsInIiOjAsInNiIjowLCJzciI6MCwic2UiOjAsImZzIjowLCJzcCI6MH0=; __hstc=59821234.75739411a4d011b2164c4f3d944ecb94.1735718442141.1735876348035.1735934407019.3; __hssrc=1; _ga_PCC6TBWJY6=GS1.1.1735934403.5.1.1735935221.0.0.0; _gat_UA-226900660-1=1; _gat_UA-226900660-2=1; _ga=GA1.1.1726154447.1735718383; _ga_HKHSWES9V9=GS1.1.1735934403.5.1.1735935221.52.0.515702806; __hssc=59821234.3.1735934407019; _ga_W6YT9YSNLH=GS1.2.1735934407.4.1.1735935225.0.0.0; _ga_CYSS87Q508=GS1.2.1735934408.4.1.1735935225.0.0.0; intercom-session-bjzkw2xf=Q05WOFFxaklPdWhSKzhmKy8ySHNlSEpZYmlEcWNodElvUnFma3dEY0g2dWFoM2g4Rk9MUlRQTXFXZXRESE5uRi0tbU5BejBPN2hnODVTQ1Q2V21ZK0hqdz09--5dd3015c5505a16c03758693d6baa30a8045cc82"
    }

    # Prepare the payload
    payload = {
        "prompt": {
            "messages": chat_history,  # from user request
            "model": "gpt-4o",
            "temperature": 0.75,
            "signatureVersion": "Jan-2024"
        },
        "variables": {
            "TASK_INPUT": ""
        },
        "options": {
            "origin": "cards/ai-task"
        },
        "origin": "cards/ai-task"
    }

    botpress_url = "https://api.botpress.cloud/v1/cognitive/chat-gpt/query"

    # Attempt to send the request
    try:
        response = requests.post(botpress_url, json=payload, headers=headers)

        # If successful (200)
        if response.status_code == 200:
            data = response.json()
            assistant_content = data.get('choices', [{}])[0].get('message', {}).get('content', '')
            return assistant_content, bot_id, workspace_id

        # If we get a 403, it could be because the bot/workspace IDs are invalid/expired
        elif response.status_code == 403:
            raise Exception("Invalid or expired bot ID.")

        # Other errors
        else:
            return f"Error {response.status_code}: {response.text}", bot_id, workspace_id

    except Exception as e:
        # If invalid/expired bot, create new workspace/bot and retry
        if "Invalid or expired bot ID" in str(e):
            # Attempt to delete old IDs if they exist
            if bot_id and workspace_id:
                delete_bot(bot_id, workspace_id)
                delete_workspace(workspace_id)

            # Create fresh IDs
            new_workspace = create_workspace()
            new_bot = create_bot(new_workspace)
            if not new_workspace or not new_bot:
                return "Failed to regenerate workspace or bot IDs.", None, None

            # Update headers with the new bot ID
            headers["x-bot-id"] = new_bot

            # Retry
            retry_response = requests.post(botpress_url, json=payload, headers=headers)
            if retry_response.status_code == 200:
                data = retry_response.json()
                assistant_content = data.get('choices', [{}])[0].get('message', {}).get('content', '')
                return assistant_content, new_bot, new_workspace
            else:
                return f"Error {retry_response.status_code}: {retry_response.text}", new_bot, new_workspace
        else:
            # Other exceptions
            return f"Unexpected error: {str(e)}", bot_id, workspace_id


# -------------------------------------------------------------------
# Flask Endpoint
# -------------------------------------------------------------------
@app.route("/chat", methods=["POST"])
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()
        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": "Could not create workspace or bot."}), 500

    # Call our function that interacts with Botpress GPT-4
    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
    GLOBAL_BOT_ID = updated_bot_id
    GLOBAL_WORKSPACE_ID = updated_workspace_id

    return jsonify({"assistant_response": assistant_response})


# -------------------------------------------------------------------
# Run the Flask app (example)
# -------------------------------------------------------------------
if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860, debug=True)