Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, jsonify, send_file, stream_template
+from flask import Flask, render_template, request, jsonify, send_file
 import ollama
 import json
 import re
@@ -10,7 +10,7 @@ import urllib3
 import pandas as pd
 import io
 import ast
-
+from groq import Groq
 
 app = Flask(__name__)
 
@@ -62,18 +62,25 @@ No details or explainations are required, just the results in the required **JSO
 """
 
 
-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-8b', 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
@@ -81,7 +88,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 {
@@ -174,13 +181,9 @@ def analyze_pdf_novelty(patent_background, url, data_type="pdf"):
     except Exception as e:
         return {"error": f"Error: {str(e)}"}
 
-# @app.route('/')
-# def home():
-#     return render_template('index.html')
-
-@app.get("/")
-def timeline():
-    return stream_template("index.html")
+@app.route('/')
+def home():
+    return render_template('index.html')
 
 @app.route('/chat', methods=['POST'])
 def chat():