Update app.py

app.py

@@ -96,26 +96,60 @@ class NewsApp:
         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, force_refresh: bool = False) -> str:
-        """獲取最新新聞並格式化顯示"""
+    def get_latest_news(self, category: str = "all", days: int = 7, 
+                       keyword: str = "", sentiment_filter: str = "all", 
+                       force_refresh: bool = False) -> str:
+        """獲取最新新聞並格式化顯示 - 增強版"""
         try:
             # 檢查是否需要刷新（避免無意義的閃爍）
             current_time = time.time()
-            if not force_refresh and (current_time - self.last_news_update) < 10:
-                # 10秒內不重複查詢，除非強制刷新
+            if not force_refresh and (current_time - self.last_news_update) < 5:
+                # 5秒內不重複查詢，除非強制刷新
                 pass
             
             self.last_news_update = current_time
             
-            # 記錄分類選擇
-            logger.info(f"獲取新聞 - 分類: {category}, 限制: {limit}")
+            # 記錄查詢參數
+            logger.info(f"獲取新聞 - 分類: {category}, 天數: {days}, 關鍵字: '{keyword}', 情緒: {sentiment_filter}")
             
-            news_data = self.db.get_recent_news(category=category, limit=limit)
-            if not news_data:
-                return f"📰 暫無 {self._get_category_name(category)} 新聞資料，請先執行爬蟲任務"
+            news_data = self.db.get_recent_news(
+                category=category, 
+                days=days, 
+                keyword=keyword, 
+                sentiment_filter=sentiment_filter
+            )
             
-            # 添加分類標題
-            category_title = f"📊 當前顯示: {self._get_category_name(category)} ({len(news_data)} 篇)"
+            if not news_data:
+                filter_desc = []
+                if category != "all":
+                    filter_desc.append(f"分類: {self._get_category_name(category)}")
+                if days > 0:
+                    filter_desc.append(f"時間: {days}天內")
+                if keyword:
+                    filter_desc.append(f"關鍵字: '{keyword}'")
+                if sentiment_filter != "all":
+                    filter_desc.append(f"情緒: {self._get_sentiment_name(sentiment_filter)}")
+                
+                filter_text = "、".join(filter_desc) if filter_desc else "所有條件"
+                return f"📰 暫無符合條件的新聞資料 ({filter_text})，請調整篩選條件或執行爬蟲任務"
+            
+            # 添加查詢結果標題
+            filter_parts = []
+            if category != "all":
+                filter_parts.append(self._get_category_name(category))
+            if days > 0:
+                filter_parts.append(f"{days}天內")
+            if keyword:
+                filter_parts.append(f"關鍵字「{keyword}」")
+            if sentiment_filter != "all":
+                filter_parts.append(f"{self._get_sentiment_name(sentiment_filter)}情緒")
+            
+            if filter_parts:
+                title_desc = " | ".join(filter_parts)
+            else:
+                title_desc = "所有新聞"
+            
+            category_title = f"📊 當前顯示: {title_desc} ({len(news_data)} 篇)"
             formatted_news = 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}"
@@ -132,6 +166,16 @@ class NewsApp:
             "tw_stock": "台股新聞"
         }
         return category_names.get(category, category)
+    
+    def _get_sentiment_name(self, sentiment: str) -> str:
+        """獲取情緒中文名稱"""
+        sentiment_names = {
+            "all": "所有",
+            "positive": "正面",
+            "negative": "負面",
+            "neutral": "中性"
+        }
+        return sentiment_names.get(sentiment, sentiment)
 
     def manual_crawl(self) -> str:
         """手動觸發爬蟲"""
@@ -145,8 +189,8 @@ class NewsApp:
             self.is_crawling = True
             self.update_progress("🚀 手動爬蟲開始")
             
-            # 使用新的即時爬蟲方法
-            results = self.crawler.crawl_all_categories(max_articles_per_category=5)
+            # 使用新的即時爬蟲方法 - 移除限制
+            results = self.crawler.crawl_all_categories(max_articles_per_category=20)  # 增加到20篇
             
             total_articles = sum(len(articles) for articles in results.values())
             result_message = f"✅ 手動爬蟲完成，總共處理 {total_articles} 篇文章"
