bugfix

app.py

@@ -201,7 +201,7 @@ def process_motion_transfer(source, prompt, mt_repaint_option, mt_repaint_image)
         # 保存上传的文件
         input_video_path = save_uploaded_file(source)
         if input_video_path is None:
-            return None, None, None
+            return None, None, None, None, None
         
         print(f"DEBUG: Repaint option: {mt_repaint_option}")
         print(f"DEBUG: Repaint image: {mt_repaint_image}")
@@ -255,8 +255,7 @@ def process_motion_transfer(source, prompt, mt_repaint_option, mt_repaint_image)
             pred_tracks, pred_visibility = generate_tracking_cotracker(video_tensor)
             tracking_path, tracking_tensor = das.visualize_tracking_cotracker(pred_tracks, pred_visibility)
             print('Export tracking video via cotracker')
-
-        # 返回处理结果，但不应用跟踪
+        
         return tracking_path, video_tensor, tracking_tensor, repaint_img_tensor, fps
     except Exception as e:
         import traceback
@@ -483,9 +482,10 @@ with gr.Blocks(title="Diffusion as Shader") as demo:
         output_video = gr.Video(label="Generated Video")
 
     with left_column:
-        source_upload = gr.UploadButton("1. Upload Source", file_types=["image", "video"])
+        gr.Markdown("### 1. Upload Source")
         gr.Markdown("Upload a video or image, We will extract the motion and space structure from it")
         source_preview = gr.Video(label="Source Preview")
+        source_upload = gr.UploadButton("Upload Source", file_types=["image", "video"])
 
         def update_source_preview(file):
             if file is None:
@@ -502,21 +502,23 @@ with gr.Blocks(title="Diffusion as Shader") as demo:
         gr.Markdown("### 2. Enter the prompt")
         common_prompt = gr.Textbox(label="Describe the scene and the motion you want to create: ", lines=2)
         
+        gr.Markdown("### 3. Select a task")
         with gr.Tabs() as task_tabs:
             # Motion Transfer tab
             with gr.TabItem("Motion Transfer"):
-                gr.Markdown("### Motion Transfer")
                 
+                gr.Markdown("#### 3.1 Process the first frame of Source")
+                gr.Markdown("DaS can produce novel videos while maintaining the features of the first frame and all the motion of the Source. You can use FLUX.1 to repaint the first frame of the Source")
                 # Simplified controls - Radio buttons for Yes/No and separate file upload
                 with gr.Row():
                     mt_repaint_option = gr.Radio(
-                        label="Repaint First Frame",
+                        label="Repaint First Frame (Optional)",
                         choices=["No", "Yes"],
                         value="No"
                     )
-                gr.Markdown("### Note: If you want to use your own image as repainted first frame, please upload the image in below.")
+                gr.Markdown("Or if you want to use your own image as repainted first frame, please upload the image in below.")
                 
-                mt_repaint_upload = gr.UploadButton("3. Upload Repaint Image (Optional)", file_types=["image"])
+                mt_repaint_upload = gr.UploadButton("Upload Repaint Image (Optional)", file_types=["image"])
                 mt_repaint_preview = gr.Image(label="Repaint Image Preview")
                 
                 # 上传文件后更新预览
@@ -550,6 +552,8 @@ with gr.Blocks(title="Diffusion as Shader") as demo:
                  gr.Markdown("Object Manipulation is not available in Huggingface Space, please deploy our [GitHub project](https://github.com/IGL-HKUST/DiffusionAsShader) on your own machine")
 
     examples_list = load_examples()
+    gr.Markdown("### Examples (For Workflow Demo Only)")
+    gr.Markdown("The following examples are only for demonstrating DaS's workflow and output quality. If you want to actually generate tracking or videos, the program will not run unless you manually upload files from your devices.")
     if examples_list:
         with gr.Blocks() as examples_block:
             gr.Examples(