Update app.py

app.py

@@ -13,6 +13,14 @@ from groq import Groq
 import PyPDF2
 import requests
 
+# --- Session State Initialization ---
+if 'is_running' not in st.session_state:
+    st.session_state.is_running = False
+if 'log' not in st.session_state:
+    st.session_state.log = []
+if 'stop_event' not in st.session_state:
+    st.session_state.stop_event = threading.Event()
+
 # --- Text Extraction Utilities ---
 def extract_text_from_pdf(pdf_path):
     try:
@@ -113,6 +121,7 @@ def send_twilio_message(client, conversation_sid, body):
     )
 
 # --- Load Knowledge Base ---
+@st.cache_resource
 def setup_knowledge_base():
     folder_path = "docs"
     all_text = ""
@@ -131,74 +140,57 @@ def setup_knowledge_base():
     index.add(np.array(embeddings).astype('float32'))
     return index, model, chunks
 
-# --- Conversation Monitor Thread ---
+# --- Conversation Monitor ---
 def start_conversation_monitor(client, index, embed_model, text_chunks):
     last_msg_index = {}
 
-    def poll_conversation(convo_sid):
-        while st.session_state.chatbot_running:
-            try:
-                question, sender, msg_index = fetch_latest_incoming_message(client, convo_sid)
-                if question and (convo_sid not in last_msg_index or msg_index > last_msg_index[convo_sid]):
-                    last_msg_index[convo_sid] = msg_index
-                    log(f"📥 New message from {sender} in {convo_sid}: {question}")
-                    context = "\n\n".join(retrieve_chunks(question, index, embed_model, text_chunks))
-                    answer = generate_answer_with_groq(question, context)
-                    send_twilio_message(client, convo_sid, answer)
-                    log(f"📤 Replied to {sender}: {answer}")
-                time.sleep(3)
-            except Exception as e:
-                log(f"❌ Error in convo {convo_sid} polling: {e}")
-                time.sleep(5)
-
-    for sid in get_whatsapp_conversation_sids(client):
-        threading.Thread(target=poll_conversation, args=(sid,), daemon=True).start()
-
-# --- Logging Utility ---
-def log(message):
-    st.session_state.logs.append(f"[{time.strftime('%H:%M:%S')}] {message}")
+    def poll():
+        while not st.session_state.stop_event.is_set():
+            sids = get_whatsapp_conversation_sids(client)
+            for convo_sid in sids:
+                try:
+                    question, sender, msg_index = fetch_latest_incoming_message(client, convo_sid)
+                    if question and (convo_sid not in last_msg_index or msg_index > last_msg_index[convo_sid]):
+                        last_msg_index[convo_sid] = msg_index
+                        context = "\n\n".join(retrieve_chunks(question, index, embed_model, text_chunks))
+                        answer = generate_answer_with_groq(question, context)
+                        send_twilio_message(client, convo_sid, answer)
+                        log_entry = f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] SID: {convo_sid} | 📥: {question} | 📤: {answer}"
+                        st.session_state.log.append(log_entry)
+                except Exception as e:
+                    st.session_state.log.append(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] ❌ Error in SID {convo_sid}: {e}")
+            time.sleep(5)
+
+    thread = threading.Thread(target=poll, daemon=True)
+    thread.start()
 
 # --- Streamlit UI ---
 st.set_page_config(page_title="Quasa – A Smart WhatsApp Chatbot", layout="wide")
 st.title("📱 Quasa – A Smart WhatsApp Chatbot")
 
-# Initialize session state
-if "chatbot_running" not in st.session_state:
-    st.session_state.chatbot_running = False
-if "logs" not in st.session_state:
-    st.session_state.logs = []
-
-account_sid = st.secrets.get("TWILIO_SID")
-auth_token = st.secrets.get("TWILIO_TOKEN")
-GROQ_API_KEY = st.secrets.get("GROQ_API_KEY")
-
-if not all([account_sid, auth_token, GROQ_API_KEY]):
-    st.warning("⚠️ Provide all credentials below:")
-    account_sid = st.text_input("Twilio SID", value=account_sid or "")
-    auth_token = st.text_input("Twilio Token", type="password", value=auth_token or "")
-    GROQ_API_KEY = st.text_input("GROQ API Key", type="password", value=GROQ_API_KEY or "")
+account_sid = st.secrets.get("TWILIO_SID") or st.text_input("Twilio SID")
+auth_token = st.secrets.get("TWILIO_TOKEN") or st.text_input("Twilio Token", type="password")
+GROQ_API_KEY = st.secrets.get("GROQ_API_KEY") or st.text_input("GROQ API Key", type="password")
 
 if all([account_sid, auth_token, GROQ_API_KEY]):
     os.environ["GROQ_API_KEY"] = GROQ_API_KEY
     client = Client(account_sid, auth_token)
 
-    if not st.session_state.chatbot_running:
-        if st.button("▶️ Start Chatbot"):
-            conversation_sids = get_whatsapp_conversation_sids(client)
-            if conversation_sids:
-                log(f"✅ {len(conversation_sids)} conversation(s) found. Loading knowledge base...")
-                index, model, chunks = setup_knowledge_base()
-                st.session_state.chatbot_running = True
-                log("🟢 Chatbot is running and will reply to new WhatsApp messages.")
-                start_conversation_monitor(client, index, model, chunks)
-            else:
-                st.error("❌ No WhatsApp conversations found.")
-    else:
-        if st.button("⏹ Stop Chatbot"):
-            st.session_state.chatbot_running = False
-            log("🔴 Chatbot stopped.")
-
-# Display logs
-st.subheader("📋 Logs")
-for log_entry in st.session_state.logs[-100:]:
-    st.text(log_entry)
+    if st.button("▶️ Start Chatbot") and not st.session_state.is_running:
+        st.session_state.stop_event.clear()
+        st.session_state.is_running = True
+        st.success("🔄 Loading knowledge base and starting monitor...")
+        index, model, chunks = setup_knowledge_base()
+        start_conversation_monitor(client, index, model, chunks)
+        st.success("🟢 Chatbot is now running...")
+
+    if st.button("⏹ Stop Chatbot") and st.session_state.is_running:
+        st.session_state.stop_event.set()
+        st.session_state.is_running = False
+        st.warning("🛑 Chatbot stopped.")
+
+    st.markdown("### 📋 Logs")
+    for entry in reversed(st.session_state.log[-100:]):
+        st.text(entry)
+else:
+    st.warning("⚠️ Please enter all required credentials.")