@@ -168,6 +212,7 @@ class NewsApp:
             return f"""
 📊 **新聞統計**
 - 總新聞數量: {stats.get('total_news', 0)}
+- 近7天新聞: {stats.get('recent_news', 0)}
 - 美股新聞: {stats.get('us_stock_count', 0)}
 - 台股新聞: {stats.get('tw_stock_count', 0)}
 - 正面新聞: {stats.get('positive_count', 0)} 😊
@@ -179,10 +224,16 @@ class NewsApp:
             logger.error(f"獲取統計資訊錯誤: {e}")
             return f"❌ 獲取統計資訊失敗: {str(e)}"
 
-    def get_news_api_data(self, category: str = "all", limit: int = 50) -> Dict:
+    def get_news_api_data(self, category: str = "all", days: int = 7, 
+                         keyword: str = "", sentiment_filter: str = "all") -> Dict:
         """獲取新聞API數據"""
         try:
-            news_data = self.db.get_recent_news(category=category, limit=limit)
+            news_data = self.db.get_recent_news(
+                category=category, 
+                days=days, 
+                keyword=keyword, 
+                sentiment_filter=sentiment_filter
+            )
             
             # 轉換為JSON友好格式
             api_data = []
@@ -221,11 +272,13 @@ app = NewsApp()
 # API 路由
 @flask_app.route('/api/news', methods=['GET'])
 def api_get_news():
-    """獲取新聞列表API"""
+    """獲取新聞列表API - 增強版"""
     category = request.args.get('category', 'all')
-    limit = int(request.args.get('limit', 50))
+    days = int(request.args.get('days', 7))
+    keyword = request.args.get('keyword', '')
+    sentiment_filter = request.args.get('sentiment', 'all')
     
-    result = app.get_news_api_data(category, limit)
+    result = app.get_news_api_data(category, days, keyword, sentiment_filter)
     return jsonify(result)
 
 @flask_app.route('/api/stats', methods=['GET'])
@@ -305,19 +358,19 @@ def create_interface():
     ) as interface:
         
         gr.Markdown("""
-        # 📈 股市新聞情緒分析器 - 即時版
+        # 📈 股市新聞情緒分析器 - 全功能版
         
         🤖 自動爬取鉅亨網美股和台股新聞，並進行即時中文情緒分析
         
         ⚡ **即時處理**: 每篇文章完成後立即分析並存檔  
         🎯 **智能分析**: 使用 RoBERTa 模型進行情緒分析  
