Update app.py

app.py

@@ -1,5 +1,4 @@
 from flask import Flask, render_template, request, jsonify, send_file
-import ollama
 import json
 import re
 from duckduckgo_search import DDGS
@@ -112,18 +111,25 @@ Return the results in JSON format with no additional text:
 """
 
 
-def ask_ollama(user_message, model='gemma3:1b', system_prompt=search_prompt):
-    response = ollama.chat(model=model, messages=[
-        {
-            "role": "system",
-            "content": system_prompt
-        },
-        {
-            "role": "user", 
-            "content": user_message
-        }
-    ])
-    ai_reply = response['message']['content']
+def ask_ollama(user_message, model='llama-3.3-70b-versatile', system_prompt=search_prompt):
+    client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+
+    response = client.chat.completions.create(
+        model=model,
+        messages=[
+            {
+                "role": "system",
+                "content": system_prompt
+            },
+            {
+                "role": "user", 
+                "content": user_message
+            }
+        ],
+        stream=False,
+    )
+
+    ai_reply = response.choices[0].message.content
     print(f"AI REPLY json:\n{ai_reply}")
 
     # Process the response to ensure we return valid JSON
@@ -131,7 +137,7 @@ def ask_ollama(user_message, model='gemma3:1b', system_prompt=search_prompt):
         # First, try to parse it directly in case it's already valid JSON
         print(f"AI REPLY:\n{ai_reply}")
         return ast.literal_eval(ai_reply.replace('json\n', '').replace('```', ''))
-    except:
+    except Exception as e:
         print(f"ERROR:\n{e}")
         # If it's not valid JSON, try to extract JSON from the text
         return {