Update app.py

app.py

@@ -1362,14 +1362,12 @@ def fetch_live_government_news():
     """Tự động kết nối và lấy dữ liệu tin tức mới nhất từ cổng thông tin"""
     news_list = []
     try:
-        # Cào dữ liệu thực tế từ chuyên mục hướng tới Chính phủ số / Dịch vụ công công nghệ mới
         url = "https://baochinhphu.vn/tien-toi-chinh-phu-so.htm"
         headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'}
         
         response = requests.get(url, headers=headers, verify=False, timeout=5)
         if response.status_code == 200:
             soup = BeautifulSoup(response.text, 'html.parser')
-            # Lấy ra 3 bài báo mới nhất dựa trên cấu trúc thẻ HTML của baochinhphu
             stories = soup.find_all('div', class_='story', limit=3)
             
             for story in stories:
@@ -1389,7 +1387,6 @@ def fetch_live_government_news():
     except Exception as e:
         print(f"Lỗi kết nối đồng bộ tin tức trực tuyến: {e}")
         
-    # Cơ chế dự phòng an toàn nếu mất kết nối mạng hoặc trang nguồn thay đổi cấu trúc thẻ HTML
     if len(news_list) < 3:
         news_list = [
             {"title": "Đẩy mạnh số hóa 100% kết quả thủ tục hành chính công", "desc": "Kế hoạch hành động thúc đẩy toàn bộ các đơn vị ứng dụng trợ lý AI nhằm giảm thiểu tối đa thời gian chờ cấp trả giấy tờ trực tiếp cho người dân...", "link": "#"},
@@ -1416,14 +1413,14 @@ def update_news_interface():
     return html_code
 
 # =====================================================================
-# BACKEND FUNCTIONS (GIỮ NGUYÊN CẤU TRÚC LOGIC CỦA BẠN)
+# BACKEND FUNCTIONS (CỦA BẠN)
 # =====================================================================
 admin_procedures = list(DICHVUCONG_DATA.values())
 
 def get_procedures_list():
     lines = ["| Mã | Tên thủ tục | Cơ quan |"]
     lines.append("|---|---|---|")
-    for p in admin_procedures[:20]: # Limit to 20 for display
+    for p in admin_procedures[:20]:
         lines.append(f"| {p['code']} | {p['name']} | {p['coquan']} |")
     return "\n".join(lines)
 
@@ -1452,7 +1449,6 @@ def process_chat_pro(audio, text, history, persona="thuong"):
         result = ai_process_pro(query, use_memory=True)
         response = f"""{result['response']}\n\n--- \n _⏱️ Xử lý: {result['process_time']}s | 📄 Tài liệu: {result['docs_count']} | 🌡️ Temp: {result['temperature']}_"""
         
-        # Cập nhật history theo định dạng Tuple [[user, bot]] cho bản Gradio cũ
         history.append((query, response))
         
         audio_out = None
@@ -1489,21 +1485,18 @@ def change_persona(new_persona):
 # CẤU HÌNH GIAO DIỆN & CUSTOM CSS
 # =====================================================================
 custom_css = """
-/* Reset font & nền */
 .gradio-container {
     font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif !important;
     background-color: #f8f9fa !important; 
 }
-/* Ẩn footer mặc định */
 footer {display: none !important;}
 
-/* Thanh Topbar chuyên nghiệp (Gradient Xanh Navy) */
 .admin-topbar {
     background: linear-gradient(135deg, #1e3c72 0%, #2a5298 100%);
     padding: 20px 30px;
     display: flex;
     align-items: center;
-    margin: -20px -20px 20px -20px; /* Tràn viền */
+    margin: -20px -20px 20px -20px;
     box-shadow: 0 4px 12px rgba(0,0,0,0.1);
     color: white;
 }
@@ -1528,7 +1521,6 @@ footer {display: none !important;}
     margin: 5px 0 0 0 !important;
     font-weight: 300 !important;
 }
-/* Style thẻ tin tức cho Tab Trang chủ */
 .news-card {
     background: #ffffff;
     padding: 14px;
@@ -1537,7 +1529,6 @@ footer {display: none !important;}
     border-radius: 6px;
     box-shadow: 0 2px 8px rgba(0,0,0,0.04);
 }
-/* Khung hướng dẫn bo góc chân trang */
 .guide-box {
     background: #ffffff;
     padding: 15px 20px;
@@ -1546,7 +1537,6 @@ footer {display: none !important;}
     box-shadow: 0 2px 8px rgba(0,0,0,0.04);
     margin-top: 5px;
 }
-/* Tinh chỉnh Nút bấm */
 button.primary {
     background-color: #0056b3 !important;
     color: white !important;
@@ -1559,14 +1549,12 @@ button.primary:hover {
     background-color: #004494 !important;
     box-shadow: 0 4px 8px rgba(0,86,179,0.3) !important;
 }
-/* Tùy chỉnh Khung Chat */
 .main-chatbot {
     border: 1px solid #dee2e6 !important;
     border-radius: 8px !important;
     background-color: #ffffff !important;
     box-shadow: 0 2px 15px rgba(0,0,0,0.03) !important;
 }
-/* Tiêu đề Box */
 .section-title {
     color: #1e3c72;
     font-weight: 600;
@@ -1575,10 +1563,8 @@ button.primary:hover {
     padding-bottom: 5px;
 }"""
 
-# Khởi tạo giao diện với Theme base màu xanh thanh lịch
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), css=custom_css) as demo:
         
