Upload folder using huggingface_hub

gradio_app.py

@@ -37,22 +37,23 @@ def load_chat(
             history = ChatInterface._tuples_to_messages(history)
         elif len(history) > 0 and isinstance(history[0], dict):
             hist = []
-            skip = False
+            merge = False
             for i, h in enumerate(history):
-                if skip:
-                    skip = False
-                    continue
                 content = h.get("content", [])
                 role = h.get("role", "")
+                ct = []
                 if isinstance(content, tuple):
                     src_path = content[0]
-                    hist.append({"role": role, "content": [{"type": "image_url", "image_url": {"url": get_gradio_url(src_path)}}]})
+                    ct = [{"type": "image_url", "image_url": {"url": get_gradio_url(src_path)}}]
                 else:
-                    hist.append({"role": role, "content": [{"type": "text", "text": content}]})
-                # if 2 consecutive messages are from the same role, merge the content
+                    ct = [{"type": "text", "text": content}]
+                if merge:
+                    hist[-1]["content"].append(ct)
+                    merge = False
+                else:
+                    hist.append({"role": role, "content": ct})
                 if i < len(history) - 1 and role == history[i + 1].get("role", ""):
-                    hist[-1]["content"].extend(history[i + 1].get("content", []))
-                    skip = True
+                    merge = True
             history = hist
         files = message.get('files', [])
         content = [