Update app.py

app.py

@@ -65,22 +65,14 @@ def get_ip_category(ip_str):
 try:
     le_action = joblib.load('le_action.pkl')
     
-    print("📂 正在處理靜態訓練基底資料...")
-    static_file = 'static_train_data.csv'
-    
-    if os.path.exists('baseline_logs.csv') and os.path.getsize('baseline_logs.csv') > 500000:
-        if not os.path.exists(static_file):
-            shutil.copy('baseline_logs.csv', static_file)
-            print(f"🛡️ 系統防護：已自動將 5 萬筆訓練資料備份為 '{static_file}'")
-            
-    if os.path.exists(static_file): base_df = pd.read_csv(static_file)
-    elif os.path.exists('baseline_logs (1).csv'): base_df = pd.read_csv('baseline_logs (1).csv')
-    else: base_df = pd.DataFrame(columns=["timestamp", "ip_address", "cookie_id", "account", "role", "action", "status"])
-
-    print("🔄 正在從 Hugging Face 雲端同步最新動態歷史資料...")
-    dynamic_df = logger_service.sync_from_hf()
-    
-    baseline_df = pd.concat([base_df, dynamic_df], ignore_index=True) if not dynamic_df.empty else base_df.copy()
+    print("📂 正在讀取本地歷史資料庫...")
+    # 👉 修改點 1: 直接讀取本地的 baseline_logs.csv，不再向雲端同步
+    if os.path.exists('baseline_logs.csv'): 
+        baseline_df = pd.read_csv('baseline_logs.csv')
+    else: 
+        # 如果檔案不存在，建立帶有標題的空 DataFrame 並存檔
+        baseline_df = pd.DataFrame(columns=["timestamp", "ip_address", "cookie_id", "account", "role", "action", "status"])
+        baseline_df.to_csv('baseline_logs.csv', index=False)
 
     baseline_df['timestamp'] = pd.to_datetime(baseline_df['timestamp'])
     baseline_df['clean_ip'] = baseline_df['ip_address'].astype(str).apply(lambda x: x.split(" ")[0])
@@ -360,6 +352,10 @@ def record_log(ip, cookie, account, role, action, status, time_mode, custom_time
         if ai_judgment != "success": status = ai_judgment
 
     new_log = {"timestamp": log_time, "ip_address": ip, "cookie_id": cookie, "account": account, "role": role, "action": action, "status": status}
+
+    new_df = pd.DataFrame([new_log])
+    new_df.to_csv('baseline_logs.csv', mode='a', header=not os.path.exists('baseline_logs.csv'), index=False)
+    
     try: logger_service.push_new_log(new_log)
     except: pass