Update app.py

app.py

@@ -442,7 +442,7 @@ def display_papers_in_sidebar(papers):
 # 4. ZIP FUNCTION
 # ─────────────────────────────────────────────────────────
 
-def create_zip_of_files_mp3(mp3_files, wav_files, input_question):
+def create_zip_of_files(md_files, 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,7 +454,10 @@ def create_zip_of_files_mp3(mp3_files, wav_files, input_question):
 
     all_content = []
     for f in all_files:
-        if f.endswith('.mp3') or f.endswith('.wav'):
+        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'):
             basename = os.path.splitext(os.path.basename(f))[0]
             words = basename.replace('_', ' ')
             all_content.append(words)
@@ -472,35 +475,6 @@ def create_zip_of_files_mp3(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
 # ─────────────────────────────────────────────────────────
@@ -633,33 +607,16 @@ def display_file_history_in_sidebar():
     col1, col4 = st.sidebar.columns(2)
     with col1:
         if st.button("🗑 Delete All"):
-            for f in all_files:
+            for f in all_md:
                 os.remove(f)
-                st.write('Deleted: ' + f)
-            #st.session_state.should_rerun = True
-            st.rerun()
-        if st.button("🗑 Delete All MD"):
-            for f in md_files:
+            for f in all_mp3:
                 os.remove(f)
-                st.write('Deleted: ' + f)
-            #st.session_state.should_rerun = True
-            st.rerun()
-        if st.button("🗑 Delete All MP3"):
-            for f in mp3_files:
+            for f in all_wav:
                 os.remove(f)
-                st.write('Deleted: ' + f)
-            for f in wav_files:
-                os.remove(f)
-                st.write('Deleted: ' + f)
-            #st.session_state.should_rerun = True
-            st.rerun()
+            st.session_state.should_rerun = True
     with col4:
-        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 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)