Fix Sagan video loading: robust fallback video ID extraction from filename/transcript, SRT file upload support, and privacy-enhanced YouTube host

app.py

@@ -61,31 +61,63 @@ def play_simulation_selection(selection: str) -> str:
     return ""
 
 def handle_generation(yt_url: str, pdf_file, doc_text: str, srt_text: str, hf_token: str):
-    # 1. Validate YouTube Link
+    # 1. Extract YouTube Video ID with fallback to uploaded filename or SRT text content
     video_id = extract_youtube_video_id(yt_url)
+    
+    # Fallback 1: Extract from the uploaded PDF/text filename if yt_url is empty
+    if not video_id and pdf_file is not None:
+        filename = os.path.basename(pdf_file.name)
+        video_id = extract_youtube_video_id(filename)
+        
+    # Fallback 2: Extract from the SRT text content (first 5 lines) if still empty
+    if not video_id and srt_text:
+        lines = srt_text.split('\n')[:5]
+        for line in lines:
+            extracted = extract_youtube_video_id(line)
+            if extracted:
+                video_id = extracted
+                break
+
     if not video_id:
         return (
             gr.update(),
-            "### ❌ Error\nInvalid YouTube URL. Please provide a valid YouTube link or 11-character Video ID.",
+            "### ❌ Error\nInvalid YouTube URL. Please provide a valid YouTube link or 11-character Video ID in the input box, or name your uploaded transcript file with the video ID (e.g. `v=VIDEO_ID.srt`).",
             gr.update()
         )
         
-    # 2. Identify Document Source
+    # 2. Identify Document Source and Transcript Source
     doc_path = None
     document_content = None
+    manual_transcript = None
+    
+    # Check if the uploaded file is actually an SRT transcript
+    uploaded_as_srt = False
     if pdf_file is not None:
-        doc_path = pdf_file.name
-    elif doc_text.strip():
-        document_content = doc_text.strip()
+        filename = os.path.basename(pdf_file.name)
+        if filename.endswith(".srt") or filename.endswith(".srt.txt") or "v=" in filename:
+            uploaded_as_srt = True
+            
+    if uploaded_as_srt:
+        # Read the uploaded SRT file as the manual transcript
+        with open(pdf_file.name, "r", encoding="utf-8") as f:
+            manual_transcript = f.read()
+        # If no separate reference document is pasted, default to using the transcript itself
+        if doc_text.strip():
+            document_content = doc_text.strip()
+        else:
+            document_content = manual_transcript
     else:
-        return (
-            gr.update(),
-            "### ❌ Error\nPlease upload a PDF/text file or paste some reference document text.",
-            gr.update()
-        )
-        
-    # 3. Clean manual transcript input if any
-    manual_transcript = srt_text.strip() if srt_text.strip() else None
+        if pdf_file is not None:
+            doc_path = pdf_file.name
+        elif doc_text.strip():
+            document_content = doc_text.strip()
+        else:
+            return (
+                gr.update(),
+                "### ❌ Error\nPlease upload a PDF/text file or paste some reference document text.",
+                gr.update()
+            )
+        manual_transcript = srt_text.strip() if srt_text.strip() else None
     
     # 4. Use provided token or default token
     token = hf_token.strip() if hf_token.strip() else DEFAULT_HF_TOKEN
@@ -298,7 +330,7 @@ with gr.Blocks(title="ReLiveStream - Interactive Replay", theme=custom_theme, cs
                     
                     pdf_input = gr.File(
                         label="Upload Reference PDF/Text",
-                        file_types=[".pdf", ".txt"],
+                        file_types=[".pdf", ".txt", ".srt"],
                         file_count="single"
                     )
                     

theater_template.html

@@ -482,6 +482,7 @@
       player = new YT.Player('player', {
         height: '100%',
         width: '100%',
+        host: 'https://www.youtube-nocookie.com',
         videoId: videoId,
         playerVars: {
           'playsinline': 1,