Update app.py

app.py

@@ -1,6 +1,4 @@
 #!/usr/bin/env python3
-# app.py - Gradio UI wrapper for your media analysis functions
-
 import os
 from io import BytesIO
 from typing import Tuple
@@ -8,17 +6,6 @@ from typing import Tuple
 import gradio as gr
 from PIL import Image
 
-# Import or copy your existing helpers here:
-# - process_media(src, custom_prompt, api_key, progress=gr.Progress())
-# - fetch_bytes, determine_media_type, ext_from_src, is_remote, safe_head, safe_get
-# - IMAGE_EXTS, VIDEO_EXTS
-# For example, if they are in analysis.py:
-# from analysis import process_media, fetch_bytes, determine_media_type, ext_from_src, is_remote, safe_head, safe_get, IMAGE_EXTS, VIDEO_EXTS
-
-# --- Begin: Minimal adapters if you want to inline small helpers (optional) ---
-# If you already have those functions in another module, import them instead of duplicating.
-# --- End adapters ---
-
 css = """
 .preview_media img, .preview_media video { max-width: 100%; height: auto; border-radius: 6px; }
 .top_preview { display:flex; gap:12px; align-items:flex-start; flex-direction:column; }
@@ -34,7 +21,6 @@ def _btn_label_for_status(status: str) -> str:
     }.get(status or "idle", "Submit")
 
 
-# preview loader returns image/video updates (use gr.update for component props)
 def load_preview(url: str):
     empty_img = gr.update(value=None, visible=False)
     empty_vid = gr.update(value=None, visible=False)
@@ -74,52 +60,37 @@ def load_preview(url: str):
         return empty_img, empty_vid, gr.update(visible=False)
 
 
-# Create the Blocks app
 def create_demo():
     with gr.Blocks(css=css, title="Media Analysis") as demo:
-        # Top preview column
         with gr.Row():
             with gr.Column(scale=1):
                 preview_image = gr.Image(label="Preview Image", type="pil", visible=False, elem_classes="preview_media")
                 preview_video = gr.Video(label="Preview Video", visible=False, elem_classes="preview_media")
-                # PiP button - shown only when a video preview is visible
                 pip_button = gr.Button("Open Video in PiP", visible=False, elem_classes="pip_button")
             with gr.Column(scale=1):
-                # Inputs and controls
                 url_input = gr.Textbox(label="Image / Video URL or local path", placeholder="https://... or /path/to/file", lines=1)
-                # Collapsible prompt (Accordion)
                 with gr.Accordion("Prompt (optional)", open=False):
                     custom_prompt = gr.Textbox(label="Prompt", lines=4, value="")
                 with gr.Accordion("Mistral API Key (optional)", open=False):
                     api_key = gr.Textbox(label="API Key", type="password", max_lines=1)
-                # Submit immediately below preview (placed here visually)
                 submit_btn = gr.Button(_btn_label_for_status("idle"), variant="primary")
                 clear_btn = gr.Button("Clear")
-                # Output area
                 final_md = gr.Markdown(value="", label="Result")
 
-        # State to track status
         status = gr.State("idle")
 
-        # Live preview: update preview image/video and PiP visibility
         url_input.change(fn=load_preview, inputs=[url_input], outputs=[preview_image, preview_video, pip_button])
 
-        # Clear handler returns exactly the outputs listed
         def do_clear():
             return "", gr.update(value=None, visible=False), gr.update(value=None, visible=False), gr.update(value="", visible=True)
         clear_btn.click(fn=do_clear, inputs=[], outputs=[url_input, preview_image, preview_video, final_md])
 
-        # Start busy — MUST return outputs for both status (gr.State) and the Button (as gr.update)
         def start_busy():
             return "busy", gr.update(value=_btn_label_for_status("busy"), interactive=False)
 
-        # Worker runs your process_media (queued)
         def worker(url: str, prompt: str, key: str, progress=gr.Progress()):
-            # process_media returns a single string result (or error string)
             return process_media(url or "", prompt or "", key or "", progress=progress)
 
-        # Finish: adapt result into status and Markdown update.
-        # Return (status_string, gr.update(value=md_text)).
         def finish(result: str):
             s = "done"
             if not result:
@@ -130,49 +101,26 @@ def create_demo():
                 s = "error"
             else:
                 md = result
-            # status as raw string (gr.State), final_md as gr.update
             return s, gr.update(value=md)
 
-        # Wire submit button: first click sets busy state & disables button
-        submit_btn.click(
-            fn=start_busy,
-            inputs=[],
-            outputs=[status, submit_btn],
-        )
+        submit_btn.click(fn=start_busy, inputs=[], outputs=[status, submit_btn])
 
-        # Then run worker (queued), produce final_md, then run finish to update status & final_md
-        submit_btn.click(
-            fn=worker,
-            inputs=[url_input, custom_prompt, api_key],
-            outputs=[final_md],
-            queue=True,
-        ).then(
-            fn=finish,
-            inputs=[final_md],
-            outputs=[status, final_md],
+        submit_btn.click(fn=worker, inputs=[url_input, custom_prompt, api_key], outputs=[final_md], queue=True).then(
+            fn=finish, inputs=[final_md], outputs=[status, final_md]
         )
 
-        # Status change -> update button label/interactivity
         def btn_label_for_state(s: str):
             return gr.update(value=_btn_label_for_status(s), interactive=(s != "busy"))
-
         status.change(fn=btn_label_for_state, inputs=[status], outputs=[submit_btn])
 
-        # PiP JS glue: open the current preview_video element in Picture-in-Picture.
-        pip_js = """
-        () => {
-            const vid = document.querySelector('#{} video, #{} video');
-            if (!vid) { return; }
-            // Request PiP
-            if (document.pictureInPictureElement) {
-                document.exitPictureInPicture();
-            } else {
-                vid.requestPictureInPicture().catch(e => console.warn('PiP failed', e));
-            }
-        }
-        """.format(preview_video.elem_id or "preview_video", preview_video.elem_id or "preview_video")
-
-        # Set PiP button to run client-side JS when clicked
+        # PiP JS: use a JS function string without Python .format interpolation
+        pip_js = (
+            "const vid = document.querySelector('video');"
+            "if (!vid) return;"
+            "if (document.pictureInPictureElement) { document.exitPictureInPicture().catch(()=>{}); }"
+            "else { vid.requestPictureInPicture().catch(()=>{}); }"
+        )
+        # Attach PiP: client-side JS (no inputs/outputs)
         pip_button.click(fn=None, _js=pip_js, inputs=[], outputs=[])
 
     return demo