当前位置：首页 > news >正文

利用预测大模型完成办公室饮水机剩余热水量

news 来源：原创 2024/7/1 5:31:14

背景

在每天上班的时候，很多同事都有喝热水的习惯，但是饮水机内的热水量总是比较少的，如何避免等待，高效的接到热水是我接下来要做的事情的动机。

理论基础

在大量真实数据的情况下，可以分析出用水紧张的时间段和用水不紧张的时间段。在用水不紧张的时间去接热水，有热水的可能性比较大。

实现思路与理论基础

每天倒水的时候记录下，当前时间对应的剩余热水量
把历史数据提交给大模型，通过提示词让大模型来总结并预测结果

依赖服务

大模型服务

实现效果

记录数据页面
预测结果页面

实现代码

后台代码

from flask import Flask, render_template, request, redirect, url_for, jsonify
import sqlite3
from datetime import datetime
import requests
app = Flask(__name__)# 连接到 SQLite 数据库
def connect_db():conn = sqlite3.connect('test_data.db')return conn# 创建数据表
def create_table():conn = connect_db()cursor = conn.cursor()cursor.execute('''CREATE TABLE IF NOT EXISTS test_data (id INTEGER PRIMARY KEY,test_date TEXT,test_datetime TEXT,water_level INTEGER,created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP)''')conn.commit()conn.close()# 插入测试数据
def insert_data(test_date, test_datetime, water_level):conn = connect_db()cursor = conn.cursor()cursor.execute('''INSERT INTO test_data (test_date, test_datetime, water_level)VALUES (?, ?, ?)''', (test_date, test_datetime, water_level))conn.commit()conn.close()# 提交测试数据
@app.route('/submit', methods=['POST'])
def submit():test_date = request.form['test_date']test_datetime = f"{test_date} {request.form['test_time']}"water_level = request.form['water_level']insert_data(test_date, test_datetime, water_level)return redirect(url_for('index'))# 删除测试数据
@app.route('/delete/<int:id>', methods=['POST'])
def delete(id):delete_data(id)return redirect(url_for('index'))# 删除测试数据
def delete_data(id):conn = connect_db()cursor = conn.cursor()cursor.execute('''DELETE FROM test_data WHERE id = ?''', (id,))conn.commit()conn.close()# 获取测试数据并按创建时间降序排序
def get_data_by_date_range(start_date, end_date, page, per_page):conn = connect_db()cursor = conn.cursor()offset = (page - 1) * per_pagecursor.execute('''SELECT * FROM test_data WHERE test_date BETWEEN ? AND ? ORDER BY created_at DESC LIMIT ? OFFSET ?''',(start_date, end_date, per_page, offset))data = cursor.fetchall()cursor.execute('''SELECT COUNT(*) FROM test_data WHERE test_date BETWEEN ? AND ?''', (start_date, end_date))total = cursor.fetchone()[0]conn.close()return data, total# 获取最新的100条test_datetime和water_level数据
def get_latest_100_data():conn = connect_db()cursor = conn.cursor()cursor.execute('''SELECT test_datetime, water_level FROM test_data ORDER BY created_at DESC LIMIT 100''')data = cursor.fetchall()conn.close()return data# 首页，显示所有测试数据
@app.route('/')
def index():today = datetime.today().strftime('%Y-%m-%d')current_time = datetime.now().strftime('%H:%M')start_date = request.args.get('start_date', today)end_date = request.args.get('end_date', today)page = int(request.args.get('page', 1))per_page = 5  # 每页显示5条记录data, total = get_data_by_date_range(start_date, end_date, page, per_page)total_pages = (total + per_page - 1) // per_page# 增加序号字段data_with_index = [(i + 1 + (page - 1) * per_page, *row) for i, row in enumerate(data)]return render_template('index.html', title="饮水机数据记录", data=data_with_index, today=today, current_time=current_time, start_date=start_date, end_date=end_date, page=page, total_pages=total_pages)# 数据预测页面
@app.route('/predict')
def predict():return render_template('predict.html', title="猜一猜")# 数据预测结果处理
@app.route('/predict_result', methods=['POST'])
def predict_result():# 获取最新的100条数据作为提示词latest_data = get_latest_100_data()prompt = "以下是最近期办公室饮水机时间和剩余热水水量的数据：---\n "for entry in latest_data:prompt += f"时间: {entry[0]}, 热水水量: {entry[1]}%\n"prompt += " 请总结规律并帮我预测时间是："+datetime.now().strftime("%Y-%m-%d %H:%M") \+ "的热水水量是多少？你需要严格按照 '预测结果是：当前水量是XXX' 的格式回答，千万不要做任何解释"# 定义要发送的 JSON 数据data = {"model": "llama3-chinese:8b","messages": [{"role": "system","content": "你是一个高冷的规律总结大师，可以根据提供的数据进行预测结果,不需要做成任何解释"},{"role": "user","content": prompt}],"options": {"temperature": 0.5},"stream": False}# 定义目标 URLurl = "http://192.168.10.70:11434/api/chat"print("prompt:"+prompt)# 发送 POST 请求，并将 JSON 数据作为请求体发送response = requests.post(url, json=data)return render_template('predict.html', title="猜一猜", prediction=response.json()['message']['content'])if __name__ == '__main__':create_table()app.run(debug=True)

