Update app.py

app.py

@@ -790,13 +790,18 @@ RESPONSE_TEMPLATES = {
     " khai_sinh": {
         "short": "Dạ để khai sinh cho bé, bác/cháu cần mang: Giấy khai sinh, CCCD của cả cha mẹ, và sổ hộ khẩu. Ra UBND xã/phường nơi cư trú là được ạ!",
         "detailed": """Dạ về khai sinh cho bé, em hướng dẫn chi tiết ạ:
+
 📋 **HỒ SƠ CẦN CHUẨN BỊ:**
 1. Giấy khai sinh (điền đầy đủ thông tin)
 2. CCCD/CMND của cả cha và mẹ
 3. Sổ hộ khẩu của cha mẹ
+
 🏢 **NƠI LÀM:** UBND xã/phường nơi cư trú
+
 ⏰ **THỜI GIAN:** 1-3 ngày
+
 💰 **LỆ PHÍ:** Miễn phí
+
 ⚠️ **LƯU Ý:** Làm trong 60 ngày kể từ khi sinh để tránh phạt nhé!"""
     }
 }
@@ -930,6 +935,7 @@ Dựa trên tài liệu và quy định hiện hành:
     # Bước 4: Tổng hợp câu trả lời
     step4 = f"""BƯỚC 4 - TỔNG HỢP TRẢ LỜI:
 Hãy trả lời theo phong cách: {persona['tone']}
+
 Lưu ý:
 - Dùng các mẫu câu: {', '.join(persona['greeting'][:2])}
 - Kết thúc bằng: {', '.join(persona['closing'][:2])}
@@ -1192,11 +1198,16 @@ QUY TRÌNH: {proc['steps']}
 
         # Bước 8: Full prompt cho LLM
         full_prompt = f"""Bạn là cán bộ tư vấn thủ tục hành chính công tại UBND. Hãy hỗ trợ người dân.
+
 {cot_prompt}
+
 CÂU HỎI: {query}
+
 {proc_info}
+
 LỊCH SỬ ĐỌC CHUẨN KHI CÓ:
 {relevant_history}
