250 lines
7.7 KiB
Python
250 lines
7.7 KiB
Python
"""
|
||
主应用入口 - 个性化饮食推荐助手
|
||
基于基座架构的完整应用
|
||
"""
|
||
|
||
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()
|