前台代码

<!DOCTYPE html>
<html lang="zh-CN">
<head><meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><title>{{ title }}</title><link href="https://stackpath.bootstrapcdn.com/bootstrap/4.5.2/css/bootstrap.min.css" rel="stylesheet"><style>.form-section, .list-section {padding: 20px;border-radius: 8px;}.form-section {background-color: #f8f9fa;border: 1px solid #e9ecef;}.list-section {background-color: #ffffff;border: 1px solid #dee2e6;}.navbar {margin-bottom: 20px;}</style>
</head>
<body><nav class="navbar navbar-expand-lg navbar-light bg-light"><a class="navbar-brand" href="/">饮水机数据记录</a><button class="navbar-toggler" type="button" data-toggle="collapse" data-target="#navbarNav" aria-controls="navbarNav" aria-expanded="false" aria-label="Toggle navigation"><span class="navbar-toggler-icon"></span></button><div class="collapse navbar-collapse" id="navbarNav"><ul class="navbar-nav"><li class="nav-item"><a class="nav-link" href="/">首页</a></li><li class="nav-item"><a class="nav-link" href="/predict">猜一猜</a></li></ul></div></nav><div class="container mt-5"><div class="row"><div class="col-md-4"><div class="form-section"><h2 class="mb-4">记录数据</h2><form action="/submit" method="post"><div class="form-group"><label for="test_date">日期:</label><input type="date" id="test_date" name="test_date" class="form-control" value="{{ today }}" required></div><div class="form-group"><label for="test_time">时间:</label><input type="time" id="test_time" name="test_time" class="form-control" value="{{ current_time }}" required></div><div class="form-group"><label for="water_level">储水量百分比:</label><select id="water_level" name="water_level" class="form-control" required><option value="0">0%</option><option value="20">20%</option><option value="40">40%</option><option value="60">60%</option><option value="80">80%</option><option value="100">100%</option></select></div><button type="submit" class="btn btn-primary btn-block">提交</button></form></div></div><div class="col-md-8"><div class="list-section"><h2 class="mb-4">已记录数据</h2><form method="get" action="/" class="mb-4"><div class="form-row"><div class="col"><label for="start_date">开始日期:</label><input type="date" id="start_date" name="start_date" class="form-control" value="{{ start_date }}" required></div><div class="col"><label for="end_date">结束日期:</label><input type="date" id="end_date" name="end_date" class="form-control" value="{{ end_date }}" required></div><div class="col align-self-end"><button type="submit" class="btn btn-secondary btn-block">筛选</button></div></div></form><table class="table table-bordered"><thead class="thead-light"><tr><th>序号</th><th>日期</th><th>时间</th><th>储水量百分比</th><th>操作</th></tr></thead><tbody>{% for row in data %}<tr><td>{{ row[0] }}</td> <!-- 这里使用序号 --><td>{{ row[2] }}</td><td>{{ row[3] }}</td><td>{{ row[4] }}%</td><td><form action="/delete/{{ row[1] }}" method="post"><button type="submit" class="btn btn-danger btn-sm">删除</button></form></td></tr>{% endfor %}</tbody></table><nav aria-label="Page navigation"><ul class="pagination justify-content-center">{% if page > 1 %}<li class="page-item"><a class="page-link" href="/?start_date={{ start_date }}&end_date={{ end_date }}&page={{ page - 1 }}">上一页</a></li>{% endif %}{% if page < total_pages %}<li class="page-item"><a class="page-link" href="/?start_date={{ start_date }}&end_date={{ end_date }}&page={{ page + 1 }}">下一页</a></li>{% endif %}</ul></nav></div></div></div></div><script src="{{ url_for('static', filename='js/jquery-3.7.1.min.js') }}"></script><script src="{{ url_for('static', filename='js/bootstrap.js') }}"></script>
</body>
</html>