Spaces:
Sleeping
Sleeping
File size: 5,429 Bytes
97b7267 f6c927c 97b7267 f6c927c 97b7267 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 | import streamlit as st
import os
from pymongo import MongoClient
import pandas as pd
from datetime import datetime
from report_generator import generate_stock_report, show_report
from utils import load_css, call_genai_summary
import re
def main():
mongo_uri = os.getenv("MONGO_URI")
mongo_dbname = os.getenv("MONGO_DBNAME")
def get_db():
client = MongoClient(mongo_uri)
db = client[mongo_dbname]
return db["reports"]
reports_history = get_db()
load_css()
if "uid" not in st.session_state:
st.warning("Vui lòng đăng nhập để sử dụng tính năng này.")
st.page_link("StockInsights.py", label="Về trang Đăng nhập", icon="🏠")
st.stop()
# ==== Giao diện trang báo cáo ===
st.markdown("<h2>Báo cáo Cổ phiếu Thông minh</h2>", unsafe_allow_html=True)
# st.markdown("<div class='report-container'>", unsafe_allow_html=True)
col_history, col_main = st.columns([1, 3])
# ===== LEFT: Report History =====
with col_history:
st.markdown("<h4>Báo cáo đã xem</h4>", unsafe_allow_html=True)
# Initialize history list
if "reports_history_list" not in st.session_state:
query = {"uid": st.session_state["uid"]}
st.session_state["reports_history_list"] = list(
reports_history.find(query).sort("created_at", -1)
)
history = st.session_state["reports_history_list"]
if not history:
st.info("Chưa có báo cáo nào được lưu.")
else:
for idx, report in enumerate(history):
if st.button(
f"{report['report_data'].get('stock_code', 'N/A')} ({report['report_data'].get('report_period', 'N/A')})",
key=f"history_btn_{idx}"
):
# Update the selected report in session state
st.session_state["selected_report"] = report
st.session_state["show_form"] = False # Hide form when viewing history
# ===== RIGHT: Report View=====
with col_main:
if st.session_state.get("selected_report") and not st.session_state.get("show_form", False):
selected = st.session_state["selected_report"]
show_report(selected["report_data"], selected["summary"], selected["report_data"]["stock_code"])
# Tạo báo cáo mới
if st.button("Tạo báo cáo mới", key="new_report_btn"):
st.session_state["show_form"] = True
st.session_state.pop("selected_report", None)
st.rerun()
else:
with st.form("report_form"):
stock_code_input = st.text_input(
"Nhập mã cổ phiếu (ví dụ: VIC, HPG...)", value="HPG").upper()
col_start, col_end = st.columns(2)
with col_start:
start_date = st.date_input(
"Từ ngày", value=pd.to_datetime("2025-05-01"))
with col_end:
end_date = st.date_input("Đến ngày", value=datetime.now())
submitted = st.form_submit_button("Tạo báo cáo", use_container_width=True)
if submitted and stock_code_input:
temp = stock_code_input.strip().upper()
if not re.fullmatch(r"[A-Z0-9]{1,6}", temp):
st.error("Mã cổ phiếu không hợp lệ. Vui lòng nhập 1-6 ký tự chữ hoa hoặc số (ví dụ: HPG, VIC).")
else:
with st.spinner(f'Đang tổng hợp và phân tích dữ liệu cho mã {stock_code_input}...'):
report_data = generate_stock_report(
stock_code_input, (str(start_date), str(end_date)))
if report_data and (report_data["overall_sentiment"]["positive_mentions"] > 0 or report_data["overall_sentiment"]["negative_mentions"] > 0):
summary = call_genai_summary(
report_data, stock_code_input, (str(start_date), str(end_date)))
else:
summary = f"Không tìm thấy đủ dữ liệu nổi bật cho mã **{stock_code_input}** trong khoảng thời gian đã chọn để tạo tóm tắt AI."
# Save to MongoDB
inserted_id = reports_history.insert_one({
"uid": st.session_state["uid"],
"report_data": report_data,
"summary": summary,
"created_at": datetime.utcnow()
}).inserted_id
# Add new report to top of history
new_report = {
"_id": inserted_id,
"report_data": report_data,
"summary": summary,
"created_at": datetime.utcnow()
}
st.session_state["reports_history_list"].insert(0, new_report)
# Show new report
st.session_state["selected_report"] = new_report
st.session_state["show_form"] = False
st.rerun()
|