Update app.py

app.py

@@ -4,11 +4,16 @@ import time
 import os
 import pandas as pd
 import datetime
+from huggingface_hub import HfApi, upload_file
 
+# Environment variables
 generated_user = os.getenv("User")
 generated_password = os.getenv("Password")
 openai_key = os.getenv("openai_key")
 assistant_id = os.getenv("ASSISTANT_ID")
+hf_token = os.getenv("HF_TOKEN")
+
+transcript_file = "transcripts.xlsx"
 
 st.set_page_config(page_title="Carfind.co.za AI Assistant", layout="wide")
 st.markdown("<h1 style='text-align: center;'>🚗 Carfind.co.za AI Assistant</h1>", unsafe_allow_html=True)
@@ -22,9 +27,7 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Load existing transcript excel or create one
-transcript_file = "transcripts.xlsx"
-
+# Save chat to Excel and push to Hugging Face repo
 def append_to_transcript(thread_id, role, message):
     timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     new_entry = {
@@ -33,18 +36,28 @@ def append_to_transcript(thread_id, role, message):
         "Role": role,
         "Message": message
     }
-
     try:
         if os.path.exists(transcript_file):
             df = pd.read_excel(transcript_file)
             df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)
         else:
             df = pd.DataFrame([new_entry])
-
         df.to_excel(transcript_file, index=False)
+        
+        # Upload updated transcripts.xlsx to Hugging Face
+        api = HfApi(token=hf_token)
+        upload_file(
+            path_or_fileobj=transcript_file,
+            path_in_repo="transcripts.xlsx",
+            repo_id="IAMTFRMZA/cfaiassistant",
+            repo_type="space",
+            token=hf_token,
+            commit_message=f"Transcript update {timestamp}"
+        )
     except Exception as e:
-        st.error(f"Failed to log conversation: {str(e)}")
+        st.error(f"Transcript logging failed: {str(e)}")
 
+# Authentication
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
 
@@ -52,7 +65,6 @@ 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
@@ -60,7 +72,7 @@ if not st.session_state.authenticated:
             time.sleep(1)
             st.rerun()
         else:
-            st.error("Incorrect username or password. Please try again.")
+            st.error("Incorrect username or password.")
 
 else:
     st.divider()
@@ -78,7 +90,7 @@ else:
             st.success("Chat cleared.")
             st.rerun()
 
-    if openai_key and assistant_id:
+    if openai_key and assistant_id and hf_token:
         client = OpenAI(api_key=openai_key)
 
         if user_input:
@@ -86,17 +98,17 @@ else:
                 thread = client.beta.threads.create()
                 st.session_state["thread_id"] = thread.id
 
+            # Add user message and log
             client.beta.threads.messages.create(
                 thread_id=st.session_state["thread_id"], role="user", content=user_input
             )
-            append_to_transcript(st.session_state["thread_id"], "User", user_input)
+            append_to_transcript(st.session_state["thread_id"], "user", user_input)
 
             try:
                 with st.spinner("Thinking and typing... 💭"):
                     run = client.beta.threads.runs.create(
                         thread_id=st.session_state["thread_id"], assistant_id=assistant_id
                     )
-
                     while True:
                         run_status = client.beta.threads.runs.retrieve(
                             thread_id=st.session_state["thread_id"], run_id=run.id
@@ -111,26 +123,24 @@ else:
 
                 assistant_icon_html = "<img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='22' style='vertical-align:middle;'/>"
 
+                # Display chat conversation
                 for msg in reversed(messages_response.data):
-                    content = msg.content[0].text.value
                     if msg.role == "user":
                         st.markdown(
                             f"<div style='background-color:#f0f0f0; color:#000000; padding:10px; border-radius:8px; margin:5px 0;'>"
-                            f"👤 <strong>You:</strong> {content}"
-                            f"</div>", 
-                            unsafe_allow_html=True
+                            f"👤 <strong>You:</strong> {msg.content[0].text.value}"
+                            f"</div>", unsafe_allow_html=True
                         )
                     else:
+                        response_text = msg.content[0].text.value
+                        append_to_transcript(st.session_state["thread_id"], "assistant", response_text)
                         st.markdown(
                             f"<div style='background-color:#D6E9FE; color:#000000; padding:10px; border-radius:8px; margin:5px 0;'>"
-                            f"{assistant_icon_html} <strong>Carfind Assistant:</strong> {content}"
-                            f"</div>", 
-                            unsafe_allow_html=True
+                            f"{assistant_icon_html} <strong>Carfind Assistant:</strong> {response_text}"
+                            f"</div>", unsafe_allow_html=True
                         )
-                        append_to_transcript(st.session_state["thread_id"], "Assistant", content)
-
             except Exception as e:
                 st.error(f"An error occurred: {str(e)}")
-
     else:
-        st.error("⚠️ OpenAI key or Assistant ID not found. Please ensure both are set as Hugging Face secrets.")
+        st.error("⚠️ Missing OpenAI key, Assistant ID or HF token. Please check Hugging Face secrets.")
+