Update app.py

app.py

@@ -27,28 +27,61 @@ flask_app = Flask(__name__)
 class NewsApp:
     def __init__(self):
         self.db = NewsDatabase()
-        self.sentiment_analyzer = SentimentAnalyzer()
-        self.crawler = CnYesNewsCrawler(
-            sentiment_analyzer=self.sentiment_analyzer,
-            database=self.db
-        )
-        self.scheduler = NewsScheduler(self.db, self.crawler, self.sentiment_analyzer)
+        
+        # 延遲初始化情緒分析器
+        self.sentiment_analyzer = None
+        self.crawler = None
+        self.scheduler = None
         
         # 進度追蹤
-        self.current_progress = "系統已就緒"
+        self.current_progress = "正在初始化系統..."
         self.is_crawling = False
+        self.is_initialized = False
         
-        # 設置爬蟲進度回調
-        self.crawler.set_progress_callback(self.update_progress)
+        # 在背景初始化重型組件
+        self._initialize_components()
         
-        # 啟動背景排程器
-        self.scheduler.start()
+        logger.info("新聞應用程式開始初始化")
+
+    def _initialize_components(self):
+        """在背景初始化重型組件"""
+        def init_task():
+            try:
+                self.update_progress("正在載入情緒分析模型...")
+                self.sentiment_analyzer = SentimentAnalyzer()
+                
+                self.update_progress("正在初始化爬蟲...")
+                self.crawler = CnYesNewsCrawler(
+                    sentiment_analyzer=self.sentiment_analyzer,
+                    database=self.db
+                )
+                
+                self.update_progress("正在設置排程器...")
+                self.scheduler = NewsScheduler(self.db, self.crawler, self.sentiment_analyzer)
+                
+                # 設置爬蟲進度回調
+                self.crawler.set_progress_callback(self.update_progress)
+                
+                # 啟動背景排程器
+                self.scheduler.start()
+                
+                self.is_initialized = True
+                self.update_progress("系統初始化完成，已就緒")
+                logger.info("所有組件初始化完成")
+                
+            except Exception as e:
+                error_msg = f"初始化失敗: {str(e)}"
+                self.update_progress(error_msg)
+                logger.error(error_msg)
         
-        logger.info("新聞應用程式初始化完成")
+        # 在背景線程中初始化
+        init_thread = threading.Thread(target=init_task, daemon=True)
+        init_thread.start()
 
     def update_progress(self, message: str):
         """更新進度信息"""
-        self.current_progress = f"{datetime.now().strftime('%H:%M:%S')} - {message}"
+        timestamp = datetime.now().strftime('%H:%M:%S')
+        self.current_progress = f"[{timestamp}] {message}"
         logger.info(f"進度更新: {message}")
 
     def get_progress(self) -> str:
@@ -60,7 +93,7 @@ class NewsApp:
         try:
             news_data = self.db.get_recent_news(category=category, limit=limit)
             if not news_data:
-                return "📰 暫無新聞資料，請稍後再試"
+                return "📰 暫無新聞資料，請先執行爬蟲任務"
             
             return format_news_for_display(news_data)
             
@@ -70,6 +103,9 @@ class NewsApp:
 
     def manual_crawl(self) -> str:
         """手動觸發爬蟲"""
+        if not self.is_initialized:
+            return "⚠️ 系統還在初始化中，請稍後再試"
+            
         if self.is_crawling:
             return "⚠️ 爬蟲正在運行中，請稍後再試"
         
@@ -179,6 +215,12 @@ def api_get_stats():
 def api_manual_crawl():
     """手動觸發爬蟲API"""
     try:
+        if not app.is_initialized:
+            return jsonify({
+                'success': False,
+                'message': '系統還在初始化中'
+            })
+            
         if app.is_crawling:
             return jsonify({
                 'success': False,
@@ -206,9 +248,17 @@ def api_get_progress():
     """獲取爬蟲進度API"""
     return jsonify({
         'progress': app.get_progress(),
-        'is_crawling': app.is_crawling
+        'is_crawling': app.is_crawling,
+        'is_initialized': app.is_initialized
     })
 
+# 定期更新函數
+def periodic_update():
+    """定期更新進度和新聞"""
+    while True:
+        time.sleep(3)  # 每3秒更新一次
+        yield [app.get_progress(), app.get_latest_news("all")]
+
 # 創建 Gradio 介面
 def create_interface():
     with gr.Blocks(
@@ -223,7 +273,7 @@ def create_interface():
         .positive-badge { background: #28a745; color: white; }
         .negative-badge { background: #dc3545; color: white; }
         .neutral-badge { background: #6c757d; color: white; }
-        .progress-box { background: #f8f9fa; border: 1px solid #dee2e6; border-radius: 5px; padding: 10px; font-family: monospace; }
+        .progress-box { background: #f8f9fa; border: 1px solid #dee2e6; border-radius: 5px; padding: 10px; font-family: monospace; font-size: 14px; }
         """
     ) as interface:
         
@@ -253,21 +303,20 @@ def create_interface():
             
             # 進度顯示
             progress_display = gr.Textbox(
-                label="📊 即時進度",
+                label="📊 系統狀態與進度",
                 value=app.get_progress(),
                 interactive=False,
-                elem_classes=["progress-box"]
+                elem_classes=["progress-box"],
+                lines=2
             )
             
             news_display = gr.HTML(label="新聞內容")
             crawl_result = gr.Textbox(label="爬取結果", visible=False)
             
-            # 自動重新整理進度
-            def update_progress_display():
-                return app.get_progress()
-            
-            def auto_refresh():
-                return app.get_latest_news("all")
+            # 更新函數
+            def update_all():
+                """更新所有信息"""
+                return app.get_progress(), app.get_latest_news("all")
             
             def refresh_news(category):
                 return app.get_latest_news(category)
@@ -278,11 +327,11 @@ def create_interface():
                 news = app.get_latest_news("all")
                 return result, news
             
-            # 定期更新進度
-            interface.load(
-                lambda: [app.get_progress(), app.get_latest_news("all")], 
-                outputs=[progress_display, news_display],
-                every=5  # 每5秒更新一次
+            # 使用定時器替代 every 參數
+            timer = gr.Timer(value=3)  # 每3秒觸發一次
+            timer.tick(
+                fn=update_all,
+                outputs=[progress_display, news_display]
             )
             
             # 綁定事件
@@ -295,6 +344,9 @@ def create_interface():
                 outputs=[crawl_result]
             )
             category_radio.change(refresh_news, inputs=[category_radio], outputs=[news_display])
+            
+            # 初始載入
+            interface.load(update_all, outputs=[progress_display, news_display])
         
         with gr.Tab("📊 統計資訊"):
             stats_display = gr.Markdown()
@@ -350,7 +402,7 @@ def create_interface():
             POST /api/crawl
             ```
             
-            #### 4. 獲取爬蟲進度
+            #### 4. 獲取系統狀態
             ```
             GET /api/progress
             ```
@@ -362,12 +414,16 @@ def create_interface():
             import requests
             
             # 獲取所有新聞
-            response = requests.get('http://localhost:7860/api/news')
+            response = requests.get('http://localhost:5000/api/news')
             news_data = response.json()
             
             # 獲取美股新聞
-            response = requests.get('http://localhost:7860/api/news?category=us_stock&limit=10')
+            response = requests.get('http://localhost:5000/api/news?category=us_stock&limit=10')
             us_news = response.json()
+            
+            # 檢查系統狀態
+            response = requests.get('http://localhost:5000/api/progress')
+            status = response.json()
             ```
             
             **JavaScript:**
@@ -398,6 +454,7 @@ def create_interface():
             - **模型**: `uer/roberta-base-finetuned-jd-binary-chinese`
             - **分類**: 正面 (綠色) / 負面 (紅色) / 中性 (灰色)
             - **準確性**: 針對中文金融新聞優化
+            - **後台載入**: 模型在背景載入，不阻塞介面啟動
             
             ### 🕷️ 新聞爬蟲
             - **來源**: 鉅亨網 (cnyes.com)
@@ -419,8 +476,9 @@ def create_interface():
             
             ---
             
-            💡 **提示**: 
-            - 首次啟動可能需要幾分鐘下載模型
+            💡 **使用提示**: 
+            - 首次啟動會在背景下載模型，請耐心等待
+            - 系統狀態會顯示初始化進度
             - 建議使用SSD硬碟提升SQLite性能
             - API接口可用於整合其他應用系統
             """)
@@ -438,6 +496,11 @@ if __name__ == "__main__":
     flask_thread = threading.Thread(target=run_flask, daemon=True)
     flask_thread.start()
     
+    print("🚀 啟動股市新聞情緒分析器...")
+    print("📊 網頁介面: http://localhost:7860")
+    print("🔌 API接口: http://localhost:5000")
+    print("💡 首次啟動需要下載模型，請耐心等待...")
+    
     # 啟動Gradio介面
     interface = create_interface()
     interface.launch(