app.py

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

# Securely get credentials
generated_user = os.getenv("User")
generated_password = os.getenv("Password")
openai_key = os.getenv("openai_key")

# Streamlit page config
st.set_page_config(page_title="Carfind.co.za AI Assistant", layout="wide")
st.title("🚗 Carfind.co.za AI Assistant")
st.caption("Chat with Carfind.co.za and find your next car fast")

# Custom CSS for styling chat input and clear button inside it
st.markdown(
    """
    <style>
    .chat-container {
        overflow-y: auto;
        max-height: 70vh;
        padding: 10px;
        border-radius: 10px;
        border: 1px solid #444;
        background-color: #111;
        margin-bottom: 80px;
    }
    .stChatInputContainer {
        position: fixed;
        bottom: 20px;
        width: 80%;
        left: 10%;
        background-color: #222;
        padding: 10px;
        border-radius: 10px;
        box-shadow: 0 0 15px rgba(0,0,0,0.5);
        display: flex;
        align-items: center;
    }
    .clear-chat-icon {
        cursor: pointer;
        margin-left: 10px;
        font-size: 1.4em;
    }
    </style>
    """,
    unsafe_allow_html=True
)

# Initialize authentication state
if "authenticated" not in st.session_state:
    st.session_state.authenticated = False

# Login screen
if not st.session_state.authenticated:
    st.subheader("🔐 Login")
    username = st.text_input("Username")
    password = st.text_input("Password", type="password")

    if username and password:
        if username == generated_user and password == generated_password:
            st.session_state.authenticated = True
            st.success("Login successful! Redirecting...")
            time.sleep(1)
            st.rerun()
        else:
            st.error("Incorrect username or password. Please try again.")

# Main chat interface
else:
    st.divider()

    if "messages" not in st.session_state:
        st.session_state["messages"] = []

    # Scrollable chat display
    with st.container():
        st.markdown('<div class="chat-container">', unsafe_allow_html=True)
        for message in st.session_state["messages"]:
            with st.chat_message(message["role"]):
                st.markdown(message["content"], unsafe_allow_html=True)
        st.markdown('</div>', unsafe_allow_html=True)

    # Custom chat input row with clear chat icon
    with st.container():
        user_input = st.chat_input("Type your message here...")
        st.markdown(
            """
            <script>
            function clearChat() {
                fetch(window.location.href, {
                    method: 'POST',
                    headers: {'Content-Type': 'application/json'},
                    body: JSON.stringify({"clear_chat": true})
                }).then(() => location.reload());
            }
            </script>
            <span class='clear-chat-icon' onclick='clearChat()'>🗑️</span>
            """,
            unsafe_allow_html=True
        )

    # Listen for clear chat POST
    if st.experimental_get_query_params().get("clear_chat"):
        st.session_state.messages = []

    if openai_key:
        client = OpenAI(api_key=openai_key)
        ASSISTANT_ID = "asst_5gQR21fOsmHil11FGBzEArA7"

        def save_transcript(messages):
            with open("chat_logs.txt", "a") as log:
                log.write(f"\n--- New Chat ({datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}) ---\n")
                for msg in messages:
                    log.write(f"{msg['role'].capitalize()}: {msg['content']}\n")
                log.write("--- End Chat ---\n")

        if user_input:
            st.session_state.messages.append({"role": "user", "content": user_input})
            with st.chat_message("user"):
                st.markdown(user_input)

            try:
                thread = client.beta.threads.create()
                thread_id = thread.id

                client.beta.threads.messages.create(
                    thread_id=thread_id,
                    role="user",
                    content=user_input
                )

                run = client.beta.threads.runs.create(
                    thread_id=thread_id,
                    assistant_id=ASSISTANT_ID
                )

                while True:
                    run_status = client.beta.threads.runs.retrieve(
                        thread_id=thread_id,
                        run_id=run.id
                    )
                    if run_status.status == "completed":
                        break
                    time.sleep(1)

                messages = client.beta.threads.messages.list(thread_id=thread_id)
                assistant_message = messages.data[0].content[0].text.value

                st.session_state.messages.append({"role": "assistant", "content": assistant_message})

                with st.chat_message("assistant"):
                    st.markdown(assistant_message, unsafe_allow_html=True)

                save_transcript(st.session_state.messages)

            except Exception as e:
                st.error(f"An error occurred: {str(e)}")

    else:
        st.error("OpenAI key not found. Please ensure it is set as a Hugging Face secret.")