Spaces:
Running
Running
File size: 3,654 Bytes
4060393 f88fdb3 4060393 6dad2a2 c645389 4060393 c0332bf f7e33cd c645389 4060393 f88fdb3 c0332bf f7e33cd 4060393 f7e33cd 6dad2a2 f7e33cd 6dad2a2 c0332bf 6dad2a2 f7e33cd 6dad2a2 f7e33cd f88fdb3 555e200 f9476ac f88fdb3 c645389 f7e33cd c645389 f7e33cd c645389 f88fdb3 c645389 f88fdb3 c645389 f88fdb3 c645389 f88fdb3 c645389 f88fdb3 c645389 f88fdb3 c645389 f88fdb3 c645389 f88fdb3 c0332bf f88fdb3 c645389 f88fdb3 c0332bf f88fdb3 f2177cd |
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 |
import streamlit as st
from streamlit_chat import message
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 UI using streamlit-chat
else:
if "messages" not in st.session_state:
st.session_state["messages"] = []
st.divider()
# Display chat history
for i, msg in enumerate(st.session_state["messages"]):
message(msg["content"], is_user=(msg["role"] == "user"), key=str(i))
# Chat input with clear chat icon
col_input, col_clear = st.columns([10, 1])
with col_input:
user_input = st.text_input("Type your message here:", key="input")
with col_clear:
if st.button("🗑️", help="Clear chat"):
st.session_state.messages = []
st.experimental_rerun()
# Handle OpenAI assistant
if user_input and openai_key:
st.session_state.messages.append({"role": "user", "content": user_input})
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")
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_response = client.beta.threads.messages.list(thread_id=thread_id)
assistant_reply = messages_response.data[0].content[0].text.value
st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
save_transcript(st.session_state.messages)
st.experimental_rerun()
except Exception as e:
st.error(f"An error occurred: {str(e)}")
elif not openai_key:
st.error("OpenAI key not found. Please ensure it is set as a Hugging Face secret.")
|