Upload 5 files

page_modules/process_video.py

@@ -8,6 +8,7 @@ import subprocess
 from pathlib import Path
 
 import streamlit as st
+from PIL import Image, ImageDraw
 
 
 def _get_video_duration(path: str) -> float:
@@ -129,9 +130,13 @@ def render_process_video_page() -> None:
                     final_video_path = None
                     try:
                         duration = _get_video_duration(str(temp_path))
+                        duration_unknown = False
                         if not duration:
-                            st.error("No s'ha pogut obtenir la durada del vídeo.")
-                            is_valid = False
+                            st.warning(
+                                "No s'ha pogut obtenir la durada del vídeo. Es continuarà assumint un màxim de 4 minuts."
+                            )
+                            duration = float(MAX_DURATION_S)
+                            duration_unknown = True
 
                         if is_valid:
                             if duration > MAX_DURATION_S:
@@ -146,7 +151,7 @@ def render_process_video_page() -> None:
                                 _transcode_video(
                                     str(temp_path),
                                     str(final_video_path),
-                                    MAX_DURATION_S if was_truncated else None,
+                                    MAX_DURATION_S if (was_truncated or duration_unknown) else None,
                                 )
                             except RuntimeError as exc:
                                 st.error(f"No s'ha pogut processar el vídeo: {exc}")
@@ -157,7 +162,8 @@ def render_process_video_page() -> None:
                                 {
                                     "status": "processed",
                                     "path": str(final_video_path),
-                                    "was_truncated": was_truncated,
+                                    "was_truncated": was_truncated or duration_unknown,
+                                    "duration_unknown": duration_unknown,
                                 }
                             )
                             st.rerun()
@@ -190,13 +196,24 @@ def render_process_video_page() -> None:
                 ]
                 st.session_state.characters_saved = False
 
+    def _load_or_placeholder(path: str, size: tuple[int, int] = (150, 150)):
+        p = Path(path)
+        if p.exists():
+            return str(p)
+        img = Image.new("RGB", size, color=(230, 230, 230))
+        d = ImageDraw.Draw(img)
+        text = "No image"
+        tw, th = d.textlength(text), 12
+        d.text(((size[0]-tw)/2, (size[1]-th)/2), text, fill=(120, 120, 120))
+        return img
+
     if st.session_state.characters_detected:
         st.subheader("Personatges detectats")
         for char in st.session_state.characters_detected:
             with st.form(key=f"form_{char['id']}"):
                 col1, col2 = st.columns(2)
                 with col1:
-                    st.image(char["image_path"], width=150)
+                    st.image(_load_or_placeholder(char["image_path"]), width=150)
                 with col2:
                     st.caption(char["description"])
                     st.text_input("Nom del personatge", key=f"name_{char['id']}")