Upload app.py

app.py

@@ -0,0 +1,72 @@
+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)