Update app.py

app.py

@@ -1,32 +1,23 @@
 import gradio as gr
 import pandas as pd
 from scraper import YahooFinanceScraper
-from sentiment_analyzer import NewsAnalyzer # --- MODIFIED: Import new class ---
+from sentiment_analyzer import NewsAnalyzer # Import new class
 import time
-from collections import Counter # --- ADDED: For counting themes/impacts ---
+from collections import Counter # For counting themes/impacts
 
 # Initialize components
-print("Initializing Yahoo Finance News Analyzer...") # --- MODIFIED ---
+print("Initializing Yahoo Finance News Analyzer...")
 scraper = YahooFinanceScraper()
-analyzer = NewsAnalyzer() # --- MODIFIED: Use new class ---
+analyzer = NewsAnalyzer() # Use new class
 
 def get_sentiment_color(sentiment):
     """กำหนดสีตาม sentiment"""
-    colors = {
-        "Positive": "🟢",
-        "Negative": "🔴",
-        "Neutral": "🟡"
-    }
+    colors = { "Positive": "🟢", "Negative": "🔴", "Neutral": "🟡" }
     return colors.get(sentiment, "⚪")
 
-# --- ADDED: New helper function for impact ---
 def get_impact_color(impact):
     """กำหนดสีตาม impact"""
-    colors = {
-        "Opportunity": "🟢", # ใช้สีเขียวเหมือน Positive
-        "Risk": "🔴",       # ใช้สีแดงเหมือน Negative
-        "Neutral": "🟡"
-    }
+    colors = { "Opportunity": "🟢", "Risk": "🔴", "Neutral": "🟡" }
     return colors.get(impact, "⚪")
 
 def analyze_news(search_type, search_input, num_articles):
@@ -34,10 +25,8 @@ def analyze_news(search_type, search_input, num_articles):
     ฟังก์ชันหลักในการวิเคราะห์ข่าว
     """
     try:
-        # Update progress
         yield "กำลังดึงข่าว...", None, None
 
-        # Fetch news based on search type
         if search_type == "Latest News":
             news_list = scraper.get_latest_news(max_articles=int(num_articles))
         elif search_type == "Stock Symbol":
@@ -57,7 +46,6 @@ def analyze_news(search_type, search_input, num_articles):
 
         yield f"พบข่าว {len(news_list)} รายการ | กำลังวิเคราะห์ sentiment, theme, และ impact...", None, None
 
-        # Analyze sentiment, theme, and impact
         results = analyzer.analyze_batch(news_list)
 
         # Create summary statistics
@@ -67,11 +55,9 @@ def analyze_news(search_type, search_input, num_articles):
         neutral = sum(1 for r in results if r['sentiment'] == 'Neutral')
         avg_score = sum(r['score'] for r in results) / total if total > 0 else 0
 
-        # --- ADDED: Theme and Impact Statistics ---
         theme_counts = Counter(r.get('theme', 'N/A') for r in results)
         impact_counts = Counter(r.get('impact', 'N/A') for r in results)
 
-        # Overall sentiment
         if positive > negative and positive > neutral:
             overall = "📈 Positive (Bullish)"
         elif negative > positive and negative > neutral:
@@ -79,7 +65,6 @@ def analyze_news(search_type, search_input, num_articles):
         else:
             overall = "📊 Neutral"
 
-        # --- MODIFIED: Updated summary block ---
         summary = f"""## 📊 สรุปผลการวิเคราะห์
 **ภาพรวม:** {overall}  |  **คะแนนเฉลี่ย:** {avg_score:.2f}/1.0
 
@@ -107,11 +92,12 @@ def analyze_news(search_type, search_input, num_articles):
             s_emoji = get_sentiment_color(result['sentiment'])
             i_emoji = get_impact_color(result.get('impact', 'N/A'))
             
