Update app.py

app.py

@@ -13,6 +13,7 @@ from groq import Groq
 import PyPDF2
 import requests
 
+# Extract text from PDF with fallback
 # --- Document Loaders ---
 def extract_text_from_pdf(pdf_path):
     try:
@@ -50,42 +51,34 @@ 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
-
+# Generate answer using Groq API with retries and timeout
 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 not set."
+
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
     }
-    prompt = f"Based on the following context, answer the question: '{question}'\n\nContext:\n{context}"
+    prompt = (
+        f"Customer asked: '{question}'\n\n"
+        f"Here is the relevant product or policy info to help:\n{context}\n\n"
+        f"Respond in a friendly and helpful tone as a toy shop support agent."
+    )
     payload = {
         "model": "llama3-8b-8192",
         "messages": [
-            {"role": "system", "content": "You are a real human and you will respond as real human. You will be asked questions and you will respond like a human."},
+            {
+                "role": "system",
+                "content": (
+                    "You are ToyBot, a friendly and helpful WhatsApp assistant for an online toy shop. "
+                    "Your goal is to politely answer customer questions, help them choose the right toys, "
+                    "provide order or delivery information, explain return policies, and guide them through purchases. "
+                    "Always sound warm, helpful, and trustworthy like a professional customer support agent."
+                )
+            },
             {"role": "user", "content": prompt},
         ],
         "temperature": 0.5,
@@ -94,17 +87,18 @@ def generate_answer_with_groq(question, context, retries=3, delay=2):
 
     for attempt in range(retries):
         try:
-            response = requests.post(url, headers=headers, json=payload)
+            response = requests.post(url, headers=headers, json=payload, timeout=10)
+            response.raise_for_status()
             result = response.json()
             return result['choices'][0]['message']['content'].strip()
-        except Exception as e:
-            if "503" in str(e) and attempt < retries - 1:
+        except requests.exceptions.HTTPError as e:
+            if response.status_code == 503 and attempt < retries - 1:
                 time.sleep(delay)
                 continue
             else:
-                return f"⚠️ Groq API Error: {str(e)}"
-
-#-----------------------------------------
+                return f"⚠️ Groq API HTTPError: {e}"
+        except Exception as e:
+            return f"⚠️ Groq API Error: {e}"
 
 # --- Twilio Chat Handlers ---
 def fetch_latest_incoming_message(account_sid, auth_token, conversation_sid):
@@ -130,7 +124,7 @@ st.title("📱 SMEHelpBot + WhatsApp (via Twilio)")
 # Load from Hugging Face secrets
 account_sid = st.secrets.get("TWILIO_SID")
 auth_token = st.secrets.get("TWILIO_TOKEN")
-conversation_sid = "CHe065d01a6b654ca0a698bfb849c980de"
+conversation_sid = "CH5093a87140a147ac926067cca276dbd1"
 GROQ_API_KEY = st.secrets.get("GROQ_API_KEY")
 
 # Fallback for testing