Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 import streamlit as st
+from streamlit_autorefresh import st_autorefresh
 from twilio.rest import Client
 from pdfminer.high_level import extract_text
 from sentence_transformers import SentenceTransformer
@@ -8,16 +9,11 @@ from transformers import AutoTokenizer
 import faiss
 import numpy as np
 import docx
-from groq import Groq
 import PyPDF2
 import requests
 
 # --- Auto-refresh every 10 seconds ---
-if "last_refresh" not in st.session_state:
-    st.session_state.last_refresh = time.time()
-elif time.time() - st.session_state.last_refresh > 10:
-    st.session_state.last_refresh = time.time()
-    st.experimental_rerun()
+count = st_autorefresh(interval=10_000, limit=None, key="refresh")
 
 # --- Document Loaders ---
 def extract_text_from_pdf(pdf_path):
@@ -59,7 +55,10 @@ def retrieve_chunks(question, index, embed_model, text_chunks, k=3):
 # --- GROQ Answer Generation ---
 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"]
+    api_key = os.environ.get("GROQ_API_KEY")
+    if not api_key:
+        return "⚠️ GROQ_API_KEY is not set."
+
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
@@ -121,9 +120,7 @@ def fetch_latest_incoming_message(account_sid, auth_token, conversation_sid):
 def send_twilio_message(account_sid, auth_token, conversation_sid, body):
     try:
         client = Client(account_sid, auth_token)
-        # Get participants to find the bot's WhatsApp identity
         participants = client.conversations.v1.conversations(conversation_sid).participants.list()
-        # Pick the first participant whose identity starts with 'whatsapp:'
         bot_identity = None
         for p in participants:
             if p.identity.startswith("whatsapp:"):
@@ -140,7 +137,6 @@ def send_twilio_message(account_sid, auth_token, conversation_sid, body):
     except Exception as e:
         return str(e)
 
-
 # --- Streamlit UI ---
 st.set_page_config(page_title="Quasa – A Smart WhatsApp Chatbot", layout="wide")
 
@@ -160,15 +156,9 @@ st.markdown('<div class="big-font">📱 Quasa – A Smart WhatsApp Chatbot</div>
 st.markdown('<div class="small-font">Talk to your documents using WhatsApp. Powered by Groq, Twilio, and RAG.</div>', unsafe_allow_html=True)
 
 # Load secrets or fallback
-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("⚠️ Some secrets are missing. Please provide them manually:")
-    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 "")
+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 Auth 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