Update app.py

app.py

@@ -1,14 +1,57 @@
+import yfinance as yf
+import requests
+from bs4 import BeautifulSoup
 from transformers import pipeline
 import gradio as gr
 
+# Load mô hình miễn phí
 pipe = pipeline("text2text-generation", model="google/flan-t5-xl")
 
-def analyze_news(text):
-    prompt = f"Tóm tắt và phân tích tin tức tài chính sau: {text}"
-    result = pipe(prompt, max_length=200)[0]["generated_text"]
-    return result
+# Hàm fetch nội dung bài báo từ link
+def fetch_article(url):
+    headers = {"User-Agent": "Mozilla/5.0"}
+    r = requests.get(url, headers=headers)
+    soup = BeautifulSoup(r.text, "html.parser")
+    paragraphs = soup.find_all("p")
+    return " ".join(p.text for p in paragraphs)
 
-gr.Interface(fn=analyze_news,
-             inputs=gr.Textbox(lines=8, placeholder="Dán nội dung tin tức Yahoo Finance ở đây..."),
-             outputs="text",
-             title="Phân tích Tin tức Tài chính (FREE)").launch()
+# Hàm xử lý chính
+def auto_analyze_news(ticker="SPY"):
+    stock = yf.Ticker(ticker)
+    news = stock.news
+
+    if not news:
+        return "Không có tin tức mới nào cho mã này."
+
+    article = news[0]  # lấy tin mới nhất
+    url = article['link']
+    title = article['title']
+
+    try:
+        content = fetch_article(url)
+    except:
+        return f"Không thể lấy nội dung từ bài: {title}"
+
+    prompt = f"""
+Tin tức: {title}
+
+{content}
+
+Hãy phân tích:
+1. Đây là tin Bullish, Bearish hay Neutral?
+2. Ảnh hưởng thế nào đến {ticker}, SPY hoặc QQQ?
+3. Tóm tắt ngắn nội dung bài viết.
+"""
+    result = pipe(prompt, max_length=512)[0]["generated_text"]
+    return f"**Tiêu đề:** {title}\n\n**Phân tích AI:**\n{result}"
+
+# Gradio giao diện
+demo = gr.Interface(
+    fn=auto_analyze_news,
+    inputs=gr.Textbox(value="SPY", label="Ticker cần lấy tin (mặc định SPY)"),
+    outputs="text",
+    title="AI Tự động Phân tích Tin tài chính mới nhất",
+    description="Bot sẽ tự lấy tin mới nhất từ Yahoo Finance qua yfinance, phân tích xu hướng và ảnh hưởng đến SPY/QQQ."
+)
+
+demo.launch()