Update app.py

app.py

@@ -13,7 +13,7 @@ import PyPDF2
 import requests
 from streamlit_autorefresh import st_autorefresh
 
-# --- Document Loaders ---
+# Extract text from PDF
 def extract_text_from_pdf(pdf_path):
     try:
         text = ""
@@ -98,23 +98,30 @@ def generate_answer_with_groq(question, context, retries=3, delay=2):
         except Exception as e:
             return f"⚠️ Groq API Error: {e}"
 
-def fetch_latest_incoming_message(account_sid, auth_token, conversation_sid):
-    client = Client(account_sid, auth_token)
+def fetch_latest_incoming_message(client, conversation_sid):
     messages = client.conversations.v1.conversations(conversation_sid).messages.list(limit=10)
     for msg in reversed(messages):
         if msg.author.startswith("whatsapp:"):
             return msg.body, msg.author, msg.index
     return None, None, None
 
-def send_twilio_message(account_sid, auth_token, conversation_sid, body):
+def send_twilio_message(client, conversation_sid, body):
     try:
-        client = Client(account_sid, auth_token)
         message = client.conversations.v1.conversations(conversation_sid).messages.create(author="system", body=body)
         return message.sid
     except Exception as e:
         return str(e)
 
-# --- Streamlit UI ---
+# Automatically get the latest WhatsApp conversation SID
+def get_latest_whatsapp_conversation_sid(client):
+    conversations = client.conversations.v1.conversations.list(limit=20)
+    for convo in conversations:
+        if convo.chat_service_sid and "whatsapp" in convo.friendly_name.lower():
+            return convo.sid
+    # fallback to the first available conversation
+    return conversations[0].sid if conversations else None
+
+# Streamlit UI
 st.set_page_config(page_title="Quasa – A Smart WhatsApp Chatbot", layout="wide")
 st.title("📱 Quasa – A Smart WhatsApp Chatbot")
 
@@ -124,17 +131,27 @@ if "last_index" not in st.session_state:
 account_sid = st.secrets.get("TWILIO_SID")
 auth_token = st.secrets.get("TWILIO_TOKEN")
 GROQ_API_KEY = st.secrets.get("GROQ_API_KEY")
-conversation_sid = st.secrets.get("TWILIO_CONVERSATION_SID")
 
-if not all([account_sid, auth_token, GROQ_API_KEY, conversation_sid]):
+if not all([account_sid, auth_token, GROQ_API_KEY]):
     st.warning("⚠️ Some secrets not found. Please enter missing credentials below:")
     account_sid = st.text_input("Twilio SID", value=account_sid or "")
     auth_token = st.text_input("Twilio Auth Token", type="password", value=auth_token or "")
     GROQ_API_KEY = st.text_input("GROQ API Key", type="password", value=GROQ_API_KEY or "")
-    conversation_sid = st.text_input("Twilio Conversation SID", value=conversation_sid or "")
 
-if all([account_sid, auth_token, GROQ_API_KEY, conversation_sid]):
+enable_autorefresh = st.checkbox("🔄 Enable Auto-Refresh", value=True)
+interval_seconds = st.selectbox("Refresh Interval (seconds)", options=[5, 10, 15, 30, 60], index=4)
+
+if enable_autorefresh:
+    st_autorefresh(interval=interval_seconds * 1000, key="auto-refresh")
+
+if all([account_sid, auth_token, GROQ_API_KEY]):
     os.environ["GROQ_API_KEY"] = GROQ_API_KEY
+    client = Client(account_sid, auth_token)
+    conversation_sid = get_latest_whatsapp_conversation_sid(client)
+
+    if conversation_sid is None:
+        st.error("❌ No WhatsApp conversation found.")
+        st.stop()
 
     @st.cache_data(show_spinner=False)
     def setup_knowledge_base():
@@ -158,31 +175,24 @@ if all([account_sid, auth_token, GROQ_API_KEY, conversation_sid]):
             st.error(f"Error setting up knowledge base: {e}")
             return None, None, None
 
-    st.info("🔄 Loading knowledge base...")
     index, embedding_model, text_chunks = setup_knowledge_base()
-
     if index is None:
         st.stop()
 
     st.success("✅ Knowledge base ready. Monitoring WhatsApp...")
 
-    enable_autorefresh = st.checkbox("🔄 Enable Auto-Refresh", value=True)
-    interval_seconds = st.selectbox("Refresh Interval (seconds)", options=[5, 10, 15, 30, 60], index=4)
-    if enable_autorefresh:
-        st_autorefresh(interval=interval_seconds * 1000, key="auto-refresh")
-
     with st.spinner("⏳ Checking for new WhatsApp messages..."):
-        question, sender, msg_index = fetch_latest_incoming_message(account_sid, auth_token, conversation_sid)
+        question, sender, msg_index = fetch_latest_incoming_message(client, conversation_sid)
         if question and msg_index > st.session_state.last_index:
             st.session_state.last_index = msg_index
             st.info(f"📥 New Question from {sender}:\n\n> {question}")
             relevant_chunks = retrieve_chunks(question, index, embedding_model, text_chunks)
             context = "\n\n".join(relevant_chunks)
             answer = generate_answer_with_groq(question, context)
-            send_twilio_message(account_sid, auth_token, conversation_sid, answer)
+            send_twilio_message(client, conversation_sid, answer)
             st.success("📤 Answer sent via WhatsApp!")
             st.markdown(f"### ✨ Answer:\n\n{answer}")
         else:
             st.caption("✅ No new message yet. Waiting for refresh...")
 else:
-    st.warning("❗ Please provide all required credentials and conversation SID.")
+    st.warning("❗ Please provide all required credentials.")