Update app.py

app.py

@@ -2,54 +2,32 @@ import streamlit as st
 import anthropic, openai, base64, cv2, glob, json, math, os, pytz, random, re, requests, time, zipfile
 from datetime import datetime
 from audio_recorder_streamlit import audio_recorder
-from collections import defaultdict, Counter
+from collections import defaultdict
 from dotenv import load_dotenv
 from gradio_client import Client
 from huggingface_hub import InferenceClient
 from PIL import Image
-from openai import OpenAI
+from streamlit_marquee import streamlit_marquee
 import asyncio
 import edge_tts
-from streamlit_marquee import streamlit_marquee
-
-st.set_page_config(
-    page_title="🚲TalkingAIResearcher🏆",
-    page_icon="🚲🏆",
-    layout="wide"
-)
 
-EDGE_TTS_VOICES = [
-    "en-US-AriaNeural",
-    "en-US-GuyNeural", 
-    "en-US-JennyNeural",
-    "en-GB-SoniaNeural"
-]
+st.set_page_config(page_title="🚲TalkingAIResearcher🏆", page_icon="🚲🏆", layout="wide")
 
-FILE_EMOJIS = {
-    "md": "📝",
-    "mp3": "🎵",
-    "wav": "🔊",
-    "txt": "📄",
-    "pdf": "📑"
-}
+EDGE_TTS_VOICES = ["en-US-AriaNeural", "en-US-GuyNeural", "en-US-JennyNeural", "en-GB-SoniaNeural"]
+FILE_EMOJIS = {"md": "📝", "mp3": "🎵", "wav": "🔊", "txt": "📄", "pdf": "📑"}
 
-# Initialize session states
-if 'tts_voice' not in st.session_state:
-    st.session_state['tts_voice'] = EDGE_TTS_VOICES[0]
-if 'audio_format' not in st.session_state:
-    st.session_state['audio_format'] = 'mp3'
-if 'messages' not in st.session_state:
-    st.session_state['messages'] = []
-if 'chat_history' not in st.session_state:
-    st.session_state['chat_history'] = []
-if 'viewing_prefix' not in st.session_state:
-    st.session_state['viewing_prefix'] = None
-if 'should_rerun' not in st.session_state:
-    st.session_state['should_rerun'] = False
-
-# API Setup
-openai_client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
-claude_client = anthropic.Anthropic(api_key=os.getenv('ANTHROPIC_API_KEY'))
+# Session state initialization
+for key, default in {
+    'tts_voice': EDGE_TTS_VOICES[0],
+    'audio_format': 'mp3',
+    'messages': [],
+    'chat_history': [],
+    'transcript_history': [],
+    'viewing_prefix': None,
+    'should_rerun': False
+}.items():
+    if key not in st.session_state:
+        st.session_state[key] = default
 
 @st.cache_resource
 def get_cached_audio_b64(file_path):
@@ -60,33 +38,25 @@ def beautify_filename(filename):
     name = os.path.splitext(filename)[0]
     return name.replace('_', ' ').replace('.', ' ')
 
-def create_zip_of_files(md_files, mp3_files, wav_files, query=''):
-    all_files = md_files + mp3_files + wav_files
-    if not all_files: return None
-    
-    timestamp = datetime.now().strftime("%y%m_%H%M")
-    zip_name = f"{timestamp}_archive.zip"
-    with zipfile.ZipFile(zip_name, 'w') as z:
-        for f in all_files:
-            z.write(f)
-    return zip_name
-
-def get_download_link(file_path, file_type="zip"):
-    with open(file_path, "rb") as f:
-        b64 = base64.b64encode(f.read()).decode()
-    ext_map = {'zip': '📦', 'mp3': '🎵', 'wav': '🔊', 'md': '📝'}
-    emoji = ext_map.get(file_type, '')
-    return f'<a href="data:application/{file_type};base64,{b64}" download="{os.path.basename(file_path)}">{emoji} Download {os.path.basename(file_path)}</a>'
-
 def load_files_for_sidebar():
