import streamlit as st
from openai import OpenAI
import time
import os
import pandas as pd
from datetime import datetime

# Load environment variables
openai_key = os.getenv("openai_key")
assistant_id = os.getenv("ASSISTANT_ID")

# Set up transcript log
transcript_file = "transcripts.xlsx"
if not os.path.exists(transcript_file):
    df = pd.DataFrame(columns=["timestamp", "thread_id", "role", "message"])
    df.to_excel(transcript_file, index=False)

def append_to_transcript(thread_id, role, message):
    df_existing = pd.read_excel(transcript_file)
    new_entry = pd.DataFrame({
        "timestamp": [datetime.now().strftime("%Y-%m-%d %H:%M:%S")],
        "thread_id": [thread_id],
        "role": [role],
        "message": [message]
    })
    updated_df = pd.concat([df_existing, new_entry], ignore_index=True)
    updated_df.to_excel(transcript_file, index=False)

# Configure page
st.set_page_config(page_title="Carfind.co.za AI Assistant", layout="wide")

# Inject custom styling
st.markdown("""
    <style>
    .block-container {padding-top: 1rem; padding-bottom: 0rem;}
    header {visibility: hidden;}
    .st-emotion-cache-18ni7ap {visibility: hidden;}
    .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; margin-bottom: 10px; }
    .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
    .stChatMessage[data-testid="stChatMessage-assistant"] { background: #D6E9FE; color: #000000; }
    div[data-testid="column"] button {
        border: none;
        background-color: transparent;
        font-size: 1.4rem;
        margin-top: 18px;
    }

    /* Logo bounce animation */
    @keyframes bounceIn {
        0%   { transform: scale(0.7); opacity: 0; }
        60%  { transform: scale(1.1); opacity: 1; }
        80%  { transform: scale(0.95); }
        100% { transform: scale(1); }
    }

    .carfind-logo {
        animation: bounceIn 0.6s ease-out;
    }
    </style>
""", unsafe_allow_html=True)

# Initialize session
if "thread_id" not in st.session_state:
    st.session_state["thread_id"] = None

# Carfind logo + powered by text (animated)
st.markdown(
    """
    <div style='text-align: center; margin-top: 20px; margin-bottom: -10px;'>
        <span style='display: inline-flex; align-items: center; gap: 8px;'>
            <img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='30' class='carfind-logo'/>
            <span style='font-size: 14px; color: gray;'>Powered by Carfind</span>
        </span>
    </div>
    """,
    unsafe_allow_html=True
)

# Chat input + inline clear button
input_col, clear_col = st.columns([9, 1])

with input_col:
    user_input = st.chat_input("Type your message here...")

with clear_col:
    if st.button("🗑️", key="clear-chat", help="Clear Chat"):
        st.session_state["thread_id"] = None
        st.rerun()

# Chat logic
if openai_key and assistant_id:
    client = OpenAI(api_key=openai_key)

    if user_input:
        if not st.session_state["thread_id"]:
            thread = client.beta.threads.create()
            st.session_state["thread_id"] = thread.id

        client.beta.threads.messages.create(
            thread_id=st.session_state["thread_id"], role="user", content=user_input
        )
        append_to_transcript(st.session_state["thread_id"], "user", user_input)

        try:
            with st.spinner("Thinking and typing... 💭"):
                run = client.beta.threads.runs.create(
                    thread_id=st.session_state["thread_id"], assistant_id=assistant_id
                )
                while True:
                    run_status = client.beta.threads.runs.retrieve(
                        thread_id=st.session_state["thread_id"], run_id=run.id
                    )
                    if run_status.status == "completed":
                        break
                    time.sleep(1)

            messages_response = client.beta.threads.messages.list(
                thread_id=st.session_state["thread_id"]
            )

            assistant_icon_html = "<img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='22' style='vertical-align:middle;'/>"
            messages_sorted = sorted(messages_response.data, key=lambda x: x.created_at, reverse=True)

            for msg in messages_sorted:
                if msg.role == "user":
                    st.markdown(
                        f"<div class='stChatMessage' data-testid='stChatMessage-user'>"
                        f"👤 <strong>You:</strong> {msg.content[0].text.value}</div>",
                        unsafe_allow_html=True
                    )
                else:
                    response_text = msg.content[0].text.value
                    append_to_transcript(st.session_state["thread_id"], "assistant", response_text)
                    st.markdown(
                        f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>"
                        f"{assistant_icon_html} <strong>Carfind Assistant:</strong> {response_text}</div>",
                        unsafe_allow_html=True
                    )

        except Exception as e:
            st.error(f"❌ An error occurred: {str(e)}")
else:
    st.error("⚠️ OpenAI key or Assistant ID not found. Please ensure both are set as Hugging Face secrets.")