+
 HÃY TRẢ LỜI NGƯỜI DÂNG:"""
 
         messages = [
@@ -1320,22 +1331,23 @@ print()
 print('='*70)
 
 # ======================================================================
-# PHẦN 12.5: GRADIO INTERFACE (BẢN TỐI ƯU - GIỮ BẢN TIN & FULL LOGIN/LOGOUT)
-# ======================================================================
+# ============================================================
+# PHẦN 12.5: GRADIO INTERFACE
+# ============================================================
 
 import gradio as gr
 print('='*70)
-print('🖥️ KHỞI TẠO GRADIO INTERFACE CHUẨN')
+print('🖥️ KHỞI TẠO GRADIO INTERFACE')
 print('='*70)
 print()
 
-# Danh sách thủ tục quản trị viên
+# Admin functions
 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]: # Giới hạn hiển thị 20 bản ghi đầu tiên
+    for p in admin_procedures[:20]: # Limit to 20 for display
         lines.append(f"| {p['code']} | {p['name']} | {p['coquan']} |")
     return "\n".join(lines)
 
@@ -1346,18 +1358,20 @@ def admin_add(code, name, docs, coquan, time_val, lephi, note, steps):
         'time': time_val, 'lephi': lephi, 'note': note, 'steps': steps
     }
     admin_procedures.append(new_proc)
-    return f"✅ Đã thêm thành công thủ tục: {code} - {name}", get_procedures_list()
+    # For simplicity, we are not updating DICHVUCONG_DATA or Pinecone here directly
+    # In a real application, you would update the backend data source
+    return f"✅ Đã thêm: {code} - {name}", get_procedures_list()
 
-# Hàm xử lý hội thoại với AI Process PRO
+# Chat function với AI Process PRO
 def process_chat_pro(audio, text, history, persona="thuong"):
     try:
         if history is None:
             history = []
 
-        # Cập nhật trạng thái nhân vật
+        # Cập nhật persona
         conversation_memory.set_persona(persona)
 
-        # Chuẩn hóa dữ liệu đầu vào
+        # Xử lý input
         query = ''
         if audio and not os.path.isdir(audio):
             query = speech_to_text(audio)
@@ -1367,16 +1381,21 @@ def process_chat_pro(audio, text, history, persona="thuong"):
         if len(query) < 3:
             return history, None, ''
 
-        # Gọi mô hình xử lý lõi AI Process PRO
+        # Dùng AI Process PRO
         result = ai_process_pro(query, use_memory=True)
 
-        # Định dạng câu trả lời hiển thị
-        response = f"""{result['response']}\n\n--- \n_⏱️ Xử lý: {result['process_time']}s | 📄 Tài liệu: {result['docs_count']} | 🌡️ Temp: {result['temperature']}_"""
+        # Format response
+        response = f"""
+
+{result['response']}
+
+---
+_⏱️ Xử lý: {result['process_time']}s | 📄 Tài liệu: {result['docs_count']} | 🌡️ Temp: {result['temperature']}_"""
 
         history.append({"role": "user", "content": query})
         history.append({"role": "assistant", "content": response})
 
-        # Xử lý chuyển đổi văn bản thành giọng nói (TTS) phản hồi
+        # TTS
         audio_out = None
         try:
             loop = asyncio.get_event_loop()
@@ -1394,7 +1413,7 @@ def process_chat_pro(audio, text, history, persona="thuong"):
     except Exception as e:
         import traceback
         traceback.print_exc()
-        error_msg = f"Hệ thống gặp lỗi: {str(e)}"
+        error_msg = f"Lỗi: {str(e)}"
         return history, None, error_msg
 
 def clear_chat_pro():
@@ -1410,23 +1429,28 @@ def change_persona(new_persona):
     }
     return f"✅ Đã chuyển sang: {persona_names.get(new_persona, new_persona)}"
 
-# Định hình cấu trúc Style CSS giao diện tĩnh bảo mật
+# Build PRO interface
 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 (Gradient Xanh Navy) */
+
+/* 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 0px -20px; /* Kết nối khít với khối bản tin bên dưới */
+    margin: -20px -20px 20px -20px; /* Tràn viền */
     box-shadow: 0 4px 12px rgba(0,0,0,0.1);
     color: white;
 }
+
 .admin-icon {
     font-size: 45px;
     margin-right: 20px;
@@ -1434,6 +1458,7 @@ footer {display: none !important;}
     padding: 10px;
     border-radius: 12px;
 }
+
 .admin-brand h1 {
     color: #ffffff !important;
     font-size: 24px !important;
@@ -1442,45 +1467,15 @@ footer {display: none !important;}
     letter-spacing: 1px;
     text-transform: uppercase;
 }
+
 .admin-brand p {
     color: #e0e0e0 !important;
     font-size: 14px !important;
     margin: 5px 0 0 0 !important;
     font-weight: 300 !important;
 }
-/* Khung bọc khu vực Tin tức nằm cố định dưới Taskbar */
-.global-news-container {
-    background: #ffffff;
-    padding: 15px 20px;
-    margin: 0px -20px 20px -20px; /* Tràn lề sát mép màn hình */
-    border-bottom: 1px solid #dee2e6;
-    box-shadow: 0 2px 8px rgba(0,0,0,0.02);
-}
-.section-title-news {
-    color: #1e3c72;
-    font-weight: 600;
-    margin-bottom: 12px;
-    border-bottom: 2px solid #e9ecef;
-    padding-bottom: 6px;
-    font-size: 15px;
-}
-.news-grid {
-    display: grid;
-    grid-template-columns: repeat(3, 1fr);
-    gap: 15px;
-}
-@media (max-width: 768px) {
-    .news-grid {
-        grid-template-columns: 1fr;
-    }
-}
-.news-card {
-    background: #f8fafc;
-    padding: 14px 16px;
-    border-left: 4px solid #1e3c72;
-    border-radius: 6px;
-    box-shadow: 0 2px 4px rgba(0,0,0,0.02);
-}
+
+/* Tinh chỉnh Nút bấm */
 button.primary {
     background-color: #0056b3 !important;
     color: white !important;
@@ -1489,16 +1484,21 @@ button.primary {
     border: none !important;
     transition: all 0.3s ease;
 }
+
 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;
@@ -1508,15 +1508,22 @@ button.primary:hover {
 }
 """
 