-    # 1. TOPBAR CHUYÊN NGHIỆP (Gradient Xanh Navy)
     gr.HTML("""
     <div class="admin-topbar">
         <div class="admin-icon">🏛️</div>
@@ -1592,16 +1578,13 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
     with gr.Tabs():
         
         # ------------------------------------------------------------------
-        # TAB 1: TRANG CHỦ (TIN TỨC Ở TRÊN NẰM NGANG - HƯỚNG DẪN Ở DƯỚI)
+        # TAB 1: TRANG CHỦ
         # ------------------------------------------------------------------
         with gr.Tab("🏠 Trang chủ"):
-            # 1. Khối tin tức trực tuyến từ Chính Phủ: Sắp xếp nằm ngang toàn bộ trang web ngay sát dưới thanh taskbar
             gr.HTML("<div class='section-title'>📰 BẢN TIN ĐIỆN TỬ DỊCH VỤ CÔNG QUỐC GIA (CẬP NHẬT TỰ ĐỘNG)</div>")
             live_news_box = gr.HTML(value="<p style='font-size:13px; color:#666;'>🔄 Đang đồng bộ thông tin mới nhất từ Cổng Dịch vụ công Quốc gia...</p>")
             
-            # 2. Khối hướng dẫn sử dụng: Di chuyển xuống dưới cùng chiếm toàn bộ không gian ngang
             gr.HTML("<div class='section-title' style='margin-top: 15px;'>📖 HƯỚNG DẪN SỬ DỤNG HỆ THỐNG TRỢ LÝ ẢO</div>")
-            
             with gr.Group(elem_classes="guide-box"):
                 gr.Markdown("""
                 Chào mừng bạn đến với **Hệ thống AI Hành chính công**. Hãy làm theo các hướng dẫn sau để khai thác tối đa tính năng:
@@ -1612,7 +1595,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
                 """)
 
         # ------------------------------------------------------------------
-        # TAB 2: TRỢ LÝ AI (HỆ THỐNG CHATBOT)
+        # TAB 2: TRỢ LÝ AI
         # ------------------------------------------------------------------
         with gr.Tab("💬 Trợ lý AI"):
             with gr.Row():
@@ -1667,11 +1650,9 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
             btn_clear.click(clear_chat_pro, [], [chat, audio_out, msg_in])
             
         # ------------------------------------------------------------------
-        # TAB 3: ADMIN PANEL (BẢO TOÀN LOG IN & LOG OUT)
+        # TAB 3: ADMIN PANEL (Sửa lỗi đồng bộ đăng nhập theo file gốc)
         # ------------------------------------------------------------------
         with gr.Tab("⚙️ Admin Panel"):
-            
-            # Khối giao diện đăng nhập (Login Form)
             with gr.Row() as login_form:
                 input_user = gr.Textbox(
                     label="Tài khoản cán bộ", 
@@ -1686,7 +1667,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
                 )
                 btn_login = gr.Button("Đăng nhập 🔑", scale=1, variant="primary")                        
             
-            # Khối nội dung Admin - Mặc định ẩn (visible=False)
             admin_content = gr.Group(visible=False)                         
             
             with admin_content:
@@ -1711,16 +1691,15 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
                 add_output = gr.Textbox(label="Kết quả xử lý")
                 procedures_list = gr.Markdown(value=get_procedures_list())
                 
-            # --- Logic xử lý Đăng nhập (Sửa lỗi ACCOUNT thành ADMIN_USERNAME tương thích tệp app.py gốc) ---
+            # Logic xử lý Đăng nhập chỉ dùng ADMIN_PASSWORD gốc của bạn
             def check_admin_login(username, password):
-                if username == ADMIN_USERNAME and password == ADMIN_PASSWORD:
+                if password == ADMIN_PASSWORD:
                     gr.Info("Đăng nhập hệ thống thành công!")
                     return gr.update(visible=False), gr.update(visible=True)
                 else:
-                    gr.Warning("Tài khoản hoặc mật khẩu không chính xác. Vui lòng kiểm tra lại!")
+                    gr.Warning("Mật khẩu không chính xác. Vui lòng kiểm tra lại!")
                     return gr.update(visible=True), gr.update(visible=False)
             
-            # --- Logic xử lý Đăng xuất ---
             def admin_logout():
                 gr.Info("Đã đăng xuất khỏi tài khoản admin.")
                 return (
@@ -1730,35 +1709,38 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
                     gr.update(value="")
                 )
                     
-            # Liên kết Sự kiện nút Đăng nhập
-            btn_login.click(
-                check_admin_login, 
-                inputs=[input_user, input_pass], 
-                outputs=[login_form, admin_content]
-            )
-            
-            # Liên kết Sự kiện nút Đăng xuất
-            btn_logout.click(
-                admin_logout,
-                inputs=[],
-                outputs=[login_form, admin_content, input_user, input_pass]
-            )
-            
-            # Sự kiện thêm thủ tục mới
-            btn_add.click(
-                admin_add,
-                [admin_code, admin_name, admin_docs, admin_coquan, admin_time, admin_lephi, admin_note, admin_steps],
-                [add_output, procedures_list]
-            )
+            btn_login.click(check_admin_login, inputs=[input_user, input_pass], outputs=[login_form, admin_content])
+            btn_logout.click(admin_logout, inputs=[], outputs=[login_form, admin_content, input_user, input_pass])
+            btn_add.click(admin_add, [admin_code, admin_name, admin_docs, admin_coquan, admin_time, admin_lephi, admin_note, admin_steps], [add_output, procedures_list])
+
+        # ------------------------------------------------------------------
+        # TAB 4: MEMORY MANAGER (Trả lại đầy đủ Tab gốc)
+        # ------------------------------------------------------------------
+        with gr.Tab("🧠 Memory Manager"):
+            gr.Markdown("### 📊 Quản lý bộ nhớ hội thoại")
+
+            memory_info = gr.Textbox(label="Ngữ cảnh hiện tại", interactive=False, lines=3)
+            memory_history = gr.Textbox(label="Lịch sử chat", interactive=False, lines=10)
+
+            with gr.Row():
+                btn_refresh = gr.Button("🔄 Làm mới")
+                btn_clear_mem = gr.Button("🗑️ Xóa memory", variant='stop')
+
+            def refresh_memory():
+                ctx = conversation_memory.get_context_summary()
+                hist = "\n".join([f"{m['role']}: {m['content'][:100]}..." for m in conversation_memory.history[-5:]])
+                return ctx or "Không có ngữ cảnh", hist or "Không có lịch sử"
+
+            btn_refresh.click(refresh_memory, [], [memory_info, memory_history])
+            btn_clear_mem.click(clear_chat_pro, [], [chat, audio_out, msg_in])
 
-    # Khởi động cơ chế tải dữ liệu thời gian thực khi khởi động trang web
     demo.load(fn=update_news_interface, inputs=[], outputs=[live_news_box])
 
 # ============================================================
 # PHẦN 12.7: LAUNCH APP PRO
 # ============================================================
 print('='*70)
-print('🚀 KHỞI ĐỘNG ỨNG DỤNG AI PRO')
+print('🚀 KHỔI ĐỘNG ỨNG DỤNG AI PRO')
 print('='*70)
 print()