-    files = [f for f in glob.glob("*.*") if not f.lower().endswith('readme.md')]
+    """Load and filter files for sidebar by timestamp prefix"""
+    files = []
+    for f in glob.glob("*.*"):
+        basename = os.path.basename(f)
+        if f.endswith('.md'):
+            if len(basename) >= 9 and re.match(r'\d{4}_\d{4}', basename[:9]):
+                files.append(f)
+        else:
+            files.append(f)
+            
     groups = defaultdict(list)
     for f in files:
         basename = os.path.basename(f)
         group_name = basename[:9] if len(basename) >= 9 else 'Other'
         groups[group_name].append(f)
+        
     return sorted(groups.items(), 
-                 key=lambda x: max(os.path.getmtime(f) for f in x[1]), 
+                 key=lambda x: max(os.path.getmtime(f) for f in x[1]),
                  reverse=True)
 
 def display_marquee_controls():
@@ -108,6 +78,23 @@ def display_marquee_controls():
         "lineHeight": "35px"
     }
 
+def get_download_link(file_path, file_type="zip"):
+    with open(file_path, "rb") as f:
+        b64 = base64.b64encode(f.read()).decode()
+    ext_map = {'zip': '📦', 'mp3': '🎵', 'wav': '🔊', 'md': '📝'}
+    emoji = ext_map.get(file_type, '')
+    return f'<a href="data:application/{file_type};base64,{b64}" download="{os.path.basename(file_path)}">{emoji} Download {os.path.basename(file_path)}</a>'
+
+def create_zip_of_files(md_files, mp3_files, wav_files, query=''):
+    all_files = md_files + mp3_files + wav_files
+    if not all_files: return None
+    timestamp = datetime.now().strftime("%y%m_%H%M")
+    zip_name = f"{timestamp}_archive.zip"
+    with zipfile.ZipFile(zip_name, 'w') as z:
+        for f in all_files:
+            z.write(f)
+    return zip_name
+
 def display_file_manager_sidebar(groups_sorted):
     st.sidebar.title("📚 File Manager")
     all_files = {'md': [], 'mp3': [], 'wav': []}
@@ -124,18 +111,18 @@ def display_file_manager_sidebar(groups_sorted):
             if st.button(f"🗑️ {ext.upper()}"):
                 [os.remove(f) for f in files]
                 st.session_state.should_rerun = True
-                
+
     if st.sidebar.button("📦 Zip All"):
-        zip_name = create_zip_of_files(
-            all_files['md'], all_files['mp3'], all_files['wav']
-        )
+        zip_name = create_zip_of_files(all_files['md'], all_files['mp3'], all_files['wav'])
         if zip_name:
             st.sidebar.markdown(get_download_link(zip_name), unsafe_allow_html=True)
 
     for group_name, files in groups_sorted:
-        timestamp = (datetime.strptime(group_name, "%y%m_%H%M").strftime("%Y-%m-%d %H:%M") 
-                    if len(group_name) == 9 else group_name)
-        
+        try:
+            timestamp = datetime.strptime(group_name, "%y%m_%H%M").strftime("%Y-%m-%d %H:%M") if len(group_name) == 9 and group_name != 'Other' else group_name
+        except ValueError:
+            timestamp = group_name
+
         with st.sidebar.expander(f"📁 {timestamp} ({len(files)})", expanded=True):
             c1, c2 = st.columns(2)
             with c1:
@@ -157,14 +144,12 @@ def display_file_manager_sidebar(groups_sorted):
                     if st.button("🔄", key=f"loop_{f}"):
                         audio_b64 = get_cached_audio_b64(f)
                         st.components.v1.html(
-                            f'''
-                            <audio id="player_{f}" loop>
+                            f'''<audio id="player_{f}" loop>
                                 <source src="data:audio/{ext};base64,{audio_b64}">
-                            </audio>
-                            <script>
+                               </audio>
+                               <script>
                                 document.getElementById("player_{f}").play();
-                            </script>
-                            ''',
+                               </script>''',
                             height=0
                         )
 
@@ -220,7 +205,7 @@ def perform_ai_lookup(query):
             **marquee_settings,
             key=f"paper_{paper['id'] or random.randint(1000,9999)}"
         )
-        st.write("")  # Spacing
+        st.write("")
     
     return papers