Add provenance section below video player in theater mode

app.py

@@ -63,30 +63,94 @@ def get_demo_4_html() -> str:
 # Global store for custom simulations
 custom_simulation_store = {
     "video_id": "",
-    "chat_data": None
+    "chat_data": None,
+    "pdf_name": None,
+    "has_pasted_text": False
 }
 
-def play_simulation_selection(selection: str) -> str:
+def get_provenance_html(selection: str) -> str:
+    apa_citations = {
+        "Steve Jobs 1983 Speech (Demo)": (
+            "Zohar, E., Bloom, P., & Inzlicht, M. (2026). Against frictionless AI. "
+            "<i>Communications Psychology</i>, 4(1), Article 402. "
+            "<a href='https://doi.org/10.1038/s44271-026-00402-1' target='_blank' style='color:#7f5af0; text-decoration:none;'>https://doi.org/10.1038/s44271-026-00402-1</a>"
+        ),
+        "Carl Sagan Demon-Haunted World (Demo)": (
+            "Giroux, H. A. (2013). Beyond dystopian education in a neoliberal society. "
+            "<i>Fast Capitalism</i>, 10(1). "
+            "<a href='https://doi.org/10.32855/fcapital.201301.010' target='_blank' style='color:#7f5af0; text-decoration:none;'>https://doi.org/10.32855/fcapital.201301.010</a>"
+        ),
+        "Edward Teller - Schrödinger's Cat (Demo)": (
+            "Claeys, G. (2010). The origins of dystopia: Wells, Huxley and Orwell. "
+            "In G. Claeys (Ed.), <i>The Cambridge Companion to Utopian Literature</i> (pp. 107–131). "
+            "Cambridge University Press. "
+            "<a href='https://doi.org/10.1017/CCOL9780521886659.005' target='_blank' style='color:#7f5af0; text-decoration:none;'>https://doi.org/10.1017/CCOL9780521886659.005</a>"
+        ),
+        "Sal Khan - Khanmigo AI Tutor (Demo)": (
+            "Pepple, D., & Muthuthantrige, N. (2026). Artificial intelligence, innovation and the new "
+            "architecture of exploitation: Towards reconfiguring humanness in the age of algorithmic labour. "
+            "<i>Journal of Innovation & Knowledge</i>, 11(1), 100878. "
+            "<a href='https://doi.org/10.1016/j.jik.2025.100878' target='_blank' style='color:#7f5af0; text-decoration:none;'>https://doi.org/10.1016/j.jik.2025.100878</a>"
+        )
+    }
+
+    if selection in apa_citations:
+        citation = apa_citations[selection]
+        return (
+            f"<div style='margin-top: 16px; padding: 14px 18px; background-color: #161a23; "
+            f"border: 1px solid rgba(255, 255, 255, 0.08); border-radius: 8px; font-family: sans-serif; "
+            f"color: #94a1b2; font-size: 0.9rem; line-height: 1.5;'>"
+            f"<span style='color: #fffffe; font-weight: 600; display: block; margin-bottom: 4px;'>📄 Provenance</span>"
+            f"Comments generated using AI from published works: {citation}"
+            f"</div>"
+        )
+    elif selection == "My Custom Simulation":
+        pdf_name = custom_simulation_store.get("pdf_name")
+        if pdf_name:
+            source_info = f"uploaded file <i>{html.escape(pdf_name)}</i>"
+        elif custom_simulation_store.get("has_pasted_text"):
+            source_info = "pasted custom reference text"
+        else:
+            source_info = None
+
+        if source_info:
+            return (
+                f"<div style='margin-top: 16px; padding: 14px 18px; background-color: #161a23; "
+                f"border: 1px solid rgba(255, 255, 255, 0.08); border-radius: 8px; font-family: sans-serif; "
+                f"color: #94a1b2; font-size: 0.9rem; line-height: 1.5;'>"
+                f"<span style='color: #fffffe; font-weight: 600; display: block; margin-bottom: 4px;'>📄 Provenance</span>"
+                f"Comments generated using AI from {source_info}."
+                f"</div>"
+            )
+
+    return ""
+
+
+def play_simulation_selection(selection: str) -> tuple[str, str]:
+    provenance_html = get_provenance_html(selection)
     if selection == "Steve Jobs 1983 Speech (Demo)":
-        return get_demo_html()
+        return get_demo_html(), provenance_html
     elif selection == "Carl Sagan Demon-Haunted World (Demo)":
-        return get_demo_2_html()
+        return get_demo_2_html(), provenance_html
     elif selection == "Edward Teller - Schrödinger's Cat (Demo)":
-        return get_demo_3_html()
+        return get_demo_3_html(), provenance_html
     elif selection == "Sal Khan - Khanmigo AI Tutor (Demo)":
-        return get_demo_4_html()
+        return get_demo_4_html(), provenance_html
     elif selection == "My Custom Simulation":
         if custom_simulation_store["video_id"] and custom_simulation_store["chat_data"]:
-            return load_template_with_data(custom_simulation_store["video_id"], custom_simulation_store["chat_data"])
+            player_html = load_template_with_data(custom_simulation_store["video_id"], custom_simulation_store["chat_data"])
         else:
-            return "<div style='color:#ff0055; text-align:center; padding:50px; font-family:sans-serif;'>No custom simulation has been generated yet. Please choose 'Configure Custom Solution'.</div>"
-    return ""
+            player_html = "<div style='color:#ff0055; text-align:center; padding:50px; font-family:sans-serif;'>No custom simulation has been generated yet. Please choose 'Configure Custom Solution'.</div>"
+        return player_html, provenance_html
+    return "", ""
+
 
 def handle_generation(yt_url: str, pdf_file, doc_text: str, srt_text: str, hf_token: str, use_ocr: bool = False):
     # 1. Validate YouTube Link
     video_id = extract_youtube_video_id(yt_url)
     if not video_id:
         return (
+            gr.update(),
             gr.update(),
             "### ❌ Error\nInvalid YouTube URL. Please provide a valid YouTube link or 11-character Video ID.",
             gr.update()
@@ -95,12 +159,17 @@ def handle_generation(yt_url: str, pdf_file, doc_text: str, srt_text: str, hf_to
     # 2. Identify Document Source
     doc_path = None
     document_content = None
+    pdf_name = None
+    has_pasted_text = False
     if pdf_file is not None:
         doc_path = pdf_file.name
+        pdf_name = os.path.basename(pdf_file.name)
     elif doc_text.strip():
         document_content = doc_text.strip()
+        has_pasted_text = True
     else:
         return (
+            gr.update(),
             gr.update(),
             "### ❌ Error\nPlease upload a PDF/text file or paste some reference document text.",
             gr.update()
@@ -110,6 +179,7 @@ def handle_generation(yt_url: str, pdf_file, doc_text: str, srt_text: str, hf_to
     manual_transcript = srt_text.strip() if srt_text.strip() else None
     if not manual_transcript:
         return (
+            gr.update(),
             gr.update(),
             "### ❌ Error\nPlease paste the timestamped video transcript. (Auto-fetching is disabled due to server IP blocks).",
             gr.update()
@@ -139,6 +209,8 @@ def handle_generation(yt_url: str, pdf_file, doc_text: str, srt_text: str, hf_to
         # Save to global store
         custom_simulation_store["video_id"] = video_id
         custom_simulation_store["chat_data"] = chat_data
+        custom_simulation_store["pdf_name"] = pdf_name
+        custom_simulation_store["has_pasted_text"] = has_pasted_text
         
         success_msg = (
             f"### 🎉 Success!\n"
@@ -148,9 +220,11 @@ def handle_generation(yt_url: str, pdf_file, doc_text: str, srt_text: str, hf_to
         
         # Create updated HTML player
         new_html = load_template_with_data(video_id, chat_data)
+        provenance_html = get_provenance_html("My Custom Simulation")
         
         return (
             new_html,
+            provenance_html,
             success_msg,
             gr.update(choices=["Steve Jobs 1983 Speech (Demo)", "Carl Sagan Demon-Haunted World (Demo)", "Edward Teller - Schrödinger's Cat (Demo)", "Sal Khan - Khanmigo AI Tutor (Demo)", "My Custom Simulation"], value="My Custom Simulation")
         )
@@ -163,11 +237,13 @@ def handle_generation(yt_url: str, pdf_file, doc_text: str, srt_text: str, hf_to
                 "document or a different video."
             )
         return (
+            gr.update(),
             gr.update(),
             error_msg,
             gr.update()
         )
 
+
 # Gradio Theme
 custom_theme = gr.themes.Default(
     primary_hue="purple",
@@ -265,7 +341,7 @@ button[role="tab"]:hover,
 }
 """
 
-with gr.Blocks(title="ReadStream", theme=custom_theme, css=custom_css) as demo:
+with gr.Blocks(title="ReadStream") as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 20px; padding-top: 10px;">
@@ -289,11 +365,14 @@ with gr.Blocks(title="ReadStream", theme=custom_theme, css=custom_css) as demo:
             # The player frame
             player_frame = gr.HTML(value=get_demo_html())
             
+            # The provenance/citation frame
+            provenance_frame = gr.HTML(value=get_provenance_html("Steve Jobs 1983 Speech (Demo)"))
+            
             # Trigger updates when selection changes
             sim_selector.change(
                 fn=play_simulation_selection,
                 inputs=[sim_selector],
-                outputs=[player_frame]
+                outputs=[player_frame, provenance_frame]
             )
             
         # Tab 2: Generator Config
@@ -358,6 +437,7 @@ with gr.Blocks(title="ReadStream", theme=custom_theme, css=custom_css) as demo:
                 ],
                 outputs=[
                     player_frame,
+                    provenance_frame,
                     status_output,
                     sim_selector
                 ]