Spaces:
Running
Running
File size: 4,030 Bytes
4060393 6dad2a2 c645389 c0332bf 4060393 c0332bf f7e33cd c645389 4060393 c0332bf f7e33cd 4060393 f7e33cd 6dad2a2 f7e33cd 6dad2a2 c0332bf 6dad2a2 f7e33cd 6dad2a2 f7e33cd 555e200 f7e33cd f9476ac c0332bf f9476ac c0332bf f9476ac c0332bf f9476ac 6dad2a2 c0332bf f9476ac c645389 f7e33cd c645389 f7e33cd c645389 f9476ac c0332bf c645389 f9476ac c645389 f7e33cd c645389 c0332bf c645389 f7e33cd c0332bf f9476ac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
import streamlit as st
from openai import OpenAI
import time
import datetime
import os
from streamlit_extras.stylable_container import stylable_container
# 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")
# 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"] = []
# Chat display with smooth scrolling
with stylable_container("chat-container", css="overflow-y: auto; height: 70vh; padding: 10px;"):
for message in st.session_state.messages:
with st.chat_message(message["role"]):
st.markdown(message["content"], unsafe_allow_html=True)
# Input row with Clear button as an icon
col_input, col_clear = st.columns([8, 1])
with col_input:
user_input = st.chat_input("Type your message here...")
with col_clear:
if st.button("🗑️", help="Clear chat"):
st.session_state.messages = []
st.rerun()
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.")
|