from dotenv import load_dotenv
import streamlit as st
import os
from utils._graph_util import run_customer_support


# Initialize categories in session state
if "categories" not in st.session_state:
    st.session_state.categories = {
        "HR": [],
        "IT": [],
        "Transportation": [],
        "Other": []
    }

def main():
    load_dotenv()
    
    # Add detailed API key verification
    api_key = os.getenv("OPENAI_API_KEY")
    if not api_key:
        st.error("❌ OpenAI API key not found! Please ensure it's set in the environment variables.")
        st.info("To set up your API key:")
        st.code("1. Go to Hugging Face Space settings\n2. Add OPENAI_API_KEY in Repository Secrets")
        st.stop()
        
    
    # Page configuration
    st.set_page_config(
        page_title="Intelligent Customer Support Agent",
        page_icon="🤖",
        layout="wide"
    )

    # Sidebar for API key
    with st.sidebar:
        #openai_api_key = st.text_input("OpenAI API Key", type="password")
        st.markdown("---")
        st.markdown("""
        ### About
        This is an AI-powered customer support agent that can answer questions or raise support ticket about the company policies and procedures:
        - HR policies
        - IT policies
        - Transportation policies
        - Other policies
        """)
        

    # Main chat interface
    st.title("🤖 Intelligent Customer Support Agent")
    st.caption("Your 24/7 AI Customer Service Representative")


    st.header("Automatic Ticket Classification Tool")
    #Capture user input
    st.write("We are here to help you, please ask your question:")
    prompt = st.text_input("🔍")

    if prompt:
        if "vector_store" not in st.session_state:
            st.error("Please load the document data first!")
            st.stop()
    
        
        response = run_customer_support(prompt)
        st.write(response.get("response"))
        #Button to create a ticket with respective department
        button = st.button("Submit ticket?")
        
        if button:
            category = response.get("category")
            st.session_state.categories[category].append(prompt)
            st.success("Ticket submitted successfully!")
            # Display category (optional)
            st.write(f"Category: {category}")
        

if __name__ == '__main__':
    main()