Update app.py

app.py

@@ -11,6 +11,9 @@ import numpy as np
 import docx
 from groq import Groq
 import PyPDF2
+from groq import Groq
+from groq.error import GroqError
+
 
 # --- Document Loaders ---
 def extract_text_from_pdf(pdf_path):
@@ -49,17 +52,27 @@ 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):
+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"])
-    response = groq_client.chat.completions.create(
-        model="llama3-8b-8192",  # Updated model name
-        messages=[
-            {"role": "system", "content": "You are an AI Assistant for Small Businesses."},
-            {"role": "user", "content": prompt},
-        ]
-    )
-    return response.choices[0].message.content
+    
+    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)}"
+
 
 # --- Twilio Chat Handlers ---
 def fetch_latest_incoming_message(account_sid, auth_token, conversation_sid):