Update app.py

app.py

@@ -11,6 +11,7 @@ import numpy as np
 import docx
 from groq import Groq
 import PyPDF2
+import requests
 
 # --- Document Loaders ---
 def extract_text_from_pdf(pdf_path):
@@ -71,13 +72,16 @@ def retrieve_chunks(question, index, embed_model, text_chunks, k=3):
 #                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"]
+    api_key = os.environ["GROQ_API_KEY"]
     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}"
     payload = {
         "model": "llama3-8b-8192",
         "messages": [
@@ -87,17 +91,19 @@ def generate_answer_with_groq(question, context, retries=3, delay=2):
         "temperature": 0.5,
         "max_tokens": 300,
     }
+
     for attempt in range(retries):
         try:
-    response = requests.post(url, headers=headers, json=payload)
-    result = response.json()
-    return result['choices'][0]['message']['content'].strip()
-    except GroqError as e:
-    if "503" in str(e) and attempt < retries - 1:
-        time.sleep(delay)
+            response = requests.post(url, headers=headers, json=payload)
+            result = response.json()
+            return result['choices'][0]['message']['content'].strip()
+        except Exception as e:
+            if "503" in str(e) and attempt < retries - 1:
+                time.sleep(delay)
                 continue
-        else:
-        return f"⚠️ Groq API Error: {str(e)}"
+            else:
+                return f"⚠️ Groq API Error: {str(e)}"
+
 #-----------------------------------------
 
 # --- Twilio Chat Handlers ---