-        🔄 **去重處理**: 自動過濾重複新聞  
+        🔍 **多條件篩選**: 支援時間段、關鍵字、情緒篩選  
         📊 **API接口**: 提供RESTful API獲取分析結果  
         """)
         
         with gr.Tab("📰 最新新聞"):
             with gr.Row():
-                with gr.Column(scale=2):
+                with gr.Column(scale=1):
                     category_radio = gr.Radio(
                         choices=[
                             ("所有新聞", "all"),
@@ -325,31 +378,56 @@ def create_interface():
                             ("台股新聞", "tw_stock")
                         ],
                         value="all",
-                        label="📋 新聞分類",
-                        info="選擇要顯示的新聞類型"
+                        label="📋 新聞分類"
+                    )
+                    
+                    days_slider = gr.Slider(
+                        minimum=0,
+                        maximum=30,
+                        value=7,
+                        step=1,
+                        label="📅 時間範圍 (天)",
+                        info="0表示不限制時間"
+                    )
+                    
+                    keyword_input = gr.Textbox(
+                        label="🔍 關鍵字搜尋",
+                        placeholder="輸入關鍵字搜尋新聞...",
+                        value=""
+                    )
+                    
+                    sentiment_radio = gr.Radio(
+                        choices=[
+                            ("所有情緒", "all"),
+                            ("正面情緒", "positive"),
+                            ("負面情緒", "negative"),
+                            ("中性情緒", "neutral")
+                        ],
+                        value="all",
+                        label="😊 情緒篩選"
+                    )
+                
+                with gr.Column(scale=2):
+                    with gr.Row():
+                        search_btn = gr.Button("🔍 搜尋新聞", variant="primary")
+                        refresh_btn = gr.Button("🔄 重新整理", variant="secondary")
+                        manual_crawl_btn = gr.Button("🚀 手動爬取", variant="secondary")
+                    
+                    # 進度顯示
+                    progress_display = gr.Textbox(
+                        label="📊 系統狀態",
+                        value=app.current_progress,
+                        interactive=False,
+                        elem_classes=["progress-box"],
+                        lines=1
                     )
-                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.current_progress,
-                interactive=False,
-                elem_classes=["progress-box"],
-                lines=1
-            )
             
             news_display = gr.HTML(
                 label="新聞內容",
-                value=app.get_latest_news("all", force_refresh=True)
+                value=app.get_latest_news("all", 7, "", "all", force_refresh=True)
             )
             crawl_result = gr.Textbox(label="爬取結果", visible=False)
             
-            # 狀態變量，用於追蹤當前選擇的分類
-            current_category = gr.State("all")
-            
             # 更新函數
             def update_progress_only():
                 """只更新進度，不更新新聞"""
@@ -357,54 +435,64 @@ def create_interface():
                 if needs_update or app.is_crawling:
                     return progress
                 else:
-                    return gr.update()  # 不更新
+                    return gr.update()
+            
+            def search_news(category, days, keyword, sentiment):
+                """搜尋新聞"""
+                logger.info(f"搜尋新聞 - 分類: {category}, 天數: {days}, 關鍵字: '{keyword}', 情緒: {sentiment}")
+                return app.get_latest_news(category, days, keyword, sentiment, force_refresh=True)
             
-            def refresh_news(category):
-                """刷新新聞內容"""
-                logger.info(f"刷新新聞 - 用戶選擇分類: {category}")
-                return app.get_latest_news(category, force_refresh=True), category
+            def refresh_current_search(category, days, keyword, sentiment):
+                """刷新當前搜尋"""
+                return app.get_latest_news(category, days, keyword, sentiment, force_refresh=True)
             
-            def handle_manual_crawl(current_cat):
+            def handle_manual_crawl(category, days, keyword, sentiment):
                 """處理手動爬蟲"""
                 result = app.manual_crawl()
-                # 爬取完成後自動刷新當前分類的新聞
-                news = app.get_latest_news(current_cat, force_refresh=True)
+                # 爬取完成後自動刷新當前搜尋
+                news = app.get_latest_news(category, days, keyword, sentiment, force_refresh=True)
                 return result, news
             
-            # 進度更新定時器 - 降低頻率，減少閃爍
-            progress_timer = gr.Timer(value=10)  # 每10秒檢查一次進度
+            # 進度更新定時器
+            progress_timer = gr.Timer(value=10)
             progress_timer.tick(
                 fn=update_progress_only,
                 outputs=[progress_display]
             )
             
             # 綁定事件
+            search_btn.click(
+                search_news,
+                inputs=[category_radio, days_slider, keyword_input, sentiment_radio],
+                outputs=[news_display]
+            )
+            
             refresh_btn.click(
-                refresh_news, 
-                inputs=[category_radio], 
-                outputs=[news_display, current_category]
+                refresh_current_search,
+                inputs=[category_radio, days_slider, keyword_input, sentiment_radio],
+                outputs=[news_display]
             )
             
             manual_crawl_btn.click(
                 handle_manual_crawl,
-                inputs=[current_category],
+                inputs=[category_radio, days_slider, keyword_input, sentiment_radio],
                 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, current_category]
+                search_news,
+                inputs=[category_radio, days_slider, keyword_input, sentiment_radio],
+                outputs=[news_display]
             )
             
             # 初始載入
             interface.load(
-                refresh_news, 
-                inputs=[gr.State("all")], 
-                outputs=[news_display, current_category]
+                lambda: app.get_latest_news("all", 7, "", "all", force_refresh=True),
+                outputs=[news_display]
             )
         
         with gr.Tab("📊 統計資訊"):
@@ -420,38 +508,20 @@ def create_interface():
             
             ### 🔗 接口列表
             
-            #### 1. 獲取新聞列表
+            #### 1. 獲取新聞列表 (增強版)
             ```
-            GET /api/news?category={all|us_stock|tw_stock}&limit={數量}
+            GET /api/news?category={all|us_stock|tw_stock}&days={天數}&keyword={關鍵字}&sentiment={all|positive|negative|neutral}
             ```
             
             **參數:**
             - `category`: 新聞分類 (可選，默認: all)
-              - `all`: 所有新聞
-              - `us_stock`: 美股新聞
-              - `tw_stock`: 台股新聞
-            - `limit`: 返回數量 (可選，默認: 50)
-            
-            **響應示例:**
-            ```json
-            {
-                "success": true,
-                "count": 10,
-                "data": [
-                    {
-                        "id": 1,
-                        "title": "美股標題",
-                        "content": "新聞內容...",
-                        "url": "https://...",
-                        "source": "鉅亨網",
-                        "category": "us_stock",
-                        "published_date": "2024-01-01T12:00:00",
-                        "sentiment": "positive",
-                        "sentiment_score": 0.85,
-                        "created_date": "2024-01-01T12:05:00"
-                    }
-                ]
-            }
+            - `days`: 時間範圍天數 (可選，默認: 7，0表示不限制)
+            - `keyword`: 關鍵字搜尋 (可選)
+            - `sentiment`: 情緒篩選 (可選，默認: all)
+            
+            **示例:**
+            ```
+            /api/news?category=us_stock&days=3&keyword=AI&sentiment=positive
             ```
             
             #### 2. 獲取統計信息
@@ -475,17 +545,13 @@ def create_interface():
             ```python
             import requests
             
-            # 獲取所有新聞
-            response = requests.get('http://localhost:5000/api/news')
+            # 獲取所有正面情緒的美股新聞
+            response = requests.get('http://localhost:5000/api/news?category=us_stock&sentiment=positive')
             news_data = response.json()
             
-            # 獲取美股新聞
-            response = requests.get('http://localhost:5000/api/news?category=us_stock&limit=10')
-            us_news = response.json()
-            
-            # 獲取台股新聞
-            response = requests.get('http://localhost:5000/api/news?category=tw_stock&limit=10')
-            tw_news = response.json()
+            # 搜尋包含"AI"關鍵字的新聞
+            response = requests.get('http://localhost:5000/api/news?keyword=AI&days=3')
+            ai_news = response.json()
             ```
             """)
         
