Update app.py

app.py

@@ -131,33 +131,43 @@ def setup_knowledge_base():
     index.add(np.array(embeddings).astype('float32'))
     return index, model, chunks
 
-# --- Monitor All Conversations ---
+# --- Conversation Monitor Thread ---
 def start_conversation_monitor(client, index, embed_model, text_chunks):
     last_msg_index = {}
 
     def poll_conversation(convo_sid):
-        while True:
+        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
-                    print(f"\n📥 New message from {sender} in {convo_sid}: {question}")
+                    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)
-                    print(f"📤 Replied to {sender}: {answer}")
+                    log(f"📤 Replied to {sender}: {answer}")
                 time.sleep(3)
             except Exception as e:
-                print(f"❌ Error in convo {convo_sid} polling:", 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}")
+
 # --- 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")
@@ -171,12 +181,24 @@ if not all([account_sid, auth_token, GROQ_API_KEY]):
 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 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 running in background and will reply to new messages.")
+    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:
-        st.error("❌ No WhatsApp conversations found.")
+        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)