Update app.py

app.py

@@ -13,14 +13,6 @@ 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:
@@ -122,7 +114,6 @@ def send_twilio_message(client, conversation_sid, body):
     )
 
 # --- Load Knowledge Base ---
-@st.cache_resource
 def setup_knowledge_base():
     folder_path = "docs"
     all_text = ""
@@ -141,57 +132,52 @@ def setup_knowledge_base():
     index.add(np.array(embeddings).astype('float32'))
     return index, model, chunks
 
-# --- Conversation Monitor ---
+# --- Monitor All Conversations ---
 def start_conversation_monitor(client, index, embed_model, text_chunks):
     last_msg_index = {}
 
-    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()
+    def poll_conversation(convo_sid):
+        while True:
+            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
+                    print(f"\n📥 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)
+                    print(f"📤 Replied to {sender}: {answer}")
+                time.sleep(3)
+            except Exception as e:
+                print(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()
 
 # --- Streamlit UI ---
 st.set_page_config(page_title="Quasa – A Smart WhatsApp Chatbot", layout="wide")
 st.title("📱 Quasa – A Smart WhatsApp Chatbot")
 
-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")
+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 "")
 
 if all([account_sid, auth_token, GROQ_API_KEY]):
     os.environ["GROQ_API_KEY"] = GROQ_API_KEY
     client = Client(account_sid, auth_token)
+    conversation_sids = get_whatsapp_conversation_sids(client)
 
-    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...")
+    if conversation_sids:
+        st.success(f"✅ {len(conversation_sids)} WhatsApp conversation(s) found. Initializing chatbot...")
         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.")
+        st.success("🟢 Chatbot is running in background and will reply to new messages.")
+    else:
+        st.error("❌ No WhatsApp conversations found.")