Spaces:
Running
Running
| 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 app configuration | |
| 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") | |
| # 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"] = [] | |
| # Display chat history | |
| for message in st.session_state.messages: | |
| role = message["role"] | |
| content = message["content"] | |
| if role == "assistant": | |
| st.markdown(f'<img src="https://www.carfind.co.za/images/Carfind-Icon.svg" width="30" style="vertical-align: middle;"> <strong>Carfind Assistant:</strong> {content}', unsafe_allow_html=True) | |
| else: | |
| st.chat_message(role).write(content) | |
| # Input with clear chat button | |
| col_clear, col_input = st.columns([1, 8]) | |
| with col_clear: | |
| if st.button("🧹 Clear Chat"): | |
| st.session_state.messages = [] | |
| st.success("Chat history cleared.") | |
| with col_input: | |
| user_input = st.chat_input("Type your message here...") | |
| 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: | |
| if len(st.session_state.messages) == 0 or user_input != st.session_state.messages[-1]["content"]: | |
| st.session_state.messages.append({"role": "user", "content": user_input}) | |
| st.chat_message("user").write(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.strip() | |
| if assistant_message.lower() != user_input.lower(): | |
| st.markdown(f'<img src="https://www.carfind.co.za/images/Carfind-Icon.svg" width="30" style="vertical-align: middle;"> <strong>Carfind Assistant:</strong> {assistant_message}', unsafe_allow_html=True) | |
| st.session_state.messages.append({"role": "assistant", "content": assistant_message}) | |
| 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.") | |