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 = None
GLOBAL_BOT_ID = None


# -------------------------------------------------------------------
# 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; ..."
    }
    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",
        "x-workspace-id": workspace_id,
        # Replace with your valid cookie below
        "cookie": "pscd=try.botpress.com; ..."
    }
    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",
        "x-workspace-id": workspace_id,
        # Replace with your valid cookie below
        "cookie": "pscd=try.botpress.com; ..."
    }
    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",
        # Replace with your valid cookie below
        "cookie": "pscd=try.botpress.com; ..."
    }
    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",
        "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; ..."
    }

    # Prepare the payload
    payload = {
        "prompt": {
            "messages": chat_history,  # from user request
            "model": "gpt-4o",
            "temperature": 0.9,
            "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)