Update app.py

app.py

@@ -4,66 +4,77 @@ import time
 import datetime
 import os
 
-# Get OpenAI API key from Hugging Face secret environment variable
+# Securely get credentials from Hugging Face secrets
+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")
-
 st.title("Carfind.co.za AI Assistant")
-st.caption("Chat with Carfind.co.za and Find your Next Car Fast")
+st.caption("Chat with Carfind.co.za and find your next car fast")
 
-# Simple login system
+# 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 st.button("Login"):
-        if username == "[email protected]" and password == "Pass.123":
+    # Auto-login once both fields are filled
+    if username and password:
+        if username == generated_user and password == generated_password:
             st.session_state.authenticated = True
-            st.success("Login successful!")
+            st.success("Login successful! Redirecting...")
+            time.sleep(1)
+            st.rerun()
+        else:
+            st.error("Incorrect username or password. Please try again.")
+
+# Main chat interface
 else:
-    # Sidebar with clear chat button
-    with st.sidebar:
-        if st.button("🧹 Clear Chat"):
-            st.session_state.messages = []
-            st.success("Chat history cleared.")
+    st.divider()
+
+    # Clear chat button positioned near chat window
+    if st.button("🧹 Clear Chat"):
+        st.session_state.messages = []
+        st.success("Chat history cleared.")
 
-    # Initialize OpenAI client
+    # Check for OpenAI key
     if not openai_key:
-        st.error("OpenAI key not found. Please make sure it is set as a secret variable in Hugging Face.")
+        st.error("OpenAI key not found. Please ensure it is set as a Hugging Face secret.")
     else:
         client = OpenAI(api_key=openai_key)
-
         ASSISTANT_ID = "asst_5gQR21fOsmHil11FGBzEArA7"
 
-        # Initialize session state for chat history
+        # Initialize message history
         if "messages" not in st.session_state:
             st.session_state["messages"] = []
 
         # Display chat history
         for message in st.session_state.messages:
-            role, content = message["role"], message["content"]
+            role = message["role"]
+            content = message["content"]
             st.chat_message(role).write(content)
 
-        # Helper function to save chat transcripts (not displayed, backend only)
+        # Save chat transcript (backend only)
         def save_transcript(messages):
-            with open("chat_logs.txt", "a") as f:
-                f.write(f"\n--- New Chat ({datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}) ---\n")
+            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:
-                    f.write(f"{msg['role'].capitalize()}: {msg['content']}\n")
-                f.write("--- End Chat ---\n")
+                    log.write(f"{msg['role'].capitalize()}: {msg['content']}\n")
+                log.write("--- End Chat ---\n")
 
-        # Process user input
+        # Handle user input
         if prompt := st.chat_input("Type your message here..."):
             st.session_state.messages.append({"role": "user", "content": prompt})
             st.chat_message("user").write(prompt)
 
             try:
-                # Create a new thread
+                # Start OpenAI chat thread
                 thread = client.beta.threads.create()
                 thread_id = thread.id
 
@@ -74,29 +85,33 @@ else:
                     content=prompt
                 )
 
-                # Run assistant
+                # Trigger assistant response
                 run = client.beta.threads.runs.create(
                     thread_id=thread_id,
                     assistant_id=ASSISTANT_ID
                 )
 
-                # Wait for completion
+                # Wait for assistant completion
                 while True:
-                    run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
+                    run_status = client.beta.threads.runs.retrieve(
+                        thread_id=thread_id,
+                        run_id=run.id
+                    )
                     if run_status.status == "completed":
                         break
                     time.sleep(1)
 
-                # Get assistant response
+                # Retrieve assistant message
                 messages = client.beta.threads.messages.list(thread_id=thread_id)
                 assistant_message = messages.data[0].content[0].text.value
 
-                # Display and store response
+                # Display and store the assistant’s response
                 st.chat_message("assistant").write(assistant_message)
                 st.session_state.messages.append({"role": "assistant", "content": assistant_message})
 
-                # Save conversation transcript silently
+                # Save chat log
                 save_transcript(st.session_state.messages)
 
             except Exception as e:
-                st.error(f"Error: {str(e)}")
+                st.error(f"An error occurred: {str(e)}")
+