Update app.py

app.py

@@ -1,6 +1,6 @@
 from flask import Flask, render_template, request, jsonify
 import sqlite3
-import requests
+import ollama
 
 # Kết nối SQL
 conn = sqlite3.connect('chat_messages.db', check_same_thread=False)
@@ -31,11 +31,9 @@ def getAnswer(role, text):
         if not any(msg["role"] == "system" for msg in previous_messages):
             previous_messages.insert(0, {"role": "system", "content": system_prompt})
 
-        # Gửi request đến Ollama API
-        payload = {"model": "llama3", "prompt": text, "stream": False}
-        response = requests.post("http://localhost:11434/api/generate", json=payload)
-        response_data = response.json()
-        bot_response = response_data.get("response", "Error: No response from AI")
+        # Gửi request đến Ollama API bằng thư viện ollama
+        response = ollama.chat(model="llama3", messages=[{"role": "user", "content": text}])
+        bot_response = response.get("message", {}).get("content", "Lỗi: Không nhận được phản hồi từ AI")
 
         # Lưu phản hồi của bot vào database
         c.execute("INSERT INTO messages VALUES (?, ?)", ("assistant", bot_response))