Update app.py

app.py

@@ -38,7 +38,7 @@ USER_DB = {
 
 log_history = []
 
-# --- 2. 核心邏輯與真實 HDBSCAN 異常偵測 ---
+# --- 3. 核心邏輯與真實 HDBSCAN 異常偵測 ---
 def check_anomaly(ip, action, log_time, overall_w, time_w, ip_w, dev_w):
     if not AI_READY: return "success" 
     if action in ["login_attempt", "logout"]: return "success"
@@ -143,7 +143,6 @@ def record_log(ip, cookie, account, role, action, status, time_mode, custom_time
 
     return pd.DataFrame(log_history)
 
-# --- 3. 視角切換與登入控制邏輯 ---
 def fast_login(target_account, ip, cookie, time_mode, custom_time, ow, tw, iw, dw):
     role, name = USER_DB[target_account][1], USER_DB[target_account][2]
     df = record_log(ip, cookie, target_account, role, "login_attempt", "success", time_mode, custom_time, ow, tw, iw, dw)
@@ -151,7 +150,6 @@ def fast_login(target_account, ip, cookie, time_mode, custom_time, ow, tw, iw, d
     welcome_msg = f"**{name}** 歡迎您  |  線上人數: 938 (⚡快速登入模式)"
     target_tab = "student_tab" if role == "學生" else "admin_tab"
     
-    # 回傳切換畫面與選定的 Tab
     return gr.update(visible=False), gr.update(visible=True), gr.update(selected=target_tab), "", welcome_msg, df, target_account
 
 def process_login(ip, cookie, account, password, captcha, time_mode, custom_time, ow, tw, iw, dw):
@@ -172,31 +170,24 @@ def process_login(ip, cookie, account, password, captcha, time_mode, custom_time
     if USER_DB[account][0] == password:
         welcome_msg = f"**{name}** 歡迎您  |  線上人數: 938"
         df = record_log(ip, cookie, account, role, "login_attempt", "success", time_mode, custom_time, ow, tw, iw, dw)
-        
         target_tab = "student_tab" if role == "學生" else "admin_tab"
+        
         return gr.update(visible=False), gr.update(visible=True), gr.update(selected=target_tab), "", welcome_msg, df
         
     df = record_log(ip, cookie, account, role, "login_attempt", "failed_password", time_mode, custom_time, ow, tw, iw, dw)
     return fail_response("密碼錯誤！", df)
 
-def toggle_role(current_account, ip, cookie, time_mode, custom_time, ow, tw, iw, dw):
-    """單鍵切換身分邏輯"""
-    # 判斷當前是誰，切換成另一個
-    target_account = "student" if current_account == "admin" else "admin"
-    
+def toggle_role(current_tab, ip, cookie, time_mode, custom_time, ow, tw, iw, dw):
+    # 根據目前的 Tab 自動判定要切換成誰
+    target_account = "admin" if current_tab == "student_tab" else "student"
     role, name = USER_DB[target_account][1], USER_DB[target_account][2]
     df = record_log(ip, cookie, target_account, role, "login_attempt", "success", time_mode, custom_time, ow, tw, iw, dw)
     
-    welcome_msg = f"**{name}** 歡迎您  |  線上人數: 938 (⚡單鍵切換模式)"
-    target_tab = "student_tab" if role == "學生" else "admin_tab"
+    welcome_msg = f"**{name}** 歡迎您  |  線上人數: 938 (⚡快速切換模式)"
+    target_tab = "admin_tab" if current_tab == "student_tab" else "student_tab"
     
     return gr.update(selected=target_tab), welcome_msg, df, target_account
 
-def logout(ip, cookie, account, time_mode, custom_time, ow, tw, iw, dw):
-    role = USER_DB.get(account, ["", "Unknown"])[1]
-    df = record_log(ip, cookie, account, role, "logout", "success", time_mode, custom_time, ow, tw, iw, dw)
-    return gr.update(visible=True), gr.update(visible=False), "", "", "", "", df
-
 def simulate_click(ip, cookie, account, button_name, time_mode, custom_time, ow, tw, iw, dw):
     role = USER_DB.get(account, ["", "Unknown"])[1]
     return record_log(ip, cookie, account, role, button_name, "success", time_mode, custom_time, ow, tw, iw, dw)
@@ -205,6 +196,11 @@ def test_anomaly(ip, cookie, account, action_name, test_time_str, ow, tw, iw, dw
     role = USER_DB.get(account, ["", "Unknown"])[1]
     return record_log(ip, cookie, account, role, action_name, "success", "自訂時間 (模擬過去/未來)", test_time_str, ow, tw, iw, dw)
 
+def logout(ip, cookie, account, time_mode, custom_time, ow, tw, iw, dw):
+    role = USER_DB.get(account, ["", "Unknown"])[1]
+    df = record_log(ip, cookie, account, role, "logout", "success", time_mode, custom_time, ow, tw, iw, dw)
+    return gr.update(visible=True), gr.update(visible=False), gr.update(), "", "", "", "", df
+
 def toggle_time_input(mode):
     return gr.update(visible=(mode == "自訂時間 (模擬過去/未來)"))
 
@@ -215,7 +211,7 @@ custom_css = """
 .ntut-title { color: #d9534f; font-size: 22px; font-weight: bold; text-align: center; margin-bottom: 20px;}
 .weight-card { padding: 10px; background: white; border-radius: 8px; box-shadow: 0 1px 3px rgba(0,0,0,0.1); margin-top: 10px; }
 
-/* 🌟 新增：隱藏 Tabs 標題列的魔法 CSS */
+/* 🌟 新增：隱藏 Tabs 標題列的語法 */
 .headless-tabs > div:first-child { display: none !important; }
 .headless-tabs { border: none !important; background: transparent !important; }
 """
@@ -255,15 +251,15 @@ with gr.Blocks(title="模擬校園入口與資安防護系統", css=custom_css)
                     welcome_text = gr.Markdown(value="", elem_classes="text-right")
                     logout_btn = gr.Button("登出", size="sm")
                 
-                # 🌟 單一切換按鈕放置區
+                # 🌟 新增：單一切換按鈕
                 with gr.Row():
-                    btn_toggle_role = gr.Button("🔄 單鍵切換身分 (Student ↔ Admin)", size="sm", variant="secondary")
+                    btn_toggle_role = gr.Button("Student ↔ Admin", size="sm", variant="secondary")
+                
                 gr.Markdown("---")
                 
-                # 🌟 加上 headless-tabs 的 CSS 類別
+                # 🌟 修改：改用 Tabs 排版並掛上 headless-tabs 樣式
                 with gr.Tabs(selected="student_tab", elem_classes="headless-tabs") as role_tabs:
-                    
-                    with gr.Tab("👨‍🎓 學生專區", id="student_tab"):
+                    with gr.Tab("student", id="student_tab"):
                         with gr.Accordion("▼ 1. 教務系統", open=True):
                             with gr.Row(): btn_eval = gr.Button("▶ 期末網路教學評量", size="sm"); btn_pre_select = gr.Button("▶ 期末網路預選系統", size="sm"); btn_add_drop = gr.Button("▶ 開學後加退選系統", size="sm")
                             with gr.Row(): btn_istudy = gr.Button("▶ 北科i學園PLUS", size="sm"); btn_card = gr.Button("▶ 學生證掛失及補發系統", size="sm"); btn_course = gr.Button("▶ 課程系統", size="sm")
@@ -275,7 +271,7 @@ with gr.Blocks(title="模擬校園入口與資安防護系統", css=custom_css)
                         with gr.Accordion("▼ 惡意操作測試區", open=True):
                             btn_hack_score = gr.Button("💀 嘗試偷改成績 (SQL Injection)", variant="stop", size="sm")
 
-                    with gr.Tab("🛡️ 管理員專區", id="admin_tab"):
+                    with gr.Tab("admin", id="admin_tab"):
                         with gr.Accordion("▼ 防護強度微調 (AI 敏感度)", open=True, elem_classes="weight-card"):
                             gr.Markdown("<span style='color:gray; font-size:0.9em;'>手動輸入 0~100 的數值。提高強度會增加誤報率，但能更有效攔截新型攻擊。</span>")
                             with gr.Row():
@@ -290,33 +286,44 @@ with gr.Blocks(title="模擬校園入口與資安防護系統", css=custom_css)
                             
                             gr.Markdown("#### 1. 選課相關系統 (具備強烈季節性)")
                             with gr.Row():
-                                t_pre_sel_norm = gr.Button("✅ 正常: 期末預選 (06-13 11:00)", size="sm"); t_pre_sel_anom = gr.Button("⚠️ 異常: 期末預選 (10-15 13:00)", size="sm")
+                                t_pre_sel_norm = gr.Button("✅ 正常: 期末網路預選系統 (06-13 11:00)", size="sm")
+                                t_pre_sel_anom = gr.Button("⚠️ 異常: 期末網路預選系統 (10-15 13:00)", size="sm")
                             with gr.Row():
-                                t_course_norm = gr.Button("✅ 正常: 課程系統 (02-25 07:50)", size="sm"); t_course_anom = gr.Button("⚠️ 異常: 課程系統 (07-15 05:00)", size="sm")
+                                t_course_norm = gr.Button("✅ 正常: 課程系統 (02-25 07:50)", size="sm")
+                                t_course_anom = gr.Button("⚠️ 異常: 課程系統 (07-15 05:00)", size="sm")
                             with gr.Row():
-                                t_add_drop_norm = gr.Button("✅ 正常: 開學加退選 (02-18 14:00)", size="sm"); t_add_drop_anom = gr.Button("⚠️ 異常: 開學加退選 (05-15 12:00)", size="sm")
+                                t_add_drop_norm = gr.Button("✅ 正常: 開學後加退選系統 (02-18 14:00)", size="sm")
+                                t_add_drop_anom = gr.Button("⚠️ 異常: 開學後加退選系統 (05-15 12:00)", size="sm")
                                 
                             gr.Markdown("#### 2. 期末與成績相關 (具備週期性)")
                             with gr.Row():
-                                t_eval_norm = gr.Button("✅ 正常: 教學評量 (12-17 13:20)", size="sm"); t_eval_anom = gr.Button("⚠️ 異常: 教學評量 (03-15 03:00)", size="sm")
+                                t_eval_norm = gr.Button("✅ 正常: 期末網路教學評量 (12-17 13:20)", size="sm")
+                                t_eval_anom = gr.Button("⚠️ 異常: 期末網路教學評量 (03-15 03:00)", size="sm")
                             with gr.Row():
-                                t_score_norm = gr.Button("✅ 正常: 成績查詢 (11-17 18:50)", size="sm"); t_score_anom = gr.Button("⚠️ 異常: 成績查詢 (02-15 03:00)", size="sm")
+                                t_score_norm = gr.Button("✅ 正常: 學業成績查詢系統 (11-17 18:50)", size="sm")
+                                t_score_anom = gr.Button("⚠️ 異常: ��業成績查詢系統 (02-15 03:00)", size="sm")
                                 
                             gr.Markdown("#### 3. 校務與行政服務 (生活與常規)")
                             with gr.Row():
-                                t_dorm_norm = gr.Button("✅ 正常: 宿舍抽籤 (07-21 23:10)", size="sm"); t_dorm_anom = gr.Button("⚠️ 異常: 宿舍抽籤 (11-15 16:00)", size="sm")
+                                t_dorm_norm = gr.Button("✅ 正常: 學生宿舍登錄(抽籤)系統 (07-21 23:10)", size="sm")
+                                t_dorm_anom = gr.Button("⚠️ 異常: 學生宿舍登錄(抽籤)系統 (11-15 16:00)", size="sm")
                             with gr.Row():
-                                t_schol_norm = gr.Button("✅ 正常: 獎學金申請 (03-12 14:00)", size="sm"); t_schol_anom = gr.Button("⚠️ 異常: 獎學金申請 (08-15 02:00)", size="sm")
+                                t_schol_norm = gr.Button("✅ 正常: 獎助學金申請系統 (03-12 14:00)", size="sm")
+                                t_schol_anom = gr.Button("⚠️ 異常: 獎助學金申請系統 (08-15 02:00)", size="sm")
                             with gr.Row():
-                                t_card_norm = gr.Button("✅ 正常: 學生證掛失 (10-23 13:40)", size="sm"); t_card_anom = gr.Button("⚠️ 異常: 學生證掛失 (08-10 03:00)", size="sm")
+                                t_card_norm = gr.Button("✅ 正常: 學生證掛失 (10-23 13:40)", size="sm")
+                                t_card_anom = gr.Button("⚠️ 異常: 學生證掛失 (08-10 03:00)", size="sm")
                                 
                             gr.Markdown("#### 4. 學習與日常輔助系統 (具備作息規律)")
                             with gr.Row():
-                                t_istudy_norm = gr.Button("✅ 正常: i學園PLUS (10-18 16:00)", size="sm"); t_istudy_anom = gr.Button("⚠️ 異常: i學園PLUS (08-20 02:00)", size="sm")
+                                t_istudy_norm = gr.Button("✅ 正常: 北科i學園PLUS (10-18 16:00)", size="sm")
+                                t_istudy_anom = gr.Button("⚠️ 異常: 北科i學園PLUS (08-20 02:00)", size="sm")
                             with gr.Row():
-                                t_leave_norm = gr.Button("✅ 正常: 學生請假 (03-29 08:45)", size="sm"); t_leave_anom = gr.Button("⚠️ 異常: 學生請假 (08-15 13:00)", size="sm")
+                                t_leave_norm = gr.Button("✅ 正常: 學生請假系統 (03-29 08:45)", size="sm")
+                                t_leave_anom = gr.Button("⚠️ 異常: 學生請假系統 (08-15 13:00)", size="sm")
                             with gr.Row():
-                                t_vdesk_norm = gr.Button("✅ 正常: 軟體雲/WebMail (03-22 18:00)", size="sm"); t_vdesk_anom = gr.Button("⚠️ 異常: 軟體雲/WebMail (08-10 00:00)", size="sm")
+                                t_vdesk_norm = gr.Button("✅ 正常: 軟體雲/WebMail (03-22 18:00)", size="sm")
+                                t_vdesk_anom = gr.Button("⚠️ 異常: 軟體雲/WebMail (08-10 00:00)", size="sm")
 
     # --- 綁定事件 ---
     time_mode.change(fn=toggle_time_input, inputs=time_mode, outputs=custom_time_input)
@@ -328,22 +335,22 @@ with gr.Blocks(title="模擬校園入口與資安防護系統", css=custom_css)
         inputs=[mock_ip, mock_cookie, acc_input, pwd_input, captcha_input, time_mode, custom_time_input] + weight_inputs, 
         outputs=[login_view, dashboard_view, role_tabs, login_msg, welcome_text, log_display] 
     )
-    
     logout_btn.click(
         fn=logout, 
         inputs=[mock_ip, mock_cookie, acc_input, time_mode, custom_time_input] + weight_inputs, 
-        outputs=[login_view, dashboard_view, acc_input, pwd_input, captcha_input, login_msg, log_display] 
+        outputs=[login_view, dashboard_view, role_tabs, acc_input, pwd_input, captcha_input, login_msg, log_display] 
     )
 
     fast_inputs = [mock_ip, mock_cookie, time_mode, custom_time_input] + weight_inputs
-    
-    btn_fast_student.click(fn=lambda ip, c, tm, ct, ow, tw, iw, dw: fast_login("student", ip, c, tm, ct, ow, tw, iw, dw), inputs=fast_inputs, outputs=[login_view, dashboard_view, role_tabs, login_msg, welcome_text, log_display, acc_input])
-    btn_fast_admin.click(fn=lambda ip, c, tm, ct, ow, tw, iw, dw: fast_login("admin", ip, c, tm, ct, ow, tw, iw, dw), inputs=fast_inputs, outputs=[login_view, dashboard_view, role_tabs, login_msg, welcome_text, log_display, acc_input])
+    fast_outputs = [login_view, dashboard_view, role_tabs, login_msg, welcome_text, log_display, acc_input] 
 
-    # 🌟 單鍵切換按鈕綁定
+    btn_fast_student.click(fn=lambda ip, c, tm, ct, ow, tw, iw, dw: fast_login("student", ip, c, tm, ct, ow, tw, iw, dw), inputs=fast_inputs, outputs=fast_outputs)
+    btn_fast_admin.click(fn=lambda ip, c, tm, ct, ow, tw, iw, dw: fast_login("admin", ip, c, tm, ct, ow, tw, iw, dw), inputs=fast_inputs, outputs=fast_outputs)
+    
+    # 🌟 綁定單一切換按���，把 role_tabs 當作 input 傳入判斷當前狀態
     btn_toggle_role.click(
         fn=toggle_role, 
-        inputs=[acc_input, mock_ip, mock_cookie, time_mode, custom_time_input] + weight_inputs, 
+        inputs=[role_tabs, mock_ip, mock_cookie, time_mode, custom_time_input] + weight_inputs, 
         outputs=[role_tabs, welcome_text, log_display, acc_input]
     )
 
@@ -365,17 +372,31 @@ with gr.Blocks(title="模擬校園入口與資安防護系統", css=custom_css)
         )
         
     test_mapping = [
-        (t_pre_sel_norm, "click_pre_select_system", "2026-06-13 11:00:00"), (t_pre_sel_anom, "click_pre_select_system", "2026-10-15 13:00:00"),
-        (t_course_norm, "click_course_system", "2026-02-25 07:50:00"), (t_course_anom, "click_course_system", "2026-07-15 05:00:00"),
-        (t_add_drop_norm, "click_add_drop_system", "2026-02-18 14:00:00"), (t_add_drop_anom, "click_add_drop_system", "2026-05-15 12:00:00"),
-        (t_eval_norm, "click_eval_system", "2026-12-17 13:20:00"), (t_eval_anom, "click_eval_system", "2026-03-15 03:00:00"),
-        (t_score_norm, "click_score_query", "2026-11-17 18:50:00"), (t_score_anom, "click_score_query", "2026-02-15 03:00:00"),
-        (t_dorm_norm, "click_dorm_system", "2026-07-21 23:10:00"), (t_dorm_anom, "click_dorm_system", "2026-11-15 16:00:00"),
-        (t_schol_norm, "click_scholarship", "2026-03-12 14:00:00"), (t_schol_anom, "click_scholarship", "2026-08-15 02:00:00"),
-        (t_card_norm, "click_card_loss", "2026-10-23 13:40:00"), (t_card_anom, "click_card_loss", "2026-08-10 03:00:00"),
-        (t_istudy_norm, "click_istudy", "2026-10-18 16:00:00"), (t_istudy_anom, "click_istudy", "2026-08-20 02:00:00"),
-        (t_leave_norm, "click_leave_system", "2026-03-29 08:45:00"), (t_leave_anom, "click_leave_system", "2026-08-15 13:00:00"),
-        (t_vdesk_norm, "click_vdesk", "2026-03-22 18:00:00"), (t_vdesk_anom, "click_vdesk", "2026-08-10 00:00:00")
+        (t_pre_sel_norm, "click_pre_select_system", "2026-06-13 11:00:00"),
+        (t_pre_sel_anom, "click_pre_select_system", "2026-10-15 13:00:00"),
+        (t_course_norm, "click_course_system", "2026-02-25 07:50:00"),
+        (t_course_anom, "click_course_system", "2026-07-15 05:00:00"),
+        (t_add_drop_norm, "click_add_drop_system", "2026-02-18 14:00:00"),
+        (t_add_drop_anom, "click_add_drop_system", "2026-05-15 12:00:00"),
+        
+        (t_eval_norm, "click_eval_system", "2026-12-17 13:20:00"),
+        (t_eval_anom, "click_eval_system", "2026-03-15 03:00:00"),
+        (t_score_norm, "click_score_query", "2026-11-17 18:50:00"),
+        (t_score_anom, "click_score_query", "2026-02-15 03:00:00"),
+        
+        (t_dorm_norm, "click_dorm_system", "2026-07-21 23:10:00"),
+        (t_dorm_anom, "click_dorm_system", "2026-11-15 16:00:00"),
+        (t_schol_norm, "click_scholarship", "2026-03-12 14:00:00"),
+        (t_schol_anom, "click_scholarship", "2026-08-15 02:00:00"),
+        (t_card_norm, "click_card_loss", "2026-10-23 13:40:00"),
+        (t_card_anom, "click_card_loss", "2026-08-10 03:00:00"),
+        
+        (t_istudy_norm, "click_istudy", "2026-10-18 16:00:00"),
+        (t_istudy_anom, "click_istudy", "2026-08-20 02:00:00"),
+        (t_leave_norm, "click_leave_system", "2026-03-29 08:45:00"),
+        (t_leave_anom, "click_leave_system", "2026-08-15 13:00:00"),
+        (t_vdesk_norm, "click_vdesk", "2026-03-22 18:00:00"),
+        (t_vdesk_anom, "click_vdesk", "2026-08-10 00:00:00")
     ]
     
     for btn, action_name, test_time_str in test_mapping: