PhilipL commited on
Commit
e1a0120
·
verified ·
1 Parent(s): c7d16a3

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -201
app.py DELETED
@@ -1,201 +0,0 @@
1
- import gradio as gr
2
- import datetime
3
- import pandas as pd
4
- import joblib
5
- from sklearn.cluster import HDBSCAN
6
- import numpy as np
7
- import logger_service
8
-
9
- # --- 1. 載入模型與歷史資料 (實現動態 Rule 的關鍵) ---
10
- try:
11
- le_action = joblib.load('le_action.pkl')
12
- le_ip = joblib.load('le_ip.pkl')
13
-
14
- # 🌟 載入歷史資料作為 HDBSCAN 即時比對的基準
15
- baseline_df = pd.read_csv('baseline_logs.csv')
16
- baseline_df['hour'] = pd.to_datetime(baseline_df['timestamp']).dt.hour
17
-
18
- # 預先將歷史資料轉成特徵矩陣
19
- baseline_X = np.column_stack((
20
- baseline_df['hour'].values,
21
- le_ip.transform(baseline_df['ip_address'].values),
22
- le_action.transform(baseline_df['action'].values)
23
- ))
24
-
25
- AI_READY = True
26
- print("✅ AI 模組與歷史基準資料載入成功!")
27
- except Exception as e:
28
- AI_READY = False
29
- print(f"⚠️ AI 模組載入失敗,請確認 .pkl 與 .csv 檔案是否已上傳。錯誤: {e}")
30
-
31
- USER_DB = {
32
- "student": ["1234", "學生", "同學"],
33
- "teacher": ["1234", "老師", "教授"],
34
- "admin": ["1234", "行政", "行政人員"],
35
- "ta": ["1234", "TA", "助教"],
36
- "sysadmin": ["1234", "系統管理員", "管理員"]
37
- }
38
-
39
- log_history = baseline_df.sort_values(by='timestamp', ascending=False).head(20).to_dict('records')
40
-
41
- # --- 3. 核心邏輯與真實 HDBSCAN 異常偵測 ---
42
- def check_anomaly(ip, action, log_time):
43
- if not AI_READY:
44
- return "success"
45
-
46
- if action in ["login_attempt", "logout"]:
47
- return "success"
48
-
49
- try:
50
- # 1. 將新進來的行為轉為代碼
51
- ip_code = le_ip.transform([ip])[0]
52
- action_code = le_action.transform([action])[0]
53
-
54
- # 2. 擷取當下行為的時間 (Hour)
55
- current_hour = pd.to_datetime(log_time).hour
56
-
57
- # 3. 組合這筆新資料的特徵向量
58
- new_data_point = np.array([[current_hour, ip_code, action_code]])
59
-
60
- # 4. 🌟 將新資料與歷史資料合併 (這就是流程圖中的「即時丟入 AI 進行判斷」)
61
- combined_X = np.vstack((baseline_X, new_data_point))
62
-
63
- # 5. 執行 HDBSCAN 分群
64
- model = HDBSCAN(min_cluster_size=5, min_samples=3)
65
- clusters = model.fit_predict(combined_X)
66
-
67
- # 6. 檢查最後一筆 (也就是剛剛送進來的新動作) 是否被分類為 -1 (雜訊/異常)
68
- if clusters[-1] == -1:
69
- return "⚠️異常_作息或行為不符"
70
- else:
71
- return "success"
72
-
73
- except ValueError:
74
- # 如果遇到沒看過的 IP 或動作,連算都不用算,絕對是異常
75
- return "⚠️異常_未知特徵"
76
-
77
- def record_log(ip, cookie, account, role, action, status, time_mode, custom_time):
78
- global baseline_X, log_history
79
- # 決定紀錄的時間
80
- if time_mode == "自訂時間 (模擬過去/未來)" and custom_time.strip() != "":
81
- log_time = custom_time
82
- else:
83
- tw_tz = datetime.timezone(datetime.timedelta(hours=8))
84
- log_time = datetime.datetime.now(tw_tz).strftime("%Y-%m-%d %H:%M:%S")
85
-
86
- # 執行 AI 檢查 (現在把 log_time 也傳進去了)
87
- if status == "success":
88
- ai_judgment = check_anomaly(ip, action, log_time)
89
- if ai_judgment != "success":
90
- status = ai_judgment
91
-
92
- new_log = {
93
- "timestamp": log_time,
94
- "ip_address": ip,
95
- "cookie_id": cookie,
96
- "account": account,
97
- "role": role,
98
- "action": action,
99
- "status": status
100
- }
101
-
102
- try:
103
- # 將新行為轉為特徵數值
104
- new_hour = pd.to_datetime(log_time).hour
105
- new_ip_code = le_ip.transform([ip])[0]
106
- new_action_code = le_action.transform([action])[0]
107
- new_feature = np.array([[new_hour, new_ip_code, new_action_code]])
108
-
109
- # 🌟 把新特徵壓入全域矩陣中
110
- baseline_X = np.vstack((baseline_X, new_feature))
111
- except Exception as e:
112
- print(f"⚠️ AI 矩陣更新跳過 (可能是未知特徵): {e}")
113
-
114
- # 呼叫外部 Service 進行存檔與上傳
115
- logger_service.push_new_log(new_log)
116
-
117
- log_history.insert(0, new_log)
118
- return pd.DataFrame(log_history)
119
-
120
- def process_login(ip, cookie, account, password, captcha, time_mode, custom_time):
121
- valid_captcha = "FEIK"
122
- if account not in USER_DB: return gr.update(visible=True), gr.update(visible=False), "帳號不存在!", gr.update(), record_log(ip, cookie, account, "Unknown", "login_attempt", "failed_no_account", time_mode, custom_time)
123
- role, name = USER_DB[account][1], USER_DB[account][2]
124
- if captcha.upper() != valid_captcha: return gr.update(visible=True), gr.update(visible=False), "驗證碼錯誤!", gr.update(), record_log(ip, cookie, account, role, "login_attempt", "failed_captcha", time_mode, custom_time)
125
- if USER_DB[account][0] == password:
126
- welcome_msg = f"**{name}** 歡迎您 | 線上人數: 938"
127
- df = record_log(ip, cookie, account, role, "login_attempt", "success", time_mode, custom_time)
128
- return gr.update(visible=False), gr.update(visible=True), "", welcome_msg, df
129
- return gr.update(visible=True), gr.update(visible=False), "密碼錯誤!", gr.update(), record_log(ip, cookie, account, role, "login_attempt", "failed_password", time_mode, custom_time)
130
-
131
- def simulate_click(ip, cookie, account, button_name, time_mode, custom_time):
132
- role = USER_DB.get(account, ["", "Unknown"])[1]
133
- return record_log(ip, cookie, account, role, button_name, "success", time_mode, custom_time)
134
-
135
- def logout(ip, cookie, account, time_mode, custom_time):
136
- role = USER_DB.get(account, ["", "Unknown"])[1]
137
- df = record_log(ip, cookie, account, role, "logout", "success", time_mode, custom_time)
138
- return gr.update(visible=True), gr.update(visible=False), "", "", "", "", df
139
-
140
- def toggle_time_input(mode):
141
- return gr.update(visible=(mode == "自訂時間 (模擬過去/未來)"))
142
-
143
- # --- 4. UI 介面設計 ---
144
- custom_css = """
145
- .mock-panel { background-color: var(--background-fill-secondary); padding: 15px; border-radius: 8px; border: 1px dashed var(--border-color-primary); }
146
- .login-container { max-width: 500px; margin: 0 auto; padding: 20px; }
147
- .placeholder-box { background-color: var(--background-fill-secondary); border: 2px dashed var(--border-color-primary); text-align: center; color: var(--body-text-color); padding: 40px 0; margin-bottom: 15px; border-radius: 5px; }
148
- .ntut-title { color: #d9534f; font-size: 22px; font-weight: bold; text-align: center; margin-bottom: 20px;}
149
- """
150
-
151
- with gr.Blocks(title="模擬校園入口與資安防護系統", css=custom_css) as demo:
152
- with gr.Row():
153
- with gr.Column(scale=1, elem_classes="mock-panel"):
154
- gr.Markdown("### ⚙️ 環境變數模擬器")
155
- mock_ip = gr.Dropdown(choices=["140.124.71.55 (校內預設)", "140.124.18.22 (宿舍)", "45.33.22.11 (國外異常IP)"], value="140.124.71.55 (校內預設)", allow_custom_value=True, label="模擬來源 IP")
156
- mock_cookie = gr.Textbox(label="模擬 Cookie Session ID", value="sess_baseline")
157
- gr.Markdown("---")
158
- time_mode = gr.Radio(label="時間設定模式", choices=["真實時間 (目前時間)", "自訂時間 (模擬過去/未來)"], value="真實時間 (目前時間)")
159
- custom_time_input = gr.Textbox(label="輸入自訂時間", placeholder="格式: YYYY-MM-DD HH:MM:SS", value="2026-03-17 03:00:00", visible=False)
160
- gr.Markdown("---")
161
- gr.Markdown("### 📊 即時 Log 與 AI 判定區")
162
- log_display = gr.Dataframe(headers=["timestamp", "ip_address", "cookie_id", "account", "role", "action", "status"], datatype=["str", "str", "str", "str", "str", "str", "str"], interactive=False, wrap=True)
163
-
164
- with gr.Column(scale=2):
165
- with gr.Column(visible=True, elem_classes="login-container") as login_view:
166
- gr.HTML("<div class='placeholder-box'>預留位置:台北科大 Logo 與 e化校園圖片</div>")
167
- gr.Markdown("<div class='ntut-title'>校園入口網站 Taipei Tech Portal</div>")
168
- login_msg = gr.Markdown(value="", visible=True)
169
- with gr.Row(): acc_input = gr.Textbox(label="使用者帳號 (Account)", placeholder="例如: student, admin...")
170
- with gr.Row(): pwd_input = gr.Textbox(label="使用者密碼 (Password)", type="password", placeholder="密碼皆為 1234")
171
- with gr.Row(): gr.HTML("<div style='background:var(--background-fill-secondary); color:var(--body-text-color); padding:10px; font-weight:bold; letter-spacing: 3px; border:1px solid var(--border-color-primary); text-align:center;'>F E I K</div>")
172
- with gr.Row(): captcha_input = gr.Textbox(label="請輸入驗證碼 (Keyin Code)")
173
- login_btn = gr.Button("登入 Login", variant="primary")
174
-
175
- with gr.Column(visible=False) as dashboard_view:
176
- gr.HTML("<div class='placeholder-box'>預留位置:學生學習歷程入口網站橫幅</div>")
177
- with gr.Row():
178
- gr.Markdown("### 資訊系統")
179
- welcome_text = gr.Markdown(value="", elem_classes="text-right")
180
- logout_btn = gr.Button("登出", size="sm")
181
- gr.Markdown("---")
182
- with gr.Accordion("▼ 1. 教務系統", open=True):
183
- with gr.Row(): btn_eval = gr.Button("▶ 期末網路教學評量", size="sm"); btn_pre_select = gr.Button("▶ 期末網路預選系統", size="sm"); btn_add_drop = gr.Button("▶ 開學後加退選系統", size="sm")
184
- with gr.Row(): btn_istudy = gr.Button("▶ 北科i學園PLUS", size="sm"); btn_card = gr.Button("▶ 學生證掛失及補發系統", size="sm"); btn_course = gr.Button("▶ 課程系統", size="sm")
185
- with gr.Row(): btn_score = gr.Button("▶ 學業成績查詢系統", size="sm")
186
- with gr.Accordion("▼ 2. 學務系統", open=True):
187
- with gr.Row(): btn_leave = gr.Button("▶ 學生請假系統", size="sm"); btn_dorm = gr.Button("▶ 學生宿舍登錄(抽籤)系統", size="sm"); btn_scholarship = gr.Button("▶ 獎助學金申請系統", size="sm")
188
- with gr.Accordion("▼ 3. 資訊服務", open=True):
189
- with gr.Row(): btn_webmail = gr.Button("▶ 網路郵局 WebMail", size="sm"); btn_vdesk = gr.Button("▶ 北科軟體雲", size="sm")
190
- with gr.Accordion("▼ 惡意操作測試區", open=True):
191
- btn_hack_score = gr.Button("💀 嘗試偷改成績 (SQL Injection)", variant="stop", size="sm")
192
-
193
- time_mode.change(fn=toggle_time_input, inputs=time_mode, outputs=custom_time_input)
194
- login_btn.click(fn=process_login, inputs=[mock_ip, mock_cookie, acc_input, pwd_input, captcha_input, time_mode, custom_time_input], outputs=[login_view, dashboard_view, login_msg, welcome_text, log_display])
195
- logout_btn.click(fn=logout, inputs=[mock_ip, mock_cookie, acc_input, time_mode, custom_time_input], outputs=[login_view, dashboard_view, acc_input, pwd_input, captcha_input, login_msg, log_display])
196
-
197
- for btn, action_name in [(btn_eval, "click_eval_system"), (btn_pre_select, "click_pre_select_system"), (btn_add_drop, "click_add_drop_system"), (btn_istudy, "click_istudy"), (btn_card, "click_card_loss"), (btn_course, "click_course_system"), (btn_score, "click_score_query"), (btn_leave, "click_leave_system"), (btn_dorm, "click_dorm_system"), (btn_scholarship, "click_scholarship"), (btn_webmail, "click_webmail"), (btn_vdesk, "click_vdesk"), (btn_hack_score, "malicious_sql_injection")]:
198
- btn.click(fn=lambda ip, c, acc, t_mode, t_custom, n=action_name: simulate_click(ip, c, acc, n, t_mode, t_custom), inputs=[mock_ip, mock_cookie, acc_input, time_mode, custom_time_input], outputs=[log_display])
199
-
200
- if __name__ == "__main__":
201
- demo.launch()