Upload folder using huggingface_hub

app.py

@@ -14,13 +14,15 @@ sys_prompt = """First output the the types of degradations in image briefly in <
 project_dir = os.path.dirname(os.path.abspath(__file__))
 
 is_spaces = os.getenv("SPACE_ID") is not None
+# temp_dir = os.path.join(project_dir, ".gradio_temp")
+temp_dir = project_dir
 if not is_spaces:
-    temp_dir = os.path.join(project_dir, ".gradio_temp")
     os.makedirs(temp_dir, exist_ok=True)
     os.environ["GRADIO_TEMP_DIR"] = temp_dir
 
 MODEL_PATH = os.getenv("MODEL_PATH", "Jiaqi-hkust/Robust-R1-RL")
 
+
 print(f"==========================================")
 print(f"Initializing application...")
 print(f"==========================================")
@@ -175,13 +177,14 @@ def get_model_handler():
         model_handler = ModelHandler(MODEL_PATH)
     return model_handler
 
+custom_css = """
+.gradio-container { font-family: 'Inter', sans-serif; }
+#chatbot { height: 650px !important; overflow-y: auto; }
+"""
+
 def create_chat_ui():
-    custom_css = """
-    .gradio-container { font-family: 'Inter', sans-serif; }
-    #chatbot { height: 650px !important; overflow-y: auto; }
-    """
 
-    with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="Robust-R1") as demo:
+    with gr.Blocks(title="Robust-R1") as demo:
         
         with gr.Row():
             gr.Markdown("# 🤖Robust-R1:Degradation-Aware Reasoning for Robust Visual Understanding")
@@ -191,7 +194,6 @@ def create_chat_ui():
                 chatbot = gr.Chatbot(
                     elem_id="chatbot",
                     label="Chat",
-                    type="messages",
                     avatar_images=(None, "https://api.dicebear.com/7.x/bottts/svg?seed=Qwen"),
                     height=650
                 )
@@ -244,14 +246,30 @@ def create_chat_ui():
             gr.Markdown("*No example images available, please manually upload images for testing*")
         
         async def respond(user_msg, history, temp, tokens):
+            print("user_msg:")
+            print(user_msg)
             text = user_msg.get("text", "").strip()
             files = user_msg.get("files", [])
-            user_content = list(files)
-            if text: user_content.append(text)
+            # 按照参考代码的格式：文件使用 {"path": x}，文本直接使用字符串
+            user_message = {"role": "user", "content": []}
+            
+            # 添加图像文件，使用 {"path": file_path} 格式
+            for file_path in files:
+                if file_path:
+                    # 确保使用绝对路径
+                    abs_path = os.path.abspath(file_path) if not os.path.isabs(file_path) else file_path
+                    user_message["content"].append({"path": abs_path})
             
-            if not files and text: user_message = {"role": "user", "content": text}
-            else: user_message = {"role": "user", "content": user_content}
+            # 添加文本，直接使用字符串
+            if text:
+                user_message["content"].append(text)
             
+            # 如果只有文本没有文件，content 保持为列表；如果都没有，content 为空列表
+            # 如果只有文本，也可以直接使用字符串（参考 Gradio 的常见用法）
+            if not files and text:
+                user_message["content"] = text
+            print("user_message:")
+            print(user_message)
             history.append(user_message)
             yield history, gr.MultimodalTextbox(value=None, interactive=False)
 
@@ -295,15 +313,19 @@ if __name__ == "__main__":
     if is_spaces:
         print(f"🚀 Running on Hugging Face Spaces: {os.getenv('SPACE_ID')}")
         demo.launch(
+            theme=gr.themes.Soft(),
+            css=custom_css,
             show_error=True,
-            allowed_paths=[project_dir] if project_dir else None
+            allowed_paths=[project_dir, temp_dir] if project_dir else None
         )
     else:
         print(f"🚀 Service is starting, please visit: http://localhost:7860")
         demo.launch(
+            theme=gr.themes.Soft(),
+            css=custom_css,
             server_name="0.0.0.0",
             server_port=7860,
             share=False,
             show_error=True,
-            allowed_paths=[project_dir]
+            allowed_paths=[project_dir, temp_dir] if project_dir else None
         )

requirements.txt

@@ -1,10 +1,10 @@
-gradio>=6.0.0
-torch>=2.0.0
-torchvision>=0.15.0
-transformers>=4.37.0
-qwen-vl-utils
-accelerate
-sentencepiece
-protobuf
-pillow
+gradio>=6.1.0
+torch>=2.0.0
+torchvision>=0.15.0
+transformers>=4.37.0
+qwen-vl-utils
+accelerate
+sentencepiece
+protobuf
+pillow
 einops