Update app.py

app.py

@@ -15,13 +15,13 @@ if not os.getenv("GROQ_API_KEY"):
 vectorizer = TfidfVectorizer(stop_words='english')
 
 # === UTILITY FUNCTIONS ===
-def call_groq_api(prompt):
+""" def call_groq_api(prompt):
     api_key = os.getenv("GROQ_API_KEY")
     if not api_key:
         return "Error: GROQ_API_KEY environment variable not set."
 
     headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
-    data = {"model": "mixtral-8x7b-32768", "messages": [{"role": "user", "content": prompt}]}
+    data = {"model": "llama-3.1-70b-versatile", "messages": [{"role": "user", "content": prompt}]}
 
     try:
         response = requests.post("https://api.groq.com/openai/v1/chat/completions", json=data, headers=headers)
@@ -33,6 +33,35 @@ def call_groq_api(prompt):
         return f"API Error: {str(e)}"
     except (KeyError, IndexError) as e:
         return f"Error parsing API response: {str(e)}"
+"""
+
+def call_groq_api(prompt):
+    api_key = os.getenv("GROQ_API_KEY")
+    if not api_key:
+        return "Error: GROQ_API_KEY environment variable not set."
+
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+
+    data = {
+        "model": "llama-3.1-70b-versatile",  # ✅ updated model
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": 0.7
+    }
+
+    try:
+        response = requests.post("https://api.groq.com/openai/v1/chat/completions", json=data, headers=headers)
+        if response.status_code != 200:
+            return f"API Error {response.status_code}: {response.text}"
+        result = response.json()
+        return result["choices"][0]["message"]["content"]
+    except requests.exceptions.RequestException as e:
+        return f"Network Error: {e}"
+    except Exception as e:
+        return f"Unexpected Error: {e}"
+
 
 def extract_text_from_pdfs(pdf_files):
     chunks, pages, file_names = [], [], []