fix example images

app.py

@@ -904,16 +904,20 @@ def update_task_ui(task):
 
 def update_image_preview(image_file):
     """Update the image preview."""
-    if image_file:
-        return image_file.name
-    return None
+    if image_file is None:
+        return None
+    if isinstance(image_file, str):
+        return image_file
+    return image_file.name if hasattr(image_file, 'name') else None
 
 
 def update_goal_preview(goal_file):
     """Update the goal preview."""
-    if goal_file:
-        return goal_file.name
-    return None
+    if goal_file is None:
+        return None
+    if isinstance(goal_file, str):
+        return goal_file
+    return goal_file.name if hasattr(goal_file, 'name') else None
 
 
 def get_download_link(selected_frame, all_paths):
@@ -1419,10 +1423,9 @@ with gr.Blocks(
                         )
 
                     with gr.Group(visible=False) as prediction_group:
-                        image_input = gr.File(
+                        image_input = gr.Image(
                             label="Upload Start Image",
-                            file_count="single",
-                            file_types=["image"],
+                            type="filepath",
                             interactive=True,
                             elem_id="image_input",
                         )
@@ -1440,17 +1443,15 @@ with gr.Blocks(
 
                     with gr.Group(visible=False) as planning_group:
                         with gr.Row():
-                            image_input_planning = gr.File(
+                            image_input_planning = gr.Image(
                                 label="Upload Start Image",
-                                file_count="single",
-                                file_types=["image"],
+                                type="filepath",
                                 interactive=True,
                                 elem_id="image_input_planning",
                             )
-                            goal_input = gr.File(
+                            goal_input = gr.Image(
                                 label="Upload Goal Image",
-                                file_count="single",
-                                file_types=["image"],
+                                type="filepath",
                                 interactive=True,
                                 elem_id="goal_input",
                             )
@@ -1704,11 +1705,15 @@ with gr.Blocks(
     )
 
     image_input.change(
-        fn=update_image_preview, inputs=[image_input], outputs=[image_preview]
+        fn=update_image_preview,
+        inputs=[image_input],
+        outputs=[image_preview]
     ).then(fn=lambda: gr.update(visible=True), inputs=[], outputs=[preview_row])
 
     goal_input.change(
-        fn=update_goal_preview, inputs=[goal_input], outputs=[goal_preview]
+        fn=update_goal_preview,
+        inputs=[goal_input],
+        outputs=[goal_preview]
     ).then(fn=lambda: gr.update(visible=True), inputs=[], outputs=[preview_row])
 
     def update_pointcloud_frames(pointcloud_paths):