Update app.py

app.py

@@ -407,15 +407,112 @@ def create_paper_audio_files(papers, input_question):
             paper['full_audio'] = None
             paper['download_base64'] = ''
 
+            
+def display_file_history_in_sidebar():
+    """
+    Shows a history of files grouped by query, with lazy loading of audio and content.
+    """
+    st.sidebar.markdown("---")
+    st.sidebar.markdown("### 📂 File History")
+
+    # Gather all files
+    md_files = glob.glob("*.md")
+    mp3_files = glob.glob("*.mp3")
+    wav_files = glob.glob("*.wav")
+    all_files = md_files + mp3_files + wav_files
+
+    if not all_files:
+        st.sidebar.write("No files found.")
+        return
+
+    # Group files by their query prefix (timestamp_query)
+    grouped_files = {}
+    for f in all_files:
+        fname = os.path.basename(f)
+        prefix = '_'.join(fname.split('_')[:6])  # Get timestamp part
+        if prefix not in grouped_files:
+            grouped_files[prefix] = {'md': [], 'audio': [], 'loaded': False}
+        
+        ext = os.path.splitext(fname)[1].lower()
+        if ext == '.md':
+            grouped_files[prefix]['md'].append(f)
+        elif ext in ['.mp3', '.wav']:
+            grouped_files[prefix]['audio'].append(f)
+
+    # Sort groups by timestamp (newest first)
+    sorted_groups = sorted(grouped_files.items(), key=lambda x: x[0], reverse=True)
+
+    # 🗑⬇️ Sidebar delete all and zip all download
+    col1, col4 = st.sidebar.columns(2)
+    with col1:
+        if st.button("🗑 Delete All"):
+            for f in all_files:
+                os.remove(f)
+            st.session_state.should_rerun = True
+    with col4:
+        if st.button("⬇️ Zip All"):
+            zip_name = create_zip_of_files(md_files, mp3_files, wav_files, 
+                                         st.session_state.get('last_query', ''))
+            if zip_name:
+                st.sidebar.markdown(get_download_link(zip_name, "zip"), 
+                                  unsafe_allow_html=True)
+
+    # Display grouped files
+    for prefix, files in sorted_groups:
+        # Get a preview of content from first MD file
+        preview = ""
+        if files['md']:
+            with open(files['md'][0], "r", encoding="utf-8") as f:
+                preview = f.read(200).replace("\n", " ")
+                if len(preview) > 200:
+                    preview += "..."
+
+        # Create unique key for this group
+        group_key = f"group_{prefix}"
+        if group_key not in st.session_state:
+            st.session_state[group_key] = False
+
+        # Display group expander
+        with st.sidebar.expander(f"📑 Query Group: {prefix}"):
+            st.write("**Preview:**")
+            st.write(preview)
+            
+            # Load full content button
+            if st.button("📖 View Full Content", key=f"btn_{prefix}"):
+                st.session_state[group_key] = True
+
+            # Only show full content and audio if button was clicked
+            if st.session_state[group_key]:
+                # Display markdown files
+                for md_file in files['md']:
+                    with open(md_file, "r", encoding="utf-8") as f:
+                        content = f.read()
+                    st.markdown("**Full Content:**")
+                    st.markdown(content)
+                    st.markdown(get_download_link(md_file, file_type="md"), 
+                              unsafe_allow_html=True)
+
+                # Display audio files
+                for audio_file in files['audio']:
+                    ext = os.path.splitext(audio_file)[1].replace('.', '')
+                    st.audio(audio_file)
+                    st.markdown(get_download_link(audio_file, file_type=ext), 
+                              unsafe_allow_html=True)
+
 def display_papers(papers, marquee_settings):
-    """Display paper info in the main area with marquee + expanders + audio."""
+    """Display paper info with both abs and PDF links."""
     st.write("## Research Papers")
     for i, paper in enumerate(papers, start=1):
-        marquee_text = f"📄 {paper['title']} | 👤 {paper['authors'][:120]} | 📝 {paper['summary'][:200]}"
+        marquee_text = f"📄 {paper['title']} | 👤 {paper['authors'][:120]}"
         display_marquee(marquee_text, marquee_settings, key_suffix=f"paper_{i}")
         
         with st.expander(f"{i}. 📄 {paper['title']}", expanded=True):
-            st.markdown(f"**{paper['date']} | {paper['title']}** — [Arxiv Link]({paper['url']})")
+            # Create PDF link by replacing 'abs' with 'pdf' in arxiv URL
+            pdf_url = paper['url'].replace('/abs/', '/pdf/')
+            st.markdown(f"""
+                **{paper['date']} | {paper['title']}**  
+                📄 [Abstract]({paper['url']}) | 📑 [PDF]({pdf_url})
+            """)
             st.markdown(f"*Authors:* {paper['authors']}")
             st.markdown(paper['summary'])
             if paper.get('full_audio'):
@@ -425,19 +522,34 @@ def display_papers(papers, marquee_settings):
                     st.markdown(paper['download_base64'], unsafe_allow_html=True)
 
 def display_papers_in_sidebar(papers):
-    """Mirrors the paper listing in the sidebar with expanders, audio, etc."""
+    """Mirrors the paper listing in sidebar with lazy loading."""
     st.sidebar.title("🎶 Papers & Audio")
     for i, paper in enumerate(papers, start=1):
+        paper_key = f"paper_{paper['url']}"
+        if paper_key not in st.session_state:
+            st.session_state[paper_key] = False
+
         with st.sidebar.expander(f"{i}. {paper['title']}"):
-            st.markdown(f"**Arxiv:** [Link]({paper['url']})")
-            if paper['full_audio']:
+            # Create PDF link
+            pdf_url = paper['url'].replace('/abs/', '/pdf/')
+            st.markdown(f"📄 [Abstract]({paper['url']}) | 📑 [PDF]({pdf_url})")
+            
+            # Preview of authors and summary
+            st.markdown(f"**Authors:** {paper['authors'][:100]}...")
+            if paper['summary']:
+                st.markdown(f"**Summary:** {paper['summary'][:200]}...")
+            
+            # Load audio button
+            if paper['full_audio'] and st.button("🎵 Load Audio", 
+                                               key=f"btn_{paper_key}"):
+                st.session_state[paper_key] = True
+            
+            # Show audio player and download only if requested
+            if st.session_state[paper_key] and paper['full_audio']:
                 st.audio(paper['full_audio'])
                 if paper['download_base64']:
                     st.markdown(paper['download_base64'], unsafe_allow_html=True)
-            st.markdown(f"**Authors:** {paper['authors']}")
-            if paper['summary']:
-                st.markdown(f"**Summary:** {paper['summary'][:300]}...")
-
+                    
 # ─────────────────────────────────────────────────────────
 # 4. ZIP FUNCTION
 # ─────────────────────────────────────────────────────────