Update app.py

app.py

@@ -50,6 +50,30 @@ def retrieve_chunks(question, index, embed_model, text_chunks, k=3):
     D, I = index.search(np.array([question_embedding]), k)
     return [text_chunks[i] for i in I[0]]
 
+#def generate_answer_with_groq(question, context, retries=3, delay=2):
+#    prompt = f"Based on the following context, answer the question: '{question}'\n\nContext:\n{context}"
+#    groq_client = Groq(api_key=os.environ["GROQ_API_KEY"])
+    
+#    for attempt in range(retries):
+#        try:
+#            response = groq_client.chat.completions.create(
+#                model="llama3-8b-8192",
+#                messages=[
+#                    {"role": "system", "content": "You are an AI Assistant for Small Businesses."},
+#                    {"role": "user", "content": prompt},
+#                ]
+#            )
+#            return response.choices[0].message.content
+#        except GroqError as e:
+#            if "503" in str(e) and attempt < retries - 1:
+#                time.sleep(delay)
+#                continue
+#            else:
+#                return f"⚠️ Groq API Error: {str(e)}"
+
+#-----------------------------------------
+import requests  # ✅ Add this at the top of your file
+
 def generate_answer_with_groq(question, context, retries=3, delay=2):
     url = "https://api.groq.com/openai/v1/chat/completions"
     api_key = os.environ["GROQ_API_KEY"]
@@ -80,6 +104,8 @@ def generate_answer_with_groq(question, context, retries=3, delay=2):
             else:
                 return f"⚠️ Groq API Error: {str(e)}"
 
+#-----------------------------------------
+
 # --- Twilio Chat Handlers ---
 def fetch_latest_incoming_message(account_sid, auth_token, conversation_sid):
     client = Client(account_sid, auth_token)
@@ -137,29 +163,20 @@ if all([account_sid, auth_token, conversation_sid, GROQ_API_KEY]):
 
     index, embedding_model, text_chunks = setup_knowledge_base()
 
-    st.success("✅ Knowledge base ready. Monitoring WhatsApp every 5 seconds...")
-
-    # --- Auto-refresh logic ---
-    st.markdown("""
-        <meta http-equiv="refresh" content="5">
-    """, unsafe_allow_html=True)
-
-    if "last_msg_index" not in st.session_state:
-        st.session_state.last_msg_index = -1
-
-    with st.spinner("Checking for new WhatsApp messages..."):
-        question, sender, msg_index = fetch_latest_incoming_message(account_sid, auth_token, conversation_sid)
-        if question and msg_index != st.session_state.last_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)
-            st.success("📤 Answer sent via WhatsApp!")
-            st.markdown(f"### ✨ Answer:\n\n{answer}")
-            st.session_state.last_msg_index = msg_index
-        else:
-            st.write("⏳ No new WhatsApp queries found.")
-
+    st.success("✅ Knowledge base ready. Monitoring WhatsApp...")
+
+    if st.button("🔁 Check for New WhatsApp Query"):
+        with st.spinner("Checking messages..."):
+            question, sender, msg_index = fetch_latest_incoming_message(account_sid, auth_token, conversation_sid)
+            if question:
+                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)
+                st.success("📤 Answer sent via WhatsApp!")
+                st.markdown(f"### ✨ Answer:\n\n{answer}")
+            else:
+                st.warning("No new messages from users found.")
 else:
     st.warning("❗ Please provide all required credentials.")