bugfix

app.py

@@ -476,6 +476,10 @@ with gr.Blocks(title="Diffusion as Shader") as demo:
         right_column = gr.Column(scale=1)
 
     with right_column:
+        
+        gr.Markdown("### 4. Generate Tracking Video")
+        gr.Markdown("'Generate Tracking Video' is used to preserve all motion from the Source. You need to generate tracking video before producing the final result.")
+        mt_run_btn = gr.Button("Generate Tracking", variant="primary", size="lg")
         tracking_video = gr.Video(label="Tracking Video")
         
         apply_tracking_btn = gr.Button("5. Generate Video", variant="primary", size="lg", interactive=False)
@@ -521,17 +525,12 @@ with gr.Blocks(title="Diffusion as Shader") as demo:
                 mt_repaint_upload = gr.UploadButton("Upload Repaint Image (Optional)", file_types=["image"])
                 mt_repaint_preview = gr.Image(label="Repaint Image Preview")
                 
-                # 上传文件后更新预览
                 mt_repaint_upload.upload(
-                    fn=update_source_preview,  # 复用相同的函数
+                    fn=update_source_preview,
                     inputs=[mt_repaint_upload],
                     outputs=[mt_repaint_preview]
                 )
-                
-                # Add run button for Motion Transfer tab
-                mt_run_btn = gr.Button("4. Generate Tracking", variant="primary", size="lg")
-                
-                # Connect to process function, but don't apply tracking
+
                 mt_run_btn.click(
                     fn=process_motion_transfer,
                     inputs=[
@@ -544,7 +543,7 @@ with gr.Blocks(title="Diffusion as Shader") as demo:
                     inputs=[tracking_video],
                     outputs=[apply_tracking_btn]
                 )
-
+                
             with gr.TabItem("Camera Control"):
                  gr.Markdown("Camera Control is not available in Huggingface Space, please deploy our [GitHub project](https://github.com/IGL-HKUST/DiffusionAsShader) on your own machine")