Update app.py

app.py

@@ -1,33 +1,35 @@
-
+import yfinance as yf
 import streamlit as st
-from playwright.sync_api import sync_playwright
 import pandas as pd
 
-st.set_page_config(page_title="UOA Barchart", layout="wide")
-st.title("Scrape Unusual Options Activity from Barchart")
+# Hàm phân tích VSA và xác suất tăng, giảm ngày hôm sau (kết hợp)
+def combined_vsa_probability(ticker):
+    data = yf.download(ticker, period='5d', interval='1d')
+
+    latest_day = data.iloc[-1]
+    open_price = latest_day['Open'].item()
+    current_price = latest_day['Close'].item() if latest_day.name.time().strftime('%H:%M') >= '16:00' else yf.Ticker(ticker).history(period="1d", interval="1m").iloc[-1]['Close'].item()
+    volume = latest_day['Volume'].item()
+    avg_volume = data['Volume'].iloc[-4:-1].mean().item()
+
+    spread = latest_day['High'].item() - latest_day['Low'].item()
+    close_position = (current_price - latest_day['Low'].item()) / spread
+
+    # Điều kiện kết hợp VSA & xác suất
+    if (current_price > open_price) and (volume > avg_volume) and close_position > 0.7:
+        probability = "Bullish (Strength): Xác suất ~75-80% tăng tiếp"
+    elif (current_price >= open_price) and close_position > 0.5 and volume >= avg_volume:
+        probability = "Bullish (Possible Strength): Xác suất ~60-65% tăng tiếp"
+    elif (current_price < open_price) and close_position < 0.3 and volume >= avg_volume:
+        probability = "Bearish (Weakness): Xác suất cao giảm hôm sau"
+    else:
+        probability = "Sideway hoặc chưa rõ tín hiệu: Quan sát thêm"
+
+    return probability
 
-@st.cache_data
-def fetch_uoa():
-    with sync_playwright() as p:
-        browser = p.chromium.launch(headless=True)
-        page = browser.new_page()
-        page.goto("https://www.barchart.com/options/unusual-activity/stocks", timeout=60000)
-        page.wait_for_selector("table")
-        html = page.content()
-        browser.close()
-    return html
+# Streamlit app
+st.title("Dự đoán xác suất tăng/giảm hôm sau (Kết hợp VSA) cho SPY & QQQ")
 
-try:
-    from bs4 import BeautifulSoup
-    soup = BeautifulSoup(fetch_uoa(), "html.parser")
-    table = soup.find("table")
-    rows = table.find_all("tr")
-    data = []
-    for row in rows[1:]:
-        cols = [col.text.strip() for col in row.find_all("td")]
-        if cols:
-            data.append(cols)
-    df = pd.DataFrame(data)
-    st.dataframe(df)
-except Exception as e:
-    st.error(f"Lỗi: {e}")
+for ticker in ["QQQ", "SPY"]:
+    probability = combined_vsa_probability(ticker)
+    st.subheader(f"{ticker}: {probability}")