@@ -497,22 +563,28 @@ def create_interface():
             - **實時分析**: 每篇文章爬取完成立即進行情緒分析
             - **即時存檔**: 分析完成後立即保存到SQLite資料庫
             - **智能刷新**: 避免無意義的頁面閃爍
-            - **分類顯示**: 支援美股、台股、全部新聞分類查看
+            - **去重處理**: 智能檢測重複文章，降低重複度閾值
+            
+            ### 🔍 多條件篩選
+            - **時間篩選**: 支援0-30天的時間範圍選擇
+            - **關鍵字搜尋**: 支援標題和內容的關鍵字搜尋
+            - **情緒篩選**: 可按正面、負面、中性情緒篩選
+            - **分類篩選**: 支援美股、台股分類查看
             
             ### 📊 情緒分析
-            - **模型**: `uer/roberta-base-finetuned-jd-binary-chinese`
-            - **分類**: 正面 (綠色) / 負面 (紅色) / 中性 (灰色)
+            - **混合模型**: RoBERTa模型 + 關鍵字分析
+            - **容錯處理**: 模型失敗時自動使用關鍵字分析
             - **準確性**: 針對中文金融新聞優化
             
             ### 🕷️ 新聞來源
             - **美股**: https://news.cnyes.com/news/cat/us_stock
             - **台股**: https://news.cnyes.com/news/cat/tw_stock_news
-            - **頻率**: 每30分鐘自動更新
+            - **無限制**: 移除文章數量限制，爬取更多新聞
             
             ### 📱 介面優化
-            - **防閃爍**: 智能更新機制，減少不必要的刷新
-            - **分類標示**: 清楚顯示當前查看的新聞分類
-            - **狀態追蹤**: 實時顯示系統運行狀態
+            - **防閃爍**: 智能更新機制
+            - **多條件篩選**: 支援複合條件搜尋
+            - **即時反饋**: 詳細的搜尋結果統計
             """)
     
     return interface
@@ -531,7 +603,7 @@ if __name__ == "__main__":
     print("🚀 啟動股市新聞情緒分析器...")
     print("📊 網頁介面: http://localhost:7860")
     print("🔌 API接口: http://localhost:5000")
-    print("💡 首次啟動需要下載模型，請耐心等待...")
+    print("💡 新功能: 時間段篩選、關鍵字搜尋、情緒篩選")
     
     # 啟動Gradio介面
     interface = create_interface()