Update app.py

app.py

@@ -19,7 +19,7 @@ import spaces
 from registry import get_model
 from core.describe_scene import describe_scene
 from core.process import process_image
-from core.input_handler import resolve_input, validate_video, validate_image
+from core.input_handler import resolve_input, validate_video, validate_image, show_preview_from_upload, show_preview_from_url
 from utils.helpers import format_error, generate_session_id
 from huggingface_hub import hf_hub_download
 
@@ -131,21 +131,30 @@ with gr.Blocks() as demo:
             # URL input
             url = gr.Textbox(label="URL (Image/Video)", visible=False)
             
+            # Preview uploaded media
+            preview_image = gr.Image(label="Preview Image", visible=False)
+            preview_video = gr.Video(label="Preview Video", visible=False)
+
             # Toggle visibility
             def toggle_inputs(selected_mode):
                 return [
                     gr.update(visible=(selected_mode == "Upload")),  # media_upload
-                    gr.update(visible=(selected_mode == "URL"))      # url
+                    gr.update(visible=(selected_mode == "URL")),      # url
+                    gr.update(visible=False),                        # preview_image
+                    gr.update(visible=False)                         # preview_video
+                
                 ]
             
             mode.change(toggle_inputs, inputs=mode, outputs=[media_upload, url])
-            
+            media_upload.change(show_preview_from_upload, inputs=media_upload, outputs=[preview_image, preview_video])
+            url.submit(show_preview_from_url, inputs=url, outputs=[preview_image, preview_video])
+
             # Visibility logic function
             def toggle_visibility(checked):
                 return gr.update(visible=checked)
             
-            def toggle_det_visibility(checked):
-                return [gr.update(visible=checked), gr.update(visible=checked)]
+            # def toggle_det_visibility(checked):
+            #     return [gr.update(visible=checked), gr.update(visible=checked)]
             
             run_det = gr.Checkbox(label="Object Detection")
             run_seg = gr.Checkbox(label="Semantic Segmentation")