Fix YouTube API initialization race condition on simulation switch

app.py

@@ -49,9 +49,9 @@ custom_simulation_store = {
 }
 
 def play_simulation_selection(selection: str) -> str:
-    if selection == "Steve Jobs on Computer Programming and Predicting AI (1983) (Demo)":
+    if selection == "Steve Jobs 1983 Speech (Demo)":
         return get_demo_html()
-    elif selection == "Carl Sagan on the Importance of Scientific Literacy (Demo)":
+    elif selection == "Carl Sagan Demon-Haunted World (Demo)":
         return get_demo_2_html()
     elif selection == "My Custom Simulation":
         if custom_simulation_store["video_id"] and custom_simulation_store["chat_data"]:
@@ -61,63 +61,31 @@ 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. Extract YouTube Video ID with fallback to uploaded filename or SRT text content
+    # 1. Validate YouTube Link
     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 in the input box, or name your uploaded transcript file with the video ID (e.g. `v=VIDEO_ID.srt`).",
+            "### ❌ Error\nInvalid YouTube URL. Please provide a valid YouTube link or 11-character Video ID.",
             gr.update()
         )
         
-    # 2. Identify Document Source and Transcript Source
+    # 2. Identify Document 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:
-        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
+        doc_path = pdf_file.name
+    elif doc_text.strip():
+        document_content = doc_text.strip()
     else:
-        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
+        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
     
     # 4. Use provided token or default token
     token = hf_token.strip() if hf_token.strip() else DEFAULT_HF_TOKEN
@@ -330,7 +298,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", ".srt"],
+                        file_types=[".pdf", ".txt"],
                         file_count="single"
                     )
                     

theater_template.html

@@ -317,9 +317,6 @@
     </div>
   </div>
 
-  <!-- YouTube IFrame API -->
-  <script src="https://www.youtube.com/iframe_api"></script>
-  
   <script>
     // Embedded Chat Data Placeholder (Replaced by Python script)
     const chatData = {{CHAT_DATA_JSON}};
@@ -482,7 +479,6 @@
       player = new YT.Player('player', {
         height: '100%',
         width: '100%',
-        host: 'https://www.youtube-nocookie.com',
         videoId: videoId,
         playerVars: {
           'playsinline': 1,
@@ -512,6 +508,20 @@
 
     // Initialize
     prepareMessages();
+
+    // Asynchronously load the YouTube IFrame Player API
+    if (window.YT && window.YT.Player) {
+      onYouTubeIframeAPIReady();
+    } else {
+      var tag = document.createElement('script');
+      tag.src = "https://www.youtube.com/iframe_api";
+      var firstScriptTag = document.getElementsByTagName('script')[0];
+      if (firstScriptTag) {
+        firstScriptTag.parentNode.insertBefore(tag, firstScriptTag);
+      } else {
+        document.head.appendChild(tag);
+      }
+    }
   </script>
 </body>
 </html>