Upload folder using huggingface_hub

gradio_app.py

@@ -11,7 +11,8 @@ DESCRIPTION = "Welcome to the AIN Arabic VLM chatbot. The best Arabic-English VL
 TOP_N_HISTORY = 2
 LOGO_PATH = "https://huggingface.co/spaces/ahmedheakl/AIN-Arabic-VLM/resolve/main/logo.jpeg"
 
-
+def get_gradio_url(path: str) -> str:
+    return f"https://{USERNAME}-{SPACE_NAME}.hf.space/gradio_api/file={path}"
 
 def load_chat(
     base_url: str,
@@ -39,7 +40,7 @@ def load_chat(
             for i, h in enumerate(history):
                 content = h.get("content", "")
                 if isinstance(content, tuple):
-                    hist.append({"role": "user", "content": [{"type": "image_url", "image_url": {"url": content[0]}}]})
+                    hist.append({"role": "user", "content": [{"type": "image_url", "image_url": {"url": get_gradio_url(content[0])}}]})
                 else:
                     hist.append({"role": "user", "content": [{"type": "text", "text": content}]})
             history = hist
@@ -49,8 +50,7 @@ def load_chat(
         ]
         if files:
             src_path = files[0]
-            image_url = f"https://{USERNAME}-{SPACE_NAME}.hf.space/gradio_api/file={src_path}"
-            content.append({"type": "image_url", "image_url": {"url": image_url}})
+            content.append({"type": "image_url", "image_url": {"url": get_gradio_url(src_path)}})
         stream = client.chat.completions.create(
             model=model,
             messages=history + [{"role": "user", "content": content}],