-# Khởi tạo Block giao diện chính thức
+# 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. Header ứng dụng công vụ kết hợp Bản tin điện tử 3 ô nằm NGAY DƯỚI THANH TASKBAR
-    
+    # 1. TOPBAR CHUYÊN NGHIỆP (Không dùng logo/tên thật)
+    gr.HTML("""
+    <div class="admin-topbar">
+        <div class="admin-icon">🏛️</div>
+        <div class="admin-brand">
+            <h1>HỆ THỐNG TRỢ LÝ ẢO TƯ VẤN HÀNH CHÍNH</h1>
+            <p>Giải đáp thông tin thủ tục - Nhanh chóng, Chính xác, Hỗ trợ 24/7</p>
+        </div>
+    </div>
+    """)
 
-    # 2. Hệ thống chuyển Tab làm việc chính
     with gr.Tabs():
-        # TAB 1: Giao diện tương tác tư vấn công dân
+        # Chat Tab
         with gr.Tab("💬 Trợ lý AI"):
             with gr.Row():
                 with gr.Column(scale=7):
@@ -1533,6 +1540,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
                         btn_send = gr.Button('GỬI CÂU HỎI', variant='primary')
                         btn_clear = gr.Button('Làm mới')
 
+                    # --- Ô HƯỚNG DẪN NHỎ THÊM VÀO ĐÂY ---
                     gr.HTML("""
                         <div style="
                             margin-top: 15px; 
@@ -1546,9 +1554,9 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
                         ">
                             <strong style="color: #1e3c72;">💡 Cách dùng nhanh:</strong>
                             <ul style="margin: 5px 0 0 15px; padding: 0;">
-                                <li>Nhập câu hỏi văn bản hoặc dùng 🎤 <b>Micro</b> để nói trực tiếp.</li>
-                                <li>Nhấn nút <b>Gửi câu hỏi</b> để hệ thống xử lý dữ liệu RAG.</li>
-                                <li>Trợ lý ảo sẽ tự động 🔊 <b>Đọc câu trả lời</b> bằng giọng nói thích hợp.</li>
+                                <li>Nhập câu hỏi hoặc dùng 🎤 <b>Micro</b>.</li>
+                                <li>Nhấn <b>Gửi</b> để nhận phản hồi từ AI.</li>
+                                <li>Hệ thống sẽ tự động 🔊 <b>Phát loa</b> câu trả lời.</li>
                             </ul>
                         </div>
                     """)
@@ -1566,34 +1574,28 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
                 inputs=[audio_in, msg_in]
             )
 
-            # Ràng buộc sự kiện Chatbot
+            # (Các sự kiện click giữ nguyên)
             btn_send.click(process_chat_pro, [audio_in, msg_in, chat], [chat, audio_out, msg_in])
             msg_in.submit(process_chat_pro, [audio_in, msg_in, chat], [chat, audio_out, msg_in])
             btn_clear.click(clear_chat_pro, [], [chat, audio_out, msg_in])
-
-        # TAB 2: Quản trị viên cập nhật thủ tục (CÓ FULL ĐĂNG NHẬP / ĐĂNG XUẤT)
         with gr.Tab("⚙️ Admin Panel"):
+            gr.Markdown("### 🔐 Xác thực quyền quản trị")
             
-            # Form đăng nhập (Tài khoản + Mật khẩu) như tài khoản Google
-            with gr.Row() as login_form:
-                with gr.Column(scale=2):
-                    input_user = gr.Textbox(label="Tài khoản cán bộ", placeholder="Nhập tên tài khoản...")
-                with gr.Column(scale=2):
-                    input_pass = gr.Textbox(label="Mật khẩu mã hóa", type="password", placeholder="Nhập mật khẩu...")
-                with gr.Column(scale=1):
-                    gr.Markdown("<br>") # Đẩy nút xuống bằng hàng với textbox
-                    btn_login = gr.Button("Đăng nhập 🔑", variant="primary")
+            with gr.Row():
+                input_pass = gr.Textbox(
+                    label="Nhập mật khẩu để mở khóa tính năng Admin", 
+                    type="password", 
+                    placeholder="nhap mat khau",
+                    scale=4
+                )
+                btn_login = gr.Button("Xác thực 🔑", scale=1)
             
-            # Khối nội dung quản trị viên - Mặc định ẩn bảo mật
+            # Group chứa nội dung Admin - Mặc định ẩn (visible=False)
             admin_content = gr.Group(visible=False) 
             
             with admin_content:
-                with gr.Row():
-                    gr.Markdown("### 🏛️ KHU VỰC CẬP NHẬT DỮ LIỆU CẤP CAO")
-                    btn_logout = gr.Button("Đăng xuất ↩️", variant="stop", scale=1, size="sm")
-                
                 gr.Markdown("---")
-                gr.Markdown("### ➕ Thêm thủ tục hành chính mới (Mô hình cập nhật ngay lập tức)")
+                gr.Markdown("### ➕ Thêm thủ tục mới (AI sẽ học ngay)")
                 with gr.Row():
                     with gr.Column():
                         admin_code = gr.Textbox(label="Mã thủ tục")
@@ -1602,64 +1604,83 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), cs
                         admin_coquan = gr.Textbox(label="Cơ quan tiếp nhận")
 
                     with gr.Column():
-                        admin_time = gr.Textbox(label="Thời gian giải quyết")
-                        admin_lephi = gr.Textbox(label="Lệ phí quy định")
-                        admin_note = gr.Textbox(label="Lưu ý quan trọng", lines=2)
-                        admin_steps = gr.Textbox(label="Quy trình thực hiện", lines=3)
+                        admin_time = gr.Textbox(label="Thời gian")
+                        admin_lephi = gr.Textbox(label="Lệ phí")
+                        admin_note = gr.Textbox(label="Lưu ý", lines=2)
+                        admin_steps = gr.Textbox(label="Quy trình", lines=3)
 
                 btn_add = gr.Button("➕ THÊM VÀO HỆ THỐNG", variant='primary', size='lg')
-                add_output = gr.Textbox(label="Trạng thái xử lý yêu cầu")
+                add_output = gr.Textbox(label="Kết quả xử lý")
                 procedures_list = gr.Markdown(value=get_procedures_list())
 
-            # Cấu trúc logic chuyển đổi trạng thái giao diện an toàn
-            def check_admin_login(username, password):
-                if username == ACCOUNT and 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!")
+            # --- Logic xử lý ẩn hiện và mật khẩu ---
+            def check_admin(password):
+                if password == ADMIN_PASSWORD:
+                    # Nếu đúng: Hiện nội dung admin, ẩn ô nhập mật khẩu và hiện thông báo thành công
+                    gr.Info("Đăng nhập thành công! Quyền quản trị đã được mở.")
                     return gr.update(visible=True), gr.update(visible=False)
+                else:
+                    # Nếu sai: Giữ ẩn nội dung và hiện cảnh báo
+                    raise gr.Error("Mật khẩu không chính xác. Vui lòng thử lại!")
+                    return gr.update(visible=False), gr.update(visible=True)
 
-            def admin_logout():
-                gr.Info("Đã đăng xuất an toàn khỏi tài khoản cán bộ.")
-                return gr.update(visible=True), gr.update(visible=False), gr.update(value=""), gr.update(value="")
-
-            # Đăng ký tương tác sự kiện xác thực
+            # Sự kiện khi nhấn nút Xác thực
             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]
+                check_admin, 
+                inputs=[input_pass], 
+                outputs=[admin_content, input_pass]
             )
 
+            # Sự kiện khi nhấn nút THÊM (giữ nguyên logic cũ của bạn)
             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]
             )
 
+        # Memory Tab (mới)
+        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])
+
+        
+
+
 
 # ======================================================================
-# PHẦN 12.7: LAUNCH SYSTEM APPLICATION
-# ======================================================================
+# ============================================================
+# PHẦN 12.7: LAUNCH APP PRO
+# ============================================================
 
 print('='*70)
-print('🚀 KHỞI ĐỘNG ỨNG DỤNG AI HÀNH CHÍNH CÔNG CHÍNH THỨC')
+print('🚀 KHỞI ĐỘNG ỨNG DỤNG AI PRO')
 print('='*70)
 print()
 
+
+
 print('='*70)
-print('  Đang đồng bộ mạng và thiết lập public link...')
-print('  Địa chỉ Web-UI sẽ xuất hiện bên dưới...')
+print('  Đang tạo public link...')
+print('  Link sẽ xuất hiện bên dưới sau vài giây...')
 print('='*70)
 print()
 
-# Khởi chạy chương trình
+# Launch với demo_pro thay vì demo
 demo.launch(
     share=True,
     server_name='0.0.0.0',