Add application file

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 from huggingface_hub import hf_hub_download
 import pandas as pd
 import tempfile
-import re
+import base64
 
 HF_REPO_ID = "stcoats/temp-duckdb-upload"
 HF_FILENAME = "ycsep.duckdb"
@@ -15,14 +15,14 @@ st.title("YCSEP Audio Dataset Viewer")
 
 # Download database if missing
 if not os.path.exists(LOCAL_PATH):
-    st.write("Downloading from HF Hub...")
-    hf_hub_download(
-        repo_id=HF_REPO_ID,
-        repo_type="dataset",
-        filename=HF_FILENAME,
-        local_dir=".",
-        local_dir_use_symlinks=False
-    )
+    with st.spinner("Downloading from HF Hub..."):
+        hf_hub_download(
+            repo_id=HF_REPO_ID,
+            repo_type="dataset",
+            filename=HF_FILENAME,
+            local_dir=".",
+            local_dir_use_symlinks=False
+        )
     st.success("Download complete.")
 
 # Connect (only once)
@@ -61,7 +61,7 @@ st.markdown(f"### Showing {len(df)} results")
 if len(df) == 0:
     st.warning("No matches found.")
 else:
-    def render_audio_tag(audio_bytes, index):
+    def get_audio_html(audio_bytes):
         try:
             if isinstance(audio_bytes, (bytes, bytearray, memoryview)):
                 data = bytes(audio_bytes)
@@ -69,46 +69,42 @@ else:
                 data = bytes(audio_bytes)
             else:
                 return ""
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
-                tmp.write(data)
-                tmp.flush()
-                return f'<audio controls preload="metadata" style="height:20px;width:100px;"><source src="file://{tmp.name}" type="audio/mpeg"></audio>'
+            b64 = base64.b64encode(data).decode("utf-8")
+            return f'<audio controls preload="metadata" style="height:20px;width:120px;"><source src="data:audio/mp3;base64,{b64}" type="audio/mpeg"></audio>'
         except Exception:
             return ""
 
-    df["audio_player"] = [render_audio_tag(b, i) for i, b in enumerate(df["audio"])]
-    df_display = df.drop(columns=["audio"])
+    df["Audio"] = df["audio"].apply(get_audio_html)
+    df.drop(columns=["audio"], inplace=True)
 
     from streamlit.components.v1 import html
 
-    def display_html_table(df):
+    def display_table_with_audio(df):
         table_html = """
         <table border='1' style='border-collapse:collapse;width:100%;font-size:13px;'>
             <thead>
                 <tr>
-                    <th style='width:8em;'>id</th>
-                    <th>channel</th>
-                    <th>video_id</th>
-                    <th>speaker</th>
-                    <th>start_time</th>
-                    <th>end_time</th>
-                    <th>upload_date</th>
-                    <th>text</th>
-                    <th>pos_tags</th>
-                    <th>audio</th>
+                    <th style='width:6em;'>id</th>
+                    <th style='width:6em;'>channel</th>
+                    <th style='width:6em;'>video_id</th>
+                    <th style='width:6em;'>speaker</th>
+                    <th style='width:6em;'>start_time</th>
+                    <th style='width:6em;'>end_time</th>
+                    <th style='width:6em;'>upload_date</th>
+                    <th style='width:25em;'>text</th>
+                    <th style='width:10em;'>pos_tags</th>
+                    <th style='width:10em;'>Audio</th>
                 </tr>
             </thead>
             <tbody>
         """
         for _, row in df.iterrows():
             table_html += "<tr>"
-            for col in ["id", "channel", "video_id", "speaker", "start_time", "end_time", "upload_date", "text", "pos_tags"]:
+            for col in ["id", "channel", "video_id", "speaker", "start_time", "end_time", "upload_date", "text", "pos_tags", "Audio"]:
                 table_html += f"<td>{row[col]}</td>"
-            table_html += f"<td>{row['audio_player']}</td>"
             table_html += "</tr>"
         table_html += "</tbody></table>"
         return table_html
 
     st.markdown("### Results Table (Sortable with Audio Column)")
-    html(display_html_table(df_display), height=800, scrolling=True)
-
+    html(display_table_with_audio(df), height=900, scrolling=True)