-            # --- MODIFIED: Added Theme and Impact to detailed view ---
+            # --- MODIFIED: Re-added 'summary' and kept 'explanation' ---
             detailed_results += f"""### {i}. {s_emoji} {result['title']}
 **Theme:** {result.get('theme', 'N/A')} | **Impact:** {i_emoji} {result.get('impact', 'N/A')}
 **Sentiment:** {result['sentiment']} | **Score:** {result['score']:.2f}
 **เผยแพร่:** {result['published']}
+**สรุปข่าว:** {result['summary'][:200]}{'...' if len(result['summary']) > 200 else ''}
 **คำอธิบาย AI:** {result['explanation']}
 [🔗 อ่านต่อ]({result['link']})
 ---"""
@@ -119,7 +105,6 @@ def analyze_news(search_type, search_input, num_articles):
         # Create DataFrame for table view
         df_data = []
         for result in results:
-            # --- MODIFIED: Added Theme and Impact to dataframe ---
             df_data.append({
                 'Title': result['title'][:60] + '...' if len(result['title']) > 60 else result['title'],
                 'Theme': result.get('theme', 'N/A'),
@@ -131,7 +116,6 @@ def analyze_news(search_type, search_input, num_articles):
             })
 
         df = pd.DataFrame(df_data)
-
         final_output = summary + detailed_results
 
         yield final_output, df, "✅ วิเคราะห์เสร็จสมบูรณ์!"
@@ -140,9 +124,8 @@ def analyze_news(search_type, search_input, num_articles):
         error_msg = f"❌ เกิดข้อผิดพลาด: {str(e)}"
         yield error_msg, None, error_msg
 
-# Create Gradio Interface
-with gr.Blocks(title="Yahoo Finance News Analyzer", theme=gr.themes.Soft()) as demo: # --- MODIFIED ---
-    
+# Create Gradio Interface (No changes needed here, keeping it as is)
+with gr.Blocks(title="Yahoo Finance News Analyzer", theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
     # 📈 Yahoo Finance News Analyzer
     วิเคราะห์ข่าวการเงินจาก Yahoo Finance ด้วย AI แบบครบวงจร
@@ -153,7 +136,7 @@ with gr.Blocks(title="Yahoo Finance News Analyzer", theme=gr.themes.Soft()) as d
     - วิเคราะห์ Impact (ผลกระทบ: โอกาส หรือ ความเสี่ยง)
     - ให้คะแนนความมั่นใจและคำอธิบาย
     - รองรับค้นหาตาม Stock Symbol และ Keyword
-    """) # --- MODIFIED ---
+    """)
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -162,13 +145,11 @@ with gr.Blocks(title="Yahoo Finance News Analyzer", theme=gr.themes.Soft()) as d
                 value="Latest News",
                 label="ประเภทการค้นหา"
             )
-            
             search_input = gr.Textbox(
                 label="Ticker Symbol / Keyword",
                 placeholder="เช่น AAPL, TSLA, AI, cryptocurrency",
                 info="ใส่เฉพาะเมื่อเลือก Stock Symbol หรือ Keyword"
             )
-            
             num_articles = gr.Slider(
                 minimum=5,
                 maximum=20,
@@ -176,9 +157,7 @@ with gr.Blocks(title="Yahoo Finance News Analyzer", theme=gr.themes.Soft()) as d
                 step=1,
                 label="จำนวนข่าว"
             )
-            
             analyze_btn = gr.Button("🔍 วิเคราะห์ข่าว", variant="primary", size="lg")
-            
             status = gr.Textbox(label="สถานะ", interactive=False)
 
         with gr.Column(scale=2):
@@ -209,9 +188,8 @@ with gr.Blocks(title="Yahoo Finance News Analyzer", theme=gr.themes.Soft()) as d
     - **Sentiment:** Positive (ดี), Negative (แย่), Neutral (กลาง)
     - **Theme:** หัวข้อหลักของข่าว (เช่น ผลประกอบการ, สินค้าใหม่)
     - **Impact:** ผลกระทบต่อบริษัท (🟢 Opportunity - โอกาส, 🔴 Risk - ความเสี่ยง)
-    """) # --- MODIFIED ---
+    """)
 
-    # Connect button to function
     analyze_btn.click(
         fn=analyze_news,
         inputs=[search_type, search_input, num_articles],