zhaojie/recommend
1
0
Fork 0
Code Issues Pull Requests 2 Actions Packages Projects Releases Wiki Activity

Initial commit: 个性化饮食推荐助手 - 包含OCR识别、AI分析、现代化界面等功能

Browse Source
This commit is contained in:
赵杰
2025-09-25 14:20:11 +01:00
commit aea5f6bf74
27 changed files with 14015 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

526
gui/quick_user_input.py Normal file
View File

@@ -0,0 +1,526 @@
"""
快速用户需求录入界面
优化用户录入流程，提高效率
"""
import tkinter as tk
from tkinter import ttk, messagebox
import customtkinter as ctk
from typing import Dict, List, Optional
from datetime import datetime
import json
class QuickUserInputDialog:
"""快速用户需求录入对话框"""
def __init__(self, parent, user_id: str):
self.parent = parent
self.user_id = user_id
self.input_data = {}
# 创建对话框
self.dialog = ctk.CTkToplevel(parent)
self.dialog.title("快速需求录入")
self.dialog.geometry("800x600")
self.dialog.transient(parent)
self.dialog.grab_set()
# 居中显示
self.dialog.geometry("+%d+%d" % (parent.winfo_rootx() + 50, parent.winfo_rooty() + 50))
self._create_widgets()
def _create_widgets(self):
"""创建界面组件"""
# 主框架
main_frame = ctk.CTkScrollableFrame(self.dialog)
main_frame.pack(fill="both", expand=True, padx=20, pady=20)
# 标题
title_label = ctk.CTkLabel(
main_frame,
text="🚀 快速需求录入",
font=ctk.CTkFont(size=24, weight="bold")
)
title_label.pack(pady=20)
# 步骤1：基础信息
self._create_basic_info_section(main_frame)
# 步骤2：饮食偏好
self._create_preferences_section(main_frame)
# 步骤3：健康目标
self._create_health_goals_section(main_frame)
# 步骤4：快速餐食记录
self._create_quick_meal_section(main_frame)
# 按钮区域
self._create_buttons(main_frame)
def _create_basic_info_section(self, parent):
"""创建基础信息区域"""
section_frame = ctk.CTkFrame(parent)
section_frame.pack(fill="x", padx=10, pady=10)
# 标题
title = ctk.CTkLabel(
section_frame,
text="1️⃣ 基础信息",
font=ctk.CTkFont(size=18, weight="bold")
)
title.pack(pady=10)
# 信息网格
info_frame = ctk.CTkFrame(section_frame)
info_frame.pack(fill="x", padx=20, pady=10)
# 姓名
name_label = ctk.CTkLabel(info_frame, text="姓名:")
name_label.grid(row=0, column=0, sticky="w", padx=10, pady=5)
self.name_var = tk.StringVar()
name_entry = ctk.CTkEntry(info_frame, textvariable=self.name_var, width=200)
name_entry.grid(row=0, column=1, sticky="w", padx=10, pady=5)
# 年龄范围
age_label = ctk.CTkLabel(info_frame, text="年龄范围:")
age_label.grid(row=0, column=2, sticky="w", padx=10, pady=5)
self.age_var = tk.StringVar(value="25-30岁")
age_menu = ctk.CTkOptionMenu(
info_frame,
variable=self.age_var,
values=["18-24岁", "25-30岁", "31-35岁", "36-40岁", "41-45岁", "46-50岁", "51-55岁", "56-60岁", "60岁以上"]
)
age_menu.grid(row=0, column=3, sticky="w", padx=10, pady=5)
# 性别
gender_label = ctk.CTkLabel(info_frame, text="性别:")
gender_label.grid(row=1, column=0, sticky="w", padx=10, pady=5)
self.gender_var = tk.StringVar(value="女")
gender_menu = ctk.CTkOptionMenu(
info_frame,
variable=self.gender_var,
values=["男", "女"]
)
gender_menu.grid(row=1, column=1, sticky="w", padx=10, pady=5)
# 身高体重范围
height_label = ctk.CTkLabel(info_frame, text="身高范围:")
height_label.grid(row=1, column=2, sticky="w", padx=10, pady=5)
self.height_var = tk.StringVar(value="160-165cm")
height_menu = ctk.CTkOptionMenu(
info_frame,
variable=self.height_var,
values=["150cm以下", "150-155cm", "155-160cm", "160-165cm", "165-170cm", "170-175cm", "175-180cm", "180cm以上"]
)
height_menu.grid(row=1, column=3, sticky="w", padx=10, pady=5)
weight_label = ctk.CTkLabel(info_frame, text="体重范围:")
weight_label.grid(row=2, column=0, sticky="w", padx=10, pady=5)
self.weight_var = tk.StringVar(value="50-55kg")
weight_menu = ctk.CTkOptionMenu(
info_frame,
variable=self.weight_var,
values=["40kg以下", "40-45kg", "45-50kg", "50-55kg", "55-60kg", "60-65kg", "65-70kg", "70-75kg", "75-80kg", "80kg以上"]
)
weight_menu.grid(row=2, column=1, sticky="w", padx=10, pady=5)
# 活动水平
activity_label = ctk.CTkLabel(info_frame, text="活动水平:")
activity_label.grid(row=2, column=2, sticky="w", padx=10, pady=5)
self.activity_var = tk.StringVar(value="中等")
activity_menu = ctk.CTkOptionMenu(
info_frame,
variable=self.activity_var,
values=["久坐", "轻度活动", "中等", "高度活动", "极度活动"]
)
activity_menu.grid(row=2, column=3, sticky="w", padx=10, pady=5)
def _create_preferences_section(self, parent):
"""创建饮食偏好区域"""
section_frame = ctk.CTkFrame(parent)
section_frame.pack(fill="x", padx=10, pady=10)
# 标题
title = ctk.CTkLabel(
section_frame,
text="2️⃣ 饮食偏好",
font=ctk.CTkFont(size=18, weight="bold")
)
title.pack(pady=10)
# 偏好网格
pref_frame = ctk.CTkFrame(section_frame)
pref_frame.pack(fill="x", padx=20, pady=10)
# 口味偏好
taste_label = ctk.CTkLabel(pref_frame, text="主要口味偏好:")
taste_label.grid(row=0, column=0, sticky="w", padx=10, pady=5)
self.taste_var = tk.StringVar(value="均衡")
taste_menu = ctk.CTkOptionMenu(
pref_frame,
variable=self.taste_var,
values=["均衡", "偏甜", "偏咸", "偏辣", "偏酸", "偏清淡"]
)
taste_menu.grid(row=0, column=1, sticky="w", padx=10, pady=5)
# 饮食类型
diet_label = ctk.CTkLabel(pref_frame, text="饮食类型:")
diet_label.grid(row=0, column=2, sticky="w", padx=10, pady=5)
self.diet_var = tk.StringVar(value="普通饮食")
diet_menu = ctk.CTkOptionMenu(
pref_frame,
variable=self.diet_var,
values=["普通饮食", "素食", "低脂饮食", "低糖饮食", "高蛋白饮食", "地中海饮食"]
)
diet_menu.grid(row=0, column=3, sticky="w", padx=10, pady=5)
# 过敏食物
allergy_label = ctk.CTkLabel(pref_frame, text="过敏食物:")
allergy_label.grid(row=1, column=0, sticky="w", padx=10, pady=5)
self.allergy_var = tk.StringVar(value="无")
allergy_menu = ctk.CTkOptionMenu(
pref_frame,
variable=self.allergy_var,
values=["无", "花生", "海鲜", "牛奶", "鸡蛋", "坚果", "大豆", "小麦", "其他"]
)
allergy_menu.grid(row=1, column=1, sticky="w", padx=10, pady=5)
# 不喜欢食物
dislike_label = ctk.CTkLabel(pref_frame, text="不喜欢食物:")
dislike_label.grid(row=1, column=2, sticky="w", padx=10, pady=5)
self.dislike_var = tk.StringVar(value="无")
dislike_menu = ctk.CTkOptionMenu(
pref_frame,
variable=self.dislike_var,
values=["无", "内脏", "辛辣", "油腻", "甜食", "酸味", "苦味", "其他"]
)
dislike_menu.grid(row=1, column=3, sticky="w", padx=10, pady=5)
def _create_health_goals_section(self, parent):
"""创建健康目标区域"""
section_frame = ctk.CTkFrame(parent)
section_frame.pack(fill="x", padx=10, pady=10)
# 标题
title = ctk.CTkLabel(
section_frame,
text="3️⃣ 健康目标",
font=ctk.CTkFont(size=18, weight="bold")
)
title.pack(pady=10)
# 目标选择
goals_frame = ctk.CTkFrame(section_frame)
goals_frame.pack(fill="x", padx=20, pady=10)
# 主要目标
main_goal_label = ctk.CTkLabel(goals_frame, text="主要目标:")
main_goal_label.grid(row=0, column=0, sticky="w", padx=10, pady=5)
self.main_goal_var = tk.StringVar(value="保持健康")
main_goal_menu = ctk.CTkOptionMenu(
goals_frame,
variable=self.main_goal_var,
values=["保持健康", "减重", "增重", "增肌", "改善消化", "提高免疫力", "控制血糖", "降低血压"]
)
main_goal_menu.grid(row=0, column=1, sticky="w", padx=10, pady=5)
# 次要目标
sub_goal_label = ctk.CTkLabel(goals_frame, text="次要目标:")
sub_goal_label.grid(row=0, column=2, sticky="w", padx=10, pady=5)
self.sub_goal_var = tk.StringVar(value="无")
sub_goal_menu = ctk.CTkOptionMenu(
goals_frame,
variable=self.sub_goal_var,
values=["无", "改善睡眠", "提高精力", "美容养颜", "延缓衰老", "改善皮肤", "增强记忆", "缓解压力"]
)
sub_goal_menu.grid(row=0, column=3, sticky="w", padx=10, pady=5)
def _create_quick_meal_section(self, parent):
"""创建快速餐食记录区域"""
section_frame = ctk.CTkFrame(parent)
section_frame.pack(fill="x", padx=10, pady=10)
# 标题
title = ctk.CTkLabel(
section_frame,
text="4️⃣ 快速餐食记录",
font=ctk.CTkFont(size=18, weight="bold")
)
title.pack(pady=10)
# 餐食选择
meal_frame = ctk.CTkFrame(section_frame)
meal_frame.pack(fill="x", padx=20, pady=10)
# 餐次选择
meal_type_label = ctk.CTkLabel(meal_frame, text="餐次:")
meal_type_label.grid(row=0, column=0, sticky="w", padx=10, pady=5)
self.meal_type_var = tk.StringVar(value="午餐")
meal_type_menu = ctk.CTkOptionMenu(
meal_frame,
variable=self.meal_type_var,
values=["早餐", "午餐", "晚餐", "加餐"]
)
meal_type_menu.grid(row=0, column=1, sticky="w", padx=10, pady=5)
# 快速食物选择
food_label = ctk.CTkLabel(meal_frame, text="快速选择食物:")
food_label.grid(row=1, column=0, sticky="w", padx=10, pady=5)
self.quick_food_var = tk.StringVar(value="米饭+鸡肉+蔬菜")
quick_food_menu = ctk.CTkOptionMenu(
meal_frame,
variable=self.quick_food_var,
values=[
"米饭+鸡肉+蔬菜",
"面条+鸡蛋+青菜",
"馒头+豆腐+白菜",
"粥+咸菜+鸡蛋",
"面包+牛奶+水果",
"饺子+汤",
"包子+豆浆",
"其他"
]
)
quick_food_menu.grid(row=1, column=1, sticky="w", padx=10, pady=5)
# 分量选择
portion_label = ctk.CTkLabel(meal_frame, text="分量:")
portion_label.grid(row=1, column=2, sticky="w", padx=10, pady=5)
self.portion_var = tk.StringVar(value="正常")
portion_menu = ctk.CTkOptionMenu(
meal_frame,
variable=self.portion_var,
values=["少量", "正常", "较多", "很多"]
)
portion_menu.grid(row=1, column=3, sticky="w", padx=10, pady=5)
# 满意度
satisfaction_label = ctk.CTkLabel(meal_frame, text="满意度:")
satisfaction_label.grid(row=2, column=0, sticky="w", padx=10, pady=5)
self.satisfaction_var = tk.IntVar(value=3)
satisfaction_slider = ctk.CTkSlider(
meal_frame,
from_=1,
to=5,
number_of_steps=4,
variable=self.satisfaction_var
)
satisfaction_slider.grid(row=2, column=1, columnspan=2, sticky="ew", padx=10, pady=5)
# 满意度标签
self.satisfaction_display = ctk.CTkLabel(meal_frame, text="3分 - 一般")
self.satisfaction_display.grid(row=2, column=3, sticky="w", padx=10, pady=5)
# 绑定滑块事件
satisfaction_slider.configure(command=self._on_satisfaction_changed)
def _create_buttons(self, parent):
"""创建按钮区域"""
button_frame = ctk.CTkFrame(parent)
button_frame.pack(fill="x", padx=10, pady=20)
# 保存按钮
save_button = ctk.CTkButton(
button_frame,
text="💾 保存所有信息",
command=self._save_all_data,
width=200,
height=50,
font=ctk.CTkFont(size=16, weight="bold")
)
save_button.pack(side="left", padx=20, pady=10)
# 取消按钮
cancel_button = ctk.CTkButton(
button_frame,
text="❌ 取消",
command=self._cancel,
width=150,
height=50
)
cancel_button.pack(side="right", padx=20, pady=10)
def _on_satisfaction_changed(self, value):
"""满意度改变事件"""
score = int(float(value))
score_texts = {
1: "1分 - 很不满意",
2: "2分 - 不满意",
3: "3分 - 一般",
4: "4分 - 满意",
5: "5分 - 很满意"
}
self.satisfaction_display.configure(text=score_texts.get(score, "3分 - 一般"))
def _save_all_data(self):
"""保存所有数据"""
try:
# 收集所有数据
self.input_data = {
'basic_info': {
'name': self.name_var.get(),
'age_range': self.age_var.get(),
'gender': self.gender_var.get(),
'height_range': self.height_var.get(),
'weight_range': self.weight_var.get(),
'activity_level': self.activity_var.get()
},
'preferences': {
'taste': self.taste_var.get(),
'diet_type': self.diet_var.get(),
'allergies': self.allergy_var.get(),
'dislikes': self.dislike_var.get()
},
'health_goals': {
'main_goal': self.main_goal_var.get(),
'sub_goal': self.sub_goal_var.get()
},
'quick_meal': {
'meal_type': self.meal_type_var.get(),
'food_combo': self.quick_food_var.get(),
'portion': self.portion_var.get(),
'satisfaction': self.satisfaction_var.get()
}
}
# 保存到数据库
if self._save_to_database():
messagebox.showinfo("成功", "所有信息保存成功！")
self.dialog.destroy()
else:
messagebox.showerror("错误", "保存失败，请重试")
except Exception as e:
messagebox.showerror("错误", f"保存失败: {str(e)}")
def _save_to_database(self) -> bool:
"""保存到数据库"""
try:
from modules.data_collection import collect_questionnaire_data, record_meal
# 保存基础信息
basic_data = self.input_data['basic_info']
age_mapping = {
"18-24岁": 21, "25-30岁": 27, "31-35岁": 33, "36-40岁": 38,
"41-45岁": 43, "46-50岁": 48, "51-55岁": 53, "56-60岁": 58, "60岁以上": 65
}
height_mapping = {
"150cm以下": 150, "150-155cm": 152, "155-160cm": 157, "160-165cm": 162,
"165-170cm": 167, "170-175cm": 172, "175-180cm": 177, "180cm以上": 180
}
weight_mapping = {
"40kg以下": 40, "40-45kg": 42, "45-50kg": 47, "50-55kg": 52,
"55-60kg": 57, "60-65kg": 62, "65-70kg": 67, "70-75kg": 72,
"75-80kg": 77, "80kg以上": 80
}
activity_mapping = {
"久坐": "sedentary", "轻度活动": "light", "中等": "moderate",
"高度活动": "high", "极度活动": "very_high"
}
basic_answers = {
'name': basic_data['name'],
'age': age_mapping.get(basic_data['age_range'], 25),
'gender': basic_data['gender'],
'height': height_mapping.get(basic_data['height_range'], 165),
'weight': weight_mapping.get(basic_data['weight_range'], 55),
'activity_level': activity_mapping.get(basic_data['activity_level'], 'moderate')
}
collect_questionnaire_data(self.user_id, 'basic', basic_answers)
# 保存口味偏好
preferences_data = self.input_data['preferences']
taste_answers = {
'taste_preference': preferences_data['taste'],
'diet_type': preferences_data['diet_type'],
'allergies': [preferences_data['allergies']] if preferences_data['allergies'] != "无" else [],
'dislikes': [preferences_data['dislikes']] if preferences_data['dislikes'] != "无" else []
}
collect_questionnaire_data(self.user_id, 'taste', taste_answers)
# 保存健康目标
health_data = self.input_data['health_goals']
health_answers = {
'main_goal': health_data['main_goal'],
'sub_goal': health_data['sub_goal']
}
collect_questionnaire_data(self.user_id, 'health', health_answers)
# 保存快速餐食记录
meal_data = self.input_data['quick_meal']
meal_type_mapping = {
"早餐": "breakfast", "午餐": "lunch", "晚餐": "dinner", "加餐": "snack"
}
# 解析食物组合
food_combo = meal_data['food_combo']
if "+" in food_combo:
foods = [food.strip() for food in food_combo.split("+")]
else:
foods = [food_combo]
# 估算分量
portion_mapping = {
"少量": "1小份", "正常": "1份", "较多": "1大份", "很多": "2份"
}
quantities = [portion_mapping.get(meal_data['portion'], "1份")] * len(foods)
meal_record = {
'date': datetime.now().strftime('%Y-%m-%d'),
'meal_type': meal_type_mapping.get(meal_data['meal_type'], 'lunch'),
'foods': foods,
'quantities': quantities,
'satisfaction_score': meal_data['satisfaction']
}
record_meal(self.user_id, meal_record)
return True
except Exception as e:
print(f"保存到数据库失败: {e}")
return False
def _cancel(self):
"""取消录入"""
self.dialog.destroy()
# 便捷函数
def show_quick_user_input_dialog(parent, user_id: str):
"""显示快速用户需求录入对话框"""
dialog = QuickUserInputDialog(parent, user_id)
parent.wait_window(dialog.dialog)
if __name__ == "__main__":
# 测试快速录入对话框
root = tk.Tk()
root.title("测试快速录入")
def test_dialog():
show_quick_user_input_dialog(root, "test_user")
test_button = tk.Button(root, text="测试快速录入", command=test_dialog)
test_button.pack(pady=20)
root.mainloop()