Upload folder using huggingface_hub

app.py

@@ -1,3 +1,4 @@
+import re
 from typing import List
 
 import gradio as gr
@@ -54,15 +55,15 @@ CHAT_EXAMPLES = [
 
 def convert_to_chat_messages(history: List[List[str]]) -> List[ChatMessage]:
     chat_messages = []
-    # for conversation in history[-1:]:
-    #     for index, message in enumerate(conversation):
-    #         if not message:
-    #             continue
-    #
-    #         message = re.sub(r'\n \n\n---\n\n參考: \n.*$', '', message, flags=re.DOTALL)
-    #         role = MessageRole.USER if index % 2 == 0 else MessageRole.ASSISTANT
-    #         chat_message = ChatMessage(role=role, content=message.strip())
-    #         chat_messages.append(chat_message)
+    for conversation in history[-1:]:
+        for index, message in enumerate(conversation):
+            if not message:
+                continue
+
+            message = re.sub(r'\n \n\n---\n\n參考: \n.*$', '', message, flags=re.DOTALL)
+            role = MessageRole.USER if index % 2 == 0 else MessageRole.ASSISTANT
+            chat_message = ChatMessage(role=role, content=message.strip())
+            chat_messages.append(chat_message)
     chat_messages.append(ChatMessage(role=MessageRole.SYSTEM, content=SYSTEM_PROMPT))
 
     return chat_messages