Update agent.py

agent.py

@@ -1,41 +1,64 @@
-from datetime import datetime
-from pipeline import fetch_data, clean_data
-import smtplib
-from email.mime.text import MIMEText
-
-# Cấu hình email
-SENDER_EMAIL = "dulichviet2406@gmail.com"
-SENDER_PASSWORD = "ppedsoakefjyrbym"  
-RECEIVER_EMAIL = "vanhai123@gmail.com"  
-
-def send_email_report(content):
-    msg = MIMEText(content)
-    msg["Subject"] = "Kết quả từ AI Agent Pipeline"
-    msg["From"] = SENDER_EMAIL
-    msg["To"] = RECEIVER_EMAIL
-
-    try:
-        with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
-            server.login(SENDER_EMAIL, SENDER_PASSWORD)
-            server.send_message(msg)
-        print("Email đã được gửi thành công.")
-    except Exception as e:
-        print("Gửi email thất bại:", e)
-
-def has_new_data():
-    return datetime.now().second % 2 == 0
-
-def run_pipeline():
-    fetch_data()
-    df_clean = clean_data()
-    return df_clean
-
-def run_agent_once():
-    if has_new_data():
-        print("Có dữ liệu mới! Đang chạy pipeline...")
-        df = run_pipeline()
-        message = f"Agent đã xử lý {len(df)} dòng dữ liệu mới."
-        send_email_report(message)
-        return message
-    else:
-        return "Chưa phát hiện dữ liệu mới."
+from datetime import datetime
+from pipeline import fetch_data, clean_data
+import smtplib
+from email.mime.text import MIMEText
+import os
+
+# ---------------------
+# ⚙️ Cấu hình email (vẫn giữ nguyên cho giáo viên kiểm tra)
+# ---------------------
+SENDER_EMAIL = "dulichviet2406@gmail.com"
+SENDER_PASSWORD = "*******************"  # App Password Gmail
+RECEIVER_EMAIL = "vanhai123@gmail.com"
+
+def send_email_report(content):
+    """
+    Gửi email thông báo kết quả xử lý.
+    ✅ Chạy được khi local có mạng.
+    ❌ Không hoạt động trên Hugging Face Spaces miễn phí do bị chặn truy cập Internet.
+    """
+    # Tắt gửi email khi đang chạy trên Hugging Face bằng biến môi trường hoặc điều kiện thủ công
+    if os.environ.get("DISABLE_EMAIL") == "true":
+        print("📭 Gửi email đã bị tắt trên môi trường online (HF Space).")
+        return
+
+    msg = MIMEText(content)
+    msg["Subject"] = "Kết quả từ AI Agent Pipeline"
+    msg["From"] = SENDER_EMAIL
+    msg["To"] = RECEIVER_EMAIL
+
+    try:
+        with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
+            server.login(SENDER_EMAIL, SENDER_PASSWORD)
+            server.send_message(msg)
+        print("✅ Email đã được gửi thành công.")
+    except Exception as e:
+        print("❌ Gửi email thất bại:", e)
+
+def has_new_data():
+    """
+    Mô phỏng phát hiện dữ liệu mới bằng cách kiểm tra nếu giây hiện tại là số chẵn.
+    """
+    return datetime.now().second % 2 == 0
+
+def run_pipeline():
+    """
+    Chạy các bước: fetch → clean → return kết quả
+    """
+    fetch_data()
+    df_clean = clean_data()
+    return df_clean
+
+def run_agent_once():
+    """
+    Hàm chính được gọi khi bấm nút 'Agent Tự Động'.
+    Nếu phát hiện có dữ liệu mới → chạy pipeline → gửi mail (nếu đang chạy local).
+    """
+    if has_new_data():
+        print("✅ Có dữ liệu mới! Đang chạy pipeline...")
+        df = run_pipeline()
+        message = f"Agent đã xử lý {len(df)} dòng dữ liệu mới."
+        send_email_report(message)
+        return message
+    else:
+        return "⏳ Chưa phát hiện dữ liệu mới."