Update app.py

app.py

@@ -1,3 +1,21 @@
+import streamlit as st
+import pandas as pd
+import numpy as np
+from sentence_transformers import SentenceTransformer
+from sklearn.metrics.pairwise import cosine_similarity
+import requests
+from datetime import datetime
+
+# Initialize session state variables
+if 'search_history' not in st.session_state:
+    st.session_state['search_history'] = []
+if 'search_columns' not in st.session_state:
+    st.session_state['search_columns'] = []
+if 'initial_search_done' not in st.session_state:
+    st.session_state['initial_search_done'] = False
+if 'hf_token' not in st.session_state:
+    st.session_state['hf_token'] = None
+
 def fetch_dataset_info_auth(dataset_id, hf_token):
     """Fetch dataset information with authentication"""
     info_url = f"https://huggingface.co/api/datasets/{dataset_id}"
@@ -163,17 +181,12 @@ def render_video_result(result):
             
         if video_url:
             st.video(video_url)
-            if st.button(f"🔊 Audio Summary", key=f"audio_{result.get('video_id', '')}"):
-                summary = f"Video summary: {result.get('title', '')}. {result.get('description', '')[:200]}"
-                audio_file = asyncio.run(generate_speech(summary))
-                if audio_file:
-                    st.audio(audio_file)
 
 def main():
-    st.title("🎥 Enhanced Video Search with Parquet Support")
+    st.title("🎥 Video Dataset Search")
     
     # Get HF token from secrets or user input
-    if 'hf_token' not in st.session_state:
+    if not st.session_state['hf_token']:
         st.session_state['hf_token'] = st.secrets.get("HF_TOKEN", None)
     
     if not st.session_state['hf_token']:
@@ -251,7 +264,7 @@ def main():
 
     # Sidebar
     with st.sidebar:
-        st.subheader("⚙️ Settings & History")
+        st.subheader("⚙️ Search History")
         if st.button("🗑️ Clear History"):
             st.session_state['search_history'] = []
             st.experimental_rerun()