Update app.py

app.py

@@ -1,72 +1,74 @@
-from flash import Flask, render_template, request, jsonify
-import sqlite3
-from openai import OpenAI
-#SQL
-
-conn = sqlite3.connect('chat_messages.db', check_same_thread=False)
-c = conn.cursor()
-
-#creat the table
-c.execute('''CREATE TABLE IF NOT EXISTS messages 
-          (role TEXT, content TEXT)''')
-
-conn.commit()
-
-stystem_prompt ="You are name oshinoko"
-def getAnswer(role, text):
-    #INSERT
-    c.execute("INSERT INTO messages VALUES (?, ?)", (role, text))
-    conn.commit()
-    #GET
-    c.execute("SELECT * FROM messages order by rowid DESC LIMIT 5")
-    previous_messages = [{"role": row[0], "content": row[1]} for row in c.fetchall()]
-
-    #REVERSE
-    previous_messages = list(reversed(previous_messages))
-    #ADD SYSTEM
-    if "system" not in [x[role] for x in previous_messages]:
-        previous_messages = [{"role": "system", "content": stystem_prompt}] + previous_messages
-    client = OpenAI(base_url="https://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()
-
-    c.execute("INSERT INTO messages VALUES (?, ?)", ("assistant", bot_response))
-    conn.commit()
-
-    return bot_response
-
-
-app = Flask(__name__)
-
-
-@app.route('/')
-def index():
-    return render_template('index.html')
-
-@app.route('/echo', methods=['POST'])
-def echo():
-    data = request.json
-    message = data['message']
-    return jsonify({'FROM': 'Echobot', 'MESSAGE': message})
-
-
-@app.route('/chat', methods=['POST'])
-def echo():
-    data = request.json
-    message = getAnswer("User",data['message'])
-    return jsonify({'FROM': 'ohsinoko', 'MESSAGE': message})
-#history
-@app.route('/history', methods=['GET'])
-def history():
-    c.execute("SELECT * FROM messages order by rowid")
-    previous_messages = [{"role": row[0], "content": row[1]} for row in c.fetchall()]
-    return jsonify(previous_messages)
-
-  
-if __name__=='__main__':
-    app.run(debug=True)
+from flask import Flask, render_template, request, jsonify
+import sqlite3
+from openai import OpenAI
+
+# Kết nối SQLite
+db_path = "chat_messages.db"
+conn = sqlite3.connect(db_path, check_same_thread=False)
+c = conn.cursor()
+
+# Tạo bảng nếu chưa tồn tại
+c.execute('''CREATE TABLE IF NOT EXISTS messages 
+          (role TEXT, content TEXT)''')
+conn.commit()
+
+# Lời nhắc hệ thống
+system_prompt = "You are name oshinoko"
+
+def getAnswer(role, text):
+    # 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()]
+    previous_messages.reverse()
+    
+    # Thêm system prompt nếu chưa có
+    if not any(msg["role"] == "system" for msg in previous_messages):
+        previous_messages.insert(0, {"role": "system", "content": system_prompt})
+    
+    # Gọi API OpenAI
+    client = OpenAI(base_url="https://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
+    c.execute("INSERT INTO messages VALUES (?, ?)", ("assistant", bot_response))
+    conn.commit()
+    
+    return bot_response
+
+# Khởi tạo Flask
+app = Flask(__name__)
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/echo', methods=['POST'])
+def echo():
+    data = request.json
+    message = data['message']
+    return jsonify({'FROM': 'Echobot', 'MESSAGE': message})
+
+@app.route('/chat', methods=['POST'])
+def chat():
+    data = request.json
+    message = getAnswer("User", data['message'])
+    return jsonify({'FROM': 'oshinoko', 'MESSAGE': message})
+
+@app.route('/history', methods=['GET'])
+def history():
+    c.execute("SELECT * FROM messages ORDER BY rowid")
+    previous_messages = [{"role": row[0], "content": row[1]} for row in c.fetchall()]
+    return jsonify(previous_messages)
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True)