Update app.py

app.py

@@ -2,62 +2,76 @@ from flask import Flask, render_template, request, jsonify
 import sqlite3
 from openai import OpenAI
 
-app = Flask(__name__)
-
-# Prompt hệ thống
-system_prompt = "You are Oshi no Ko"
+# Hàm tạo database nếu chưa tồn tại
+def init_db():
+    with sqlite3.connect('chat_messages.db') as conn:
+        c = conn.cursor()
+        c.execute('''CREATE TABLE IF NOT EXISTS messages 
+                     (role TEXT, content TEXT)''')
+        conn.commit()
 
+# Hàm lấy kết nối database
 def get_db_connection():
-    """Tạo một kết nối SQLite mới cho mỗi request."""
     conn = sqlite3.connect('chat_messages.db', check_same_thread=False)
-    conn.row_factory = sqlite3.Row
     return conn
 
+# Prompt hệ thống
+system_prompt = "You are name Oshi no Ko"
+
 def getAnswer(role, text):
     try:
         conn = get_db_connection()
         c = conn.cursor()
-
-        # Thêm tin nhắn vào database
+        
+        # Lưu tin nhắn vào database
         c.execute("INSERT INTO messages VALUES (?, ?)", (role, text))
         conn.commit()
-
+        
         # Lấy 5 tin nhắn gần nhất
         c.execute("SELECT * FROM messages ORDER BY rowid DESC LIMIT 5")
         previous_messages = [{"role": row[0], "content": row[1]} for row in c.fetchall()]
-
+        conn.close()
+        
         # Đảo ngược thứ tự tin nhắn
         previous_messages.reverse()
-
+        
         # Kiểm tra nếu chưa có system prompt thì thêm vào
         if not any(msg["role"] == "system" for msg in previous_messages):
             previous_messages.insert(0, {"role": "system", "content": system_prompt})
-
-        # Kết nối API OpenAI (hoặc Ollama)
-        client = OpenAI(base_url="http://localhost:8000/v1", api_key="ollama")  # Kiểm tra lại URL của bạn
+        
+        # Kết nối API OpenAI
+        client = OpenAI(base_url="http://localhost:8000/v1", api_key="ollama")
         response = client.chat.completions.create(
             model="llama3",
             messages=previous_messages,
             temperature=0.7,
         )
-
+        
         bot_response = response.choices[0].message.content.strip()
-
+        
         # Lưu phản hồi của bot vào database
+        conn = get_db_connection()
+        c = conn.cursor()
         c.execute("INSERT INTO messages VALUES (?, ?)", ("assistant", bot_response))
         conn.commit()
         conn.close()
-
+        
         return bot_response
     except Exception as e:
         return f"Error: {str(e)}"
 
+# Khởi tạo Flask app
+app = Flask(__name__)
+
+# Tạo database khi ứng dụng khởi động
+init_db()
+
 @app.route('/')
 def index():
     return render_template('index.html')
 
 @app.route('/echo', methods=['POST'])
-def echo_message():
+def echo():
     data = request.json
     message = data.get('message', '')
     return jsonify({'FROM': 'Echobot', 'MESSAGE': message})
@@ -79,4 +93,4 @@ def history():
     return jsonify(previous_messages)
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host='0.0.0.0', port=7860, debug=True)