Update src/pages/page_report.py

src/pages/page_report.py

@@ -5,6 +5,7 @@ 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")
@@ -84,34 +85,38 @@ def main():
                 submitted = st.form_submit_button("Tạo báo cáo", use_container_width=True)
 
             if submitted and stock_code_input:
-                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
+                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()