"""
Utility functions
"""

import re
from io import BytesIO
from PIL import Image
import tempfile
import streamlit as st
import requests
import pandas as pd

from langchain.agents import create_csv_agent
from langchain.llms import OpenAI

from pyairtable import Table

airtable_logo_url = "https://seeklogo.com/images/Q/question-cube-logo-2041FEA436-seeklogo.com.png"

def extract_ids_from_base_url(base_url):
    """
    Extract base and table ID or name from the base URL using regular expressions
    """
    pattern = r'https://airtable.com/([\w\d]+)/(.*?)(?:/|$)'
    match = re.match(pattern, base_url)
  
    if match:
        base_id = match.group(1)
        table_id = match.group(2)

        return dict(base_id=base_id, table_id=table_id)
    else:
        raise ValueError("Invalid base URL")

def airtable_to_csv():
    """
    Convert Airtable contents into csv
    """
    access_token = st.session_state["AIRTABLE_PAT"]
    
    # Extract the base and table ID from the base URL
    ids_from_url = extract_ids_from_base_url(st.session_state["AIRTABLE_URL"]) 
    base_id, table_id = ids_from_url['base_id'], ids_from_url['table_id']
    
    # Initialize Airtable Python SDK
    table = Table(access_token, base_id, table_id)

    # Get all records from the table
    all_records = table.all()

    # Extract the data from the JSON response and create a pandas DataFrame
    rows = []
    for record in all_records:
        row = record['fields']
        row['id'] = record['id']
        rows.append(row)
    df = pd.DataFrame(rows)

    with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
        df.to_csv(tmp_file.name, index=False)

    print(tmp_file.name)
    return tmp_file.name

def clear_submit():
    """
    Clears the 'submit' value in the session state.
    """
    st.session_state["submit"] = False

def run_agent(file_name, query):
    """
    Runs the agent on the given file with the specified query.
    """
    openai_key = st.session_state["OPENAI_API_KEY"]
    agent = create_csv_agent(OpenAI(temperature=0, openai_api_key=openai_key), file_name, verbose=True)
    return agent.run(query).__str__()

def validate_api_key(api_key_input):
    """
    Validates the provided API key.
    """
    api_key_regex = r"^sk-"
    api_key_valid = re.match(api_key_regex, api_key_input) is not None
    return api_key_valid

def validate_pat(airtable_pat_input):
    """
    Validates the provided Airtable personal access token (PAT).
    """
    airtable_pat_regex = r"^pat"
    airtable_pat_valid = re.match(airtable_pat_regex, airtable_pat_input) is not None
    return airtable_pat_valid

def validate_base_url(airtable_base_url_input):
    """
    Validates the provided Airtable base URL.
    """
    airtable_base_url_regex = r"^https:\/\/airtable.com\/app[^\/]+\/tbl[^\/]"
    airtable_base_url_valid = re.match(airtable_base_url_regex, airtable_base_url_input) is not None
    return airtable_base_url_valid

def set_logo_and_page_config():
    """
    Sets the Airtable logo image and page config.
    """
    response = requests.get(airtable_logo_url)
    im = Image.open(BytesIO(response.content))
    st.set_page_config(page_title="Airtable-QA", page_icon=im, layout="wide")
    st.image(airtable_logo_url, width=50)
    st.header("Airtable-QA")
    
def populate_markdown():
    """
    Populates markdown for sidebar.
    """
    st.markdown(
            "## How to use\n"
            "1. Enter your [OpenAI API key](https://platform.openai.com/account/api-keys) below\n" 
            "2. Enter your [Airtable Personal Access Token](https://airtable.com/developers/web/guides/personal-access-tokens#creating-a-token) & Base URL 🔑\n"
            "3. Ask any question that can be answered from Airtable Base\n")
    api_key_input = st.text_input(
            "OpenAI API Key",
            type="password",
            placeholder="sk-...",
            help="You can get your API key from https://platform.openai.com/account/api-keys", 
            value=st.session_state.get("OPENAI_API_KEY", ""))
    airtable_pat_input = st.text_input(
            "Airtable Personal Access Token",
            type="password",
            placeholder="pat...",
            help="You can get your Airtable PAT from https://airtable.com/developers/web/guides/personal-access-tokens#creating-a-token",
            value=st.session_state.get("AIRTABLE_PAT", ""))
    airtable_base_url_input = st.text_input(
            "Airtable Base URL",
            type="password",
            placeholder="https://airtable.com/app.../tbl...",
            help="You can get your Airtable Base URL by simply copy pasting the URL",
            value=st.session_state.get("AIRTABLE_URL", ""))
    return api_key_input, airtable_pat_input, airtable_base_url_input