Update app.py

app.py

@@ -442,7 +442,7 @@ def display_papers_in_sidebar(papers):
 # 4. ZIP FUNCTION
 # ─────────────────────────────────────────────────────────
 
-def create_zip_of_files(md_files, mp3_files, wav_files, input_question):
+def create_zip_of_files_mp3(mp3_files, wav_files, input_question):
     """
     Zip up all relevant files, limiting the final zip name to ~20 chars
     to avoid overly long base64 strings. 
@@ -454,10 +454,7 @@ def create_zip_of_files(md_files, mp3_files, wav_files, input_question):
 
     all_content = []
     for f in all_files:
-        if f.endswith('.md'):
-            with open(f, 'r', encoding='utf-8') as file:
-                all_content.append(file.read())
-        elif f.endswith('.mp3') or f.endswith('.wav'):
+        if f.endswith('.mp3') or f.endswith('.wav'):
             basename = os.path.splitext(os.path.basename(f))[0]
             words = basename.replace('_', ' ')
             all_content.append(words)
@@ -475,6 +472,35 @@ def create_zip_of_files(md_files, mp3_files, wav_files, input_question):
             z.write(f)
     return short_zip_name
 
+def create_zip_of_files_md(md_files input_question):
+    """
+    Zip up all relevant files, limiting the final zip name to ~20 chars
+    to avoid overly long base64 strings. 
+    """
+    md_files = [f for f in md_files if os.path.basename(f).lower() != 'readme.md']
+    all_files = md_files + mp3_files + wav_files
+    if not all_files:
+        return None
+
+    all_content = []
+    for f in all_files:
+        if f.endswith('.md'):
+            with open(f, 'r', encoding='utf-8') as file:
+                all_content.append(file.read())
+    
+    all_content.append(input_question)
+    combined_content = " ".join(all_content)
+    info_terms = get_high_info_terms(combined_content, top_n=10)
+    
+    timestamp = format_timestamp_prefix()
+    name_text = '-'.join(term for term in info_terms[:5])  
+    short_zip_name = (timestamp + "_" + name_text)[:20] + ".zip"
+
+    with zipfile.ZipFile(short_zip_name, 'w') as z:
+        for f in all_files:
+            z.write(f)
+    return short_zip_name
+    
 # ─────────────────────────────────────────────────────────
 # 5. MAIN LOGIC: AI LOOKUP & VOICE INPUT
 # ─────────────────────────────────────────────────────────
@@ -614,9 +640,23 @@ def display_file_history_in_sidebar():
             for f in all_wav:
                 os.remove(f)
             st.session_state.should_rerun = True
+        if st.button("🗑 Delete All MD"):
+            for f in all_md:
+                os.remove(f)
+            st.session_state.should_rerun = True
+        if st.button("🗑 Delete All MP3"):
+            for f in all_mp3:
+                os.remove(f)
+            for f in all_wav:
+                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 st.button("⬇️ Zip All MD"):
+            zip_name = create_zip_of_files_md(md_files, st.session_state.get('last_query', ''))
+            if zip_name:
+                st.sidebar.markdown(get_download_link(zip_name, "zip"), unsafe_allow_html=True)
+        if st.button("⬇️ Zip All MP3"):
+            zip_name = create_zip_of_files_mp3(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)