Update app.py

app.py

@@ -26,6 +26,7 @@ def get_db_history(profile: gr.OAuthProfile | None):
         return []
 
 def load_selected_chat(evt: gr.SelectData, db_data):
+    # Lấy dữ liệu từ item được chọn
     selected_item = db_data[evt.index]
     return selected_item['history'], selected_item['session_id']
 
@@ -61,7 +62,7 @@ def chat_engine(message, history, session_id, profile: gr.OAuthProfile | None):
     except Exception as e:
         yield gr.update(), new_history + [{"role": "assistant", "content": f"Lỗi: {str(e)}"}], session_id
 
-# 4. UI - BỎ THAM SỐ CSS KHỎI BLOCKS
+# 4. UI
 with gr.Blocks() as demo:
     session_id = gr.State(str(uuid.uuid4()))
     db_state = gr.State([])
@@ -73,6 +74,7 @@ with gr.Blocks() as demo:
             new_chat_btn = gr.Button("＋ NEW WORKSPACE", variant="primary")
             
             gr.Markdown("### 🗄️ Database Memory")
+            # Dataset ban đầu
             history_list = gr.Dataset(
                 components=[gr.Textbox(visible=False)],
                 label="Lịch sử chat",
@@ -81,20 +83,23 @@ with gr.Blocks() as demo:
             )
 
         with gr.Column(scale=4):
-            # FIX: Bỏ type="messages"
             chatbot = gr.Chatbot(show_label=False, height="85vh")
             msg_input = gr.MultimodalTextbox(show_label=False, placeholder="Hỏi NodeAI...", container=False)
 
+    # FIX: Hàm refresh trả về đối tượng gr.Dataset mới
     def refresh_data(profile: gr.OAuthProfile | None):
-        if not profile: return [], []
+        if not profile: 
+            return [], gr.Dataset(samples=[])
         data = get_db_history(profile)
         samples = [[item['title']] for item in data]
-        return data, samples
+        # QUAN TRỌNG: Phải return gr.Dataset(...) thay vì chỉ return samples
+        return data, gr.Dataset(samples=samples)
 
     demo.load(refresh_data, None, [db_state, history_list])
 
     history_list.select(fn=load_selected_chat, inputs=[db_state], outputs=[chatbot, session_id])
 
+    # Sau khi chat xong, gọi refresh để cập nhật Sidebar
     msg_input.submit(chat_engine, [msg_input, chatbot, session_id], [msg_input, chatbot, session_id]).then(
         refresh_data, None, [db_state, history_list]
     )
@@ -102,5 +107,4 @@ with gr.Blocks() as demo:
     new_chat_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
 
 if __name__ == "__main__":
-    # FIX: Chuyển css xuống launch()
     demo.launch(css=custom_css)