fix new features

app.py

@@ -69,7 +69,6 @@ def initialize_dataset():
             print("Dataset mới đã được tạo và đẩy lên Huggingface Hub.")
         except Exception as push_e:
             print(f"Lỗi khi đẩy Dataset lên Hub: {push_e}")
-    return
 
 def save_conversation(user_id: str, messages: List[Tuple[str, str]]):
     """
@@ -262,7 +261,13 @@ def generate_response(prompt: str, chat_history: List[Tuple[str, str]], user_id:
     outputs = []
     for text in streamer:
         outputs.append(text)
-        yield "".join(outputs)
+        # Yields the entire conversation history with the latest assistant message updated
+        conversation_formatted[-1]["content"] += text
+        # Convert to list of tuples
+        display_history = []
+        for msg in conversation_formatted:
+            display_history.append((msg["role"], msg["content"]))
+        yield display_history
 
     # Lưu phản hồi vào Dataset
     response = "".join(outputs)
@@ -400,35 +405,32 @@ def generate(
     top_p: float = 0.9,
     top_k: int = 50,
     repetition_penalty: float = 1.2,
-) -> Iterator[str]:
+) -> Iterator[List[Tuple[str, str]]]:
     """
     Hàm chính để xử lý đầu vào của người dùng và tạo phản hồi.
     """
     # Thông báo về việc phân tích đầu vào
-    yield "🔍 Đang phân tích truy vấn của bạn..."
+    yield [("system", "🔍 Đang phân tích truy vấn của bạn...")]
 
     # Lấy user_id từ trạng thái
     user_id = get_user_id(user_state)
 
-     # Thêm tin nhắn của người dùng vào lịch sử cuộc trò chuyện
-    chat_history.append((message, "")) 
-
     # Xác định hàm nào sẽ được gọi dựa trên tin nhắn của người dùng
     function_call = process_query(message)
 
     # Thông báo về hàm được chọn
     if function_call["name"] == "web_search":
-        yield "🛠️ Đã chọn chức năng: Tìm kiếm trên web."
+        yield [("system", "🛠️ Đã chọn chức năng: Tìm kiếm trên web.")]
     elif function_call["name"] == "summarize_query":
-        yield "🛠️ Đã chọn chức năng: Tóm tắt văn bản."
+        yield [("system", "🛠️ Đã chọn chức năng: Tóm tắt văn bản.")]
     elif function_call["name"] == "sentiment_analysis":
-        yield "🛠️ Đã chọn chức năng: Phân tích tâm lý."
+        yield [("system", "🛠️ Đã chọn chức năng: Phân tích tâm lý.")]
     elif function_call["name"] == "new_topic":
-        yield "🛠️ Đã chọn chức năng: Chủ đề mới."
+        yield [("system", "🛠️ Đã chọn chức năng: Chủ đề mới.")]
     elif function_call["name"] in ["general_query", "hard_query"]:
-        yield "🛠️ Đã chọn chức năng: Trả lời câu hỏi."
+        yield [("system", "🛠️ Đã chọn chức năng: Trả lời câu hỏi.")]
     else:
-        yield "⚠️ Không thể xác định chức năng phù hợp."
+        yield [("system", "⚠️ Không thể xác định chức năng phù hợp.")]
 
     # Xử lý lời gọi hàm và sinh phản hồi tương ứng
     response_iterator = handle_functions(
@@ -443,14 +445,14 @@ def generate(
         repetition_penalty=repetition_penalty
     )
 
+    # Start with the existing chat history
+    updated_chat_history = chat_history.copy()
+
     for response in response_iterator:
-        # Thay thế tin nhắn cuối cùng của người dùng với phản hồi của hệ thống
-        chat_history[-1] = (message, response)  
-        yield chat_history
-    # Lưu phản hồi cuối cùng vào Dataset
-    save_conversation(user_id, chat_history)
-    yield chat_history
-    
+        # Append assistant response
+        updated_chat_history.append(("assistant", response))
+        yield updated_chat_history
+
 # Định nghĩa các ví dụ để hướng dẫn người dùng
 EXAMPLES = [
     ["Xin chào! Bạn khỏe không?"],