Update main.py

main.py

@@ -238,11 +238,7 @@ def setup_message_handler():
     def handle_message(event):
         user_message = event.message.text.strip()
 
-        # 1. 處理特殊指令
-        if user_message == "再見":
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="Bye!"))
-            return
-        
+        # 處理 podcast 指令
         if user_message.lower() == "podcast":
             audio_url = "https://huggingface.co/spaces/tkkbbo332/stockline/resolve/main/podcast.mp3"
             line_bot_api.reply_message(
@@ -253,71 +249,30 @@ def setup_message_handler():
 
         response_text = None
 
-        # 2. 嘗試使用 Gemini 進行 PDF 分析 (如果訊息為股票代號)
-        if user_message.isdigit() and len(user_message) >= 4 and working_status and gemini_client:
-            try:
-                doc_url = f"https://www.twse.com.tw/pdf/ch/{user_message}_ch.pdf"
-                doc_data = httpx.get(doc_url, timeout=10.0)
-                
-                if doc_data.status_code == 200:
-                    with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as temp_file:
-                        temp_file.write(doc_data.content)
-                        temp_file_path = temp_file.name
-                    
-                    print(f"Uploading {temp_file_path} for analysis...")
-                    sample_doc = gemini_client.upload_file(path=temp_file_path)
-                    
-                    model = genai.GenerativeModel(
-                        model_name="models/gemini-1.5-flash", 
-                        generation_config=generation_config,
-                        system_instruction=system_instruction
-                    )
-                    completion = model.generate_content(sample_doc)
-                    response_text = completion.text
-                    
-                    # 清理暫存檔
-                    try:
-                        os.unlink(temp_file_path)
-                    except:
-                        pass
-                else:
-                    print(f"查無股票代號 {user_message} 的公開說明書 (HTTP Status: {doc_data.status_code})。")
-            except Exception as e:
-                print(f"Gemini 執行出錯: {e}")
-                response_text = "Gemini AI 分析時發生錯誤，請稍後再試。"
-                # 確保清理暫存檔
-                try:
-                    if 'temp_file_path' in locals():
-                        os.unlink(temp_file_path)
-                except:
-                    pass
-
-        # 3. 如果 Gemini 未觸發或失敗，則嘗試 yfinance 技術分析
-        if response_text is None:
-            stock_symbol = None
-            stock_name = None
-            for name, symbol in TAIWAN_STOCKS.items():
-                # 檢查名稱或代號 (例如 '2330' in '2330.TW')
-                if name in user_message or symbol.split('.')[0] in user_message:
-                    stock_symbol = symbol
-                    stock_name = name
-                    break
+        # 嘗試進行股票名稱分析
+        stock_symbol = None
+        stock_name = None
+        for name, symbol in TAIWAN_STOCKS.items():
+            # 檢查名稱或代號 (例如 '2330' in '2330.TW')
+            if name in user_message or symbol.split('.')[0] in user_message:
+                stock_symbol = symbol
+                stock_name = name
+                break
+        
+        if stock_symbol:
+            stock_data = get_stock_data(stock_symbol, period='6mo')
+            data_with_indicators = calculate_technical_indicators(stock_data)
             
-            if stock_symbol:
-                stock_data = get_stock_data(stock_symbol, period='6mo')
-                data_with_indicators = calculate_technical_indicators(stock_data)
-                
-                report_part = generate_analysis_report(data_with_indicators, stock_name)
-                outlook_part = generate_market_outlook(data_with_indicators, stock_name)
-                response_text = report_part + outlook_part
+            report_part = generate_analysis_report(data_with_indicators, stock_name)
+            outlook_part = generate_market_outlook(data_with_indicators, stock_name)
+            response_text = report_part + outlook_part
         
-        # 4. 如果都沒有匹配，回覆預設訊息與網址
+        # 如果沒有匹配，回覆預設訊息與網址
         if response_text is None:
             response_text = (
                 f"無法識別您的指令。\n\n"
-                f"👉 輸入台灣上市公司【股票代號】(如: 2330) 進行 AI 深度分析。\n"
                 f"👉 輸入【股票名稱】(如: 台積電) 查詢即時技術指標。\n\n"
-                f"您也可以試試輸入 'podcast' 或 '再見'。\n\n"
+                f"您也可以試試輸入 'podcast'。\n\n"
                 f"🔗 更多功能請訪問：\nhttps://huggingface.co/spaces/AlanRex/AITEST"
             )