Initial commit: 个性化饮食推荐助手 - 包含OCR识别、AI分析、现代化界面等功能
This commit is contained in:
赵杰
249
main.py
Normal file
249
main.py
Normal file
@@ -0,0 +1,249 @@
|
||||
"""
|
||||
主应用入口 - 个性化饮食推荐助手
|
||||
基于基座架构的完整应用
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
import logging
|
||||
|
||||
# 添加项目根目录到Python路径
|
||||
project_root = Path(__file__).parent
|
||||
sys.path.insert(0, str(project_root))
|
||||
|
||||
from core.base import BaseConfig, AppCore, ModuleManager, ModuleType, initialize_app, cleanup_app
|
||||
from modules.data_collection import DataCollectionModule
|
||||
from modules.ai_analysis import AIAnalysisModule
|
||||
from modules.recommendation_engine import RecommendationEngine
|
||||
from modules.ocr_calorie_recognition import OCRCalorieRecognitionModule
|
||||
from gui.main_window import MainWindow
|
||||
import tkinter as tk
|
||||
from tkinter import messagebox
|
||||
|
||||
# 配置日志
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
handlers=[
|
||||
logging.FileHandler('logs/app.log', encoding='utf-8'),
|
||||
logging.StreamHandler()
|
||||
]
|
||||
)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class DietRecommendationApp:
|
||||
"""饮食推荐应用主类"""
|
||||
|
||||
def __init__(self):
|
||||
self.config = self._load_config()
|
||||
self.app_core: Optional[AppCore] = None
|
||||
self.main_window: Optional[MainWindow] = None
|
||||
self.is_running = False
|
||||
|
||||
def _load_config(self) -> BaseConfig:
|
||||
"""加载配置"""
|
||||
config = BaseConfig()
|
||||
|
||||
# 从环境变量加载API密钥
|
||||
config.qwen_api_key = os.getenv('QWEN_API_KEY')
|
||||
|
||||
# 从.env文件加载配置
|
||||
env_file = Path('.env')
|
||||
if env_file.exists():
|
||||
try:
|
||||
from dotenv import load_dotenv
|
||||
load_dotenv()
|
||||
config.qwen_api_key = os.getenv('QWEN_API_KEY')
|
||||
except Exception:
|
||||
# 如果.env文件有编码问题，跳过加载
|
||||
pass
|
||||
|
||||
return config
|
||||
|
||||
def initialize(self) -> bool:
|
||||
"""初始化应用"""
|
||||
try:
|
||||
logger.info("正在初始化饮食推荐应用...")
|
||||
|
||||
# 创建必要的目录
|
||||
self._create_directories()
|
||||
|
||||
# 初始化应用核心
|
||||
if not initialize_app(self.config):
|
||||
logger.error("应用核心初始化失败")
|
||||
return False
|
||||
|
||||
self.app_core = AppCore(self.config)
|
||||
|
||||
# 注册所有模块
|
||||
if not self._register_modules():
|
||||
logger.error("模块注册失败")
|
||||
return False
|
||||
|
||||
# 启动应用核心
|
||||
if not self.app_core.start():
|
||||
logger.error("应用核心启动失败")
|
||||
return False
|
||||
|
||||
self.is_running = True
|
||||
logger.info("饮食推荐应用初始化成功")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"应用初始化失败: {e}")
|
||||
return False
|
||||
|
||||
def _create_directories(self):
|
||||
"""创建必要的目录"""
|
||||
directories = [
|
||||
'data',
|
||||
'data/users',
|
||||
'data/training',
|
||||
'models',
|
||||
'logs',
|
||||
'gui'
|
||||
]
|
||||
|
||||
for directory in directories:
|
||||
Path(directory).mkdir(parents=True, exist_ok=True)
|
||||
|
||||
def _register_modules(self) -> bool:
|
||||
"""注册所有模块"""
|
||||
try:
|
||||
module_manager = ModuleManager(self.config)
|
||||
|
||||
# 注册数据采集模块
|
||||
data_collection_module = DataCollectionModule(self.config)
|
||||
if not module_manager.register_module(data_collection_module):
|
||||
logger.error("数据采集模块注册失败")
|
||||
return False
|
||||
|
||||
# 注册AI分析模块
|
||||
ai_analysis_module = AIAnalysisModule(self.config)
|
||||
if not module_manager.register_module(ai_analysis_module):
|
||||
logger.error("AI分析模块注册失败")
|
||||
return False
|
||||
|
||||
# 注册推荐引擎模块
|
||||
recommendation_module = RecommendationEngine(self.config)
|
||||
if not module_manager.register_module(recommendation_module):
|
||||
logger.error("推荐引擎模块注册失败")
|
||||
return False
|
||||
|
||||
# 注册OCR热量识别模块
|
||||
ocr_module = OCRCalorieRecognitionModule(self.config)
|
||||
if not module_manager.register_module(ocr_module):
|
||||
logger.error("OCR热量识别模块注册失败")
|
||||
return False
|
||||
|
||||
# 初始化模块管理器
|
||||
if not module_manager.initialize_all():
|
||||
logger.error("模块管理器初始化失败")
|
||||
return False
|
||||
|
||||
# 将模块管理器设置到应用核心
|
||||
self.app_core.module_manager = module_manager
|
||||
logger.info("所有模块注册成功")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"模块注册失败: {e}")
|
||||
return False
|
||||
|
||||
def run_gui(self):
|
||||
"""运行GUI界面"""
|
||||
try:
|
||||
logger.info("启动移动端GUI界面...")
|
||||
|
||||
# 使用移动端界面，传递应用核心
|
||||
from gui.mobile_main_window import MobileMainWindow
|
||||
self.main_window = MobileMainWindow(self.app_core)
|
||||
|
||||
# 启动GUI主循环
|
||||
self.main_window.run()
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"GUI启动失败: {e}")
|
||||
messagebox.showerror("错误", f"GUI启动失败: {str(e)}")
|
||||
|
||||
def _on_closing(self):
|
||||
"""窗口关闭事件处理"""
|
||||
try:
|
||||
if messagebox.askokcancel("退出", "确定要退出应用吗？"):
|
||||
self.shutdown()
|
||||
except Exception as e:
|
||||
logger.error(f"关闭应用失败: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
def shutdown(self):
|
||||
"""关闭应用"""
|
||||
try:
|
||||
logger.info("正在关闭应用...")
|
||||
|
||||
# 关闭GUI
|
||||
if self.main_window:
|
||||
self.main_window.destroy()
|
||||
|
||||
# 关闭应用核心
|
||||
if self.app_core:
|
||||
self.app_core.stop()
|
||||
|
||||
# 清理资源
|
||||
cleanup_app()
|
||||
|
||||
self.is_running = False
|
||||
logger.info("应用关闭完成")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"应用关闭失败: {e}")
|
||||
finally:
|
||||
sys.exit(0)
|
||||
|
||||
def get_app_status(self) -> dict:
|
||||
"""获取应用状态"""
|
||||
if not self.app_core:
|
||||
return {"status": "not_initialized"}
|
||||
|
||||
module_status = self.app_core.module_manager.get_module_status()
|
||||
|
||||
return {
|
||||
"status": "running" if self.is_running else "stopped",
|
||||
"modules": module_status,
|
||||
"config": {
|
||||
"app_name": self.config.app_name,
|
||||
"version": self.config.version,
|
||||
"debug": self.config.debug
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
try:
|
||||
# 创建应用实例
|
||||
app = DietRecommendationApp()
|
||||
|
||||
# 初始化应用
|
||||
if not app.initialize():
|
||||
logger.error("应用初始化失败，退出")
|
||||
sys.exit(1)
|
||||
|
||||
# 运行GUI
|
||||
app.run_gui()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
logger.info("用户中断，正在退出...")
|
||||
sys.exit(0)
|
||||
except Exception as e:
|
||||
logger.error(f"应用运行失败: {e}")
|
||||
import traceback
|
||||
logger.error(f"详细错误信息: {traceback.format_exc()}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user