Update app.py

app.py

@@ -416,7 +416,15 @@ def create_demo():
         with gr.Row():
             with gr.Column(scale=1):
                 preview_image = gr.Image(label="Preview Image", type="filepath", elem_classes="preview_media", visible=False)
-                preview_video = gr.Video(label="Preview Video", elem_classes="preview_media", visible=False, format="mp4")
+                preview_video = gr.Video(label="Preview Video", elem_classes="preview_media", visible=False, format="mp4", show_controls=False)  
+                def _play_pause(video_path):
+                    # simple JS to toggle playback – Gradio lets you inject JS via `gr.HTML`
+                    return f'''
+                    <script>
+                    const vid = document.querySelector('video[src="{video_path}"]');
+                    if (vid) vid.paused ? vid.play() : vid.pause();
+                    </script>
+                    '''
                 preview_status = gr.Textbox(label="Preview status", interactive=False, lines=2, value="", visible=True)
             with gr.Column(scale=2):
                 url_input = gr.Textbox(label="Image / Video URL", placeholder="https://...", lines=1)
@@ -653,7 +661,7 @@ def create_demo():
                     # write with a proper video extension
                     tmp_video = _temp_file(raw, suffix=ext_from_src(url) or ".mp4")
                     progress(0.15, desc="Preparing preview")
-                    preview_path = _convert_video_for_preview_if_needed(tmp_video)
+                    preview_path = _make_preview(url, raw)
         
                     progress(0.25, desc="Running full‑video analysis")
                     result = analyze_video_cohesive(client, tmp_video, prompt, progress=progress)
@@ -670,7 +678,7 @@ def create_demo():
                     raw = fetch_bytes(url, progress=progress)
         
                     # preview image (always JPEG for consistency)
-                    preview_path = _temp_file(convert_to_jpeg_bytes(raw, base_h=1024), suffix=".jpg")
+                    preview_path = _make_preview(url, raw)
         
                     progress(0.20, desc="Running image analysis")
                     result = analyze_image_structured(client, raw, prompt, progress=progress)
@@ -736,7 +744,25 @@ def create_demo():
                 "error": "Retry",
             }
             return labels.get(s, "Submit")  # Default case
+
+        # after submit_btn / clear_btn definitions
+        cancel_btn = gr.Button("Cancel", variant="secondary")
+        
+        def _cancel():
+            # abort any queued jobs
+            demo.cancel_all()
+            return "idle", "", "", "idle", "Idle", "", ""
         
+        cancel_btn.click(
+            fn=_cancel,
+            inputs=[],
+            outputs=[
+                status_state, output_md, preview_path_state,
+                progress_md, preview_status, preview_image, preview_video,
+            ],
+            queue=False,
+        )
+            
         status_state.change(fn=lambda s: _btn_label_for_status(s), inputs=[status_state], outputs=[submit_btn])
 
         # map status to progress text
@@ -744,11 +770,29 @@ def create_demo():
             return {"idle":"Idle","busy":"Processing…","done":"Completed","error":"Error — see output"}.get(s, s)
         status_state.change(fn=status_to_progress_text, inputs=[status_state], outputs=[progress_md])
 
+        preview_cache = {}   # module‑level dict: url → (preview_path, is_video)
+
+        def _make_preview(url: str, raw: bytes) -> str:
+            # reuse existing preview if we already have one for this URL
+            if url in preview_cache:
+                return preview_cache[url][0]
+        
+            if determine_media_type(url)[1]:          # video
+                tmp = _temp_file(raw, suffix=ext_from_src(url) or ".mp4")
+                preview = _convert_video_for_preview_if_needed(tmp)
+                # keep the raw temp file for later clean‑up
+                preview_cache[url] = (preview, True)
+            else:                                      # image
+                preview = _temp_file(convert_to_jpeg_bytes(raw, base_h=1024), suffix=".jpg")
+                preview_cache[url] = (preview, False)
+            return preview
+
         # when preview_path_state changes, update preview components appropriately
         # Update preview logic in apply_preview function
-        def apply_preview(path: str):
+        def apply_preview(path: str, last_path: str):
+            if path == last_path:
             if not path:
-                return gr.update(value=None, visible=False), gr.update(value=None, visible=False), ""
+                return gr.update(), gr.update(), ""
             
             try:
                 # Check for valid image previews