Update app.py

app.py

@@ -38,6 +38,10 @@ class NewsApp:
         self.is_crawling = False
         self.is_initialized = False
         
+        # 上次新聞更新時間，用於防止無意義的刷新
+        self.last_news_update = 0
+        self.last_progress_update = 0
+        
         # 在背景初始化重型組件
         self._initialize_components()
         
@@ -82,25 +86,53 @@ class NewsApp:
         """更新進度信息"""
         timestamp = datetime.now().strftime('%H:%M:%S')
         self.current_progress = f"[{timestamp}] {message}"
+        self.last_progress_update = time.time()
         logger.info(f"進度更新: {message}")
 
-    def get_progress(self) -> str:
-        """獲取當前進度"""
-        return self.current_progress
+    def get_progress(self) -> tuple:
+        """獲取當前進度和是否需要更新"""
+        current_time = time.time()
+        # 只有在進度真的有更新時才返回新內容
+        needs_update = (current_time - self.last_progress_update) < 5  # 5秒內的更新才顯示
+        return self.current_progress, needs_update
 
-    def get_latest_news(self, category: str = "all", limit: int = 50) -> str:
+    def get_latest_news(self, category: str = "all", limit: int = 50, force_refresh: bool = False) -> str:
         """獲取最新新聞並格式化顯示"""
         try:
+            # 檢查是否需要刷新（避免無意義的閃爍）
+            current_time = time.time()
+            if not force_refresh and (current_time - self.last_news_update) < 10:
+                # 10秒內不重複查詢，除非強制刷新
+                pass
+            
+            self.last_news_update = current_time
+            
+            # 記錄分類選擇
+            logger.info(f"獲取新聞 - 分類: {category}, 限制: {limit}")
+            
             news_data = self.db.get_recent_news(category=category, limit=limit)
             if not news_data:
-                return "📰 暫無新聞資料，請先執行爬蟲任務"
+                return f"📰 暫無 {self._get_category_name(category)} 新聞資料，請先執行爬蟲任務"
+            
+            # 添加分類標題
+            category_title = f"📊 當前顯示: {self._get_category_name(category)} ({len(news_data)} 篇)"
+            formatted_news = format_news_for_display(news_data)
             
-            return format_news_for_display(news_data)
+            return f"<div style='background: #e3f2fd; padding: 10px; margin-bottom: 15px; border-radius: 5px; text-align: center; font-weight: bold;'>{category_title}</div>{formatted_news}"
             
         except Exception as e:
             logger.error(f"獲取新聞時發生錯誤: {e}")
             return f"❌ 獲取新聞時發生錯誤: {str(e)}"
 
+    def _get_category_name(self, category: str) -> str:
+        """獲取分類中文名稱"""
+        category_names = {
+            "all": "所有新聞",
+            "us_stock": "美股新聞", 
+            "tw_stock": "台股新聞"
+        }
+        return category_names.get(category, category)
+
     def manual_crawl(self) -> str:
         """手動觸發爬蟲"""
         if not self.is_initialized:
@@ -246,19 +278,14 @@ def api_manual_crawl():
 @flask_app.route('/api/progress', methods=['GET'])
 def api_get_progress():
     """獲取爬蟲進度API"""
+    progress, needs_update = app.get_progress()
     return jsonify({
-        'progress': app.get_progress(),
+        'progress': progress,
         'is_crawling': app.is_crawling,
-        'is_initialized': app.is_initialized
+        'is_initialized': app.is_initialized,
+        'needs_update': needs_update
     })
 
-# 定期更新函數
-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(
@@ -292,61 +319,93 @@ def create_interface():
             with gr.Row():
                 with gr.Column(scale=2):
                     category_radio = gr.Radio(
-                        choices=["all", "us_stock", "tw_stock"],
+                        choices=[
+                            ("所有新聞", "all"),
+                            ("美股新聞", "us_stock"), 
+                            ("台股新聞", "tw_stock")
+                        ],
                         value="all",
-                        label="新聞分類",
+                        label="📋 新聞分類",
                         info="選擇要顯示的新聞類型"
                     )
                 with gr.Column(scale=1):
                     refresh_btn = gr.Button("🔄 重新整理", variant="primary")
                     manual_crawl_btn = gr.Button("🚀 手動爬取", variant="secondary")
             
-            # 進度顯示
+            # 進度顯示 - 只在有更新時顯示
             progress_display = gr.Textbox(
-                label="📊 系統狀態與進度",
-                value=app.get_progress(),
+                label="📊 系統狀態",
+                value=app.current_progress,
                 interactive=False,
                 elem_classes=["progress-box"],
-                lines=2
+                lines=1
             )
             
-            news_display = gr.HTML(label="新聞內容")
+            news_display = gr.HTML(
+                label="新聞內容",
+                value=app.get_latest_news("all", force_refresh=True)
+            )
             crawl_result = gr.Textbox(label="爬取結果", visible=False)
             
+            # 狀態變量，用於追蹤當前選擇的分類
+            current_category = gr.State("all")
+            
             # 更新函數
-            def update_all():
-                """更新所有信息"""
-                return app.get_progress(), app.get_latest_news("all")
+            def update_progress_only():
+                """只更新進度，不更新新聞"""
+                progress, needs_update = app.get_progress()
+                if needs_update or app.is_crawling:
+                    return progress
+                else:
+                    return gr.update()  # 不更新
             
             def refresh_news(category):
-                return app.get_latest_news(category)
+                """刷新新聞內容"""
+                logger.info(f"刷新新聞 - 用戶選擇分類: {category}")
+                return app.get_latest_news(category, force_refresh=True), category
             
-            def handle_manual_crawl():
+            def handle_manual_crawl(current_cat):
+                """處理手動爬蟲"""
                 result = app.manual_crawl()
-                # 爬取完成後自動刷新新聞
-                news = app.get_latest_news("all")
+                # 爬取完成後自動刷新當前分類的新聞
+                news = app.get_latest_news(current_cat, force_refresh=True)
                 return result, news
             
-            # 使用定時器替代 every 參數
-            timer = gr.Timer(value=3)  # 每3秒觸發一次
-            timer.tick(
-                fn=update_all,
-                outputs=[progress_display, news_display]
+            # 進度更新定時器 - 降低頻率，減少閃爍
+            progress_timer = gr.Timer(value=10)  # 每10秒檢查一次進度
+            progress_timer.tick(
+                fn=update_progress_only,
+                outputs=[progress_display]
             )
             
             # 綁定事件
-            refresh_btn.click(refresh_news, inputs=[category_radio], outputs=[news_display])
+            refresh_btn.click(
+                refresh_news, 
+                inputs=[category_radio], 
+                outputs=[news_display, current_category]
+            )
+            
             manual_crawl_btn.click(
-                handle_manual_crawl, 
+                handle_manual_crawl,
+                inputs=[current_category],
                 outputs=[crawl_result, news_display]
             ).then(
                 lambda: gr.update(visible=True), 
                 outputs=[crawl_result]
             )
-            category_radio.change(refresh_news, inputs=[category_radio], outputs=[news_display])
+            
+            category_radio.change(
+                refresh_news, 
+                inputs=[category_radio], 
+                outputs=[news_display, current_category]
+            )
             
             # 初始載入
-            interface.load(update_all, outputs=[progress_display, news_display])
+            interface.load(
+                refresh_news, 
+                inputs=[gr.State("all")], 
+                outputs=[news_display, current_category]
+            )
         
         with gr.Tab("📊 統計資訊"):
             stats_display = gr.Markdown()
@@ -368,6 +427,9 @@ def create_interface():
             
             **參數:**
             - `category`: 新聞分類 (可選，默認: all)
+              - `all`: 所有新聞
+              - `us_stock`: 美股新聞
+              - `tw_stock`: 台股新聞
             - `limit`: 返回數量 (可選，默認: 50)
             
             **響應示例:**
@@ -421,22 +483,9 @@ def create_interface():
             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:**
-            ```javascript
-            // 獲取新聞
-            fetch('/api/news?category=tw_stock')
-                .then(response => response.json())
-                .then(data => console.log(data));
-            
-            // 觸發爬蟲
-            fetch('/api/crawl', {method: 'POST'})
-                .then(response => response.json())
-                .then(data => console.log(data));
+            # 獲取台股新聞
+            response = requests.get('http://localhost:5000/api/news?category=tw_stock&limit=10')
+            tw_news = response.json()
             ```
             """)
         
@@ -447,40 +496,23 @@ def create_interface():
             ### ⚡ 即時處理
             - **實時分析**: 每篇文章爬取完成立即進行情緒分析
             - **即時存檔**: 分析完成後立即保存到SQLite資料庫
-            - **進度追蹤**: 實時顯示爬蟲和分析進度
-            - **去重檢查**: 存檔前檢查標題相似度避免重複
+            - **智能刷新**: 避免無意義的頁面閃爍
+            - **分類顯示**: 支援美股、台股、全部新聞分類查看
             
             ### 📊 情緒分析
             - **模型**: `uer/roberta-base-finetuned-jd-binary-chinese`
             - **分類**: 正面 (綠色) / 負面 (紅色) / 中性 (灰色)
             - **準確性**: 針對中文金融新聞優化
-            - **後台載入**: 模型在背景載入，不阻塞介面啟動
             
-            ### 🕷️ 新聞爬蟲
-            - **來源**: 鉅亨網 (cnyes.com)
-            - **分類**: 美股、台股新聞
+            ### 🕷️ 新聞來源
+            - **美股**: https://news.cnyes.com/news/cat/us_stock
+            - **台股**: https://news.cnyes.com/news/cat/tw_stock_news
             - **頻率**: 每30分鐘自動更新
-            - **反爬蟲**: 隨機延遲、User-Agent輪換
-            
-            ### 💾 資料管理
-            - **儲存**: SQLite 本地資料庫 (news.db)
-            - **索引**: 針對URL、分類、時間、情緒建立索引
-            - **保留期**: 自動清理兩週前的資料
-            - **效能**: 優化查詢，支持並發存取
-            
-            ### 🔌 API接口
-            - **REST API**: 提供完整的RESTful API
-            - **JSON格式**: 標準JSON響應格式
-            - **錯誤處理**: 完善的錯誤處理和狀態碼
-            - **跨域支持**: 支持CORS跨域請求
-            
-            ---
-            
-            💡 **使用提示**: 
-            - 首次啟動會在背景下載模型，請耐心等待
-            - 系統狀態會顯示初始化進度
-            - 建議使用SSD硬碟提升SQLite性能
-            - API接口可用於整合其他應用系統
+            
+            ### 📱 介面優化
+            - **防閃爍**: 智能更新機制，減少不必要的刷新
+            - **分類標示**: 清楚顯示當前查看的新聞分類
+            - **狀態追蹤**: 實時顯示系統運行狀態
             """)
     
     return interface