Update app.py

app.py

@@ -57,6 +57,10 @@ if 'edit_new_name' not in st.session_state:
     st.session_state['edit_new_name'] = ""
 if 'edit_new_content' not in st.session_state:
     st.session_state['edit_new_content'] = ""
+if 'viewing_file' not in st.session_state:
+    st.session_state['viewing_file'] = None
+if 'viewing_file_type' not in st.session_state:
+    st.session_state['viewing_file_type'] = None
 if 'should_rerun' not in st.session_state:
     st.session_state['should_rerun'] = False
 
@@ -84,14 +88,14 @@ def generate_filename(prompt, file_type="md"):
     return f"{date_str}_{safe}.{file_type}"
 
 def create_file(filename, prompt, response):
-    # Just create the file. Do not rerun automatically here.
+    # Creating file does not trigger immediate rerun
     with open(filename, 'w', encoding='utf-8') as f:
         f.write(prompt + "\n\n" + response)
 
 def get_download_link(file):
     with open(file, "rb") as f:
         b64 = base64.b64encode(f.read()).decode()
-    return f'<a href="data:file/txt;base64,{b64}" download="{os.path.basename(file)}">📂 Download {os.path.basename(file)}</a>'
+    return f'<a href="data:file/zip;base64,{b64}" download="{os.path.basename(file)}">📂 Download {os.path.basename(file)}</a>'
 
 @st.cache_resource
 def speech_synthesis_html(result):
@@ -106,7 +110,7 @@ def speech_synthesis_html(result):
     components.html(html_code, height=0)
 
 async def edge_tts_generate_audio(text, voice="en-US-AriaNeural", rate=0, pitch=0):
-    # Create mp3 file. Do not trigger rerun here.
+    # Just create mp3 file, no immediate rerun
     if not text.strip():
         return None
     rate_str = f"{rate:+d}%"
@@ -122,7 +126,8 @@ def speak_with_edge_tts(text, voice="en-US-AriaNeural", rate=0, pitch=0):
 def play_and_download_audio(file_path):
     if file_path and os.path.exists(file_path):
         st.audio(file_path)
-        st.markdown(get_download_link(file_path), unsafe_allow_html=True)
+        dl_link = f'<a href="data:audio/mpeg;base64,{base64.b64encode(open(file_path,"rb").read()).decode()}" download="{os.path.basename(file_path)}">Download {os.path.basename(file_path)}</a>'
+        st.markdown(dl_link, unsafe_allow_html=True)
 
 def process_image(image_path, user_prompt):
     with open(image_path, "rb") as imgf:
@@ -142,10 +147,10 @@ def process_image(image_path, user_prompt):
     return resp.choices[0].message.content
 
 def process_audio(audio_path):
-    # Creating transcription does not force rerun here.
     with open(audio_path, "rb") as f:
         transcription = openai_client.audio.transcriptions.create(model="whisper-1", file=f)
     st.session_state.messages.append({"role": "user", "content": transcription.text})
+    # No immediate rerun
     return transcription.text
 
 def process_video(video_path, seconds_per_frame=1):
@@ -200,7 +205,6 @@ def perform_ai_lookup(q, vocal_summary=True, extended_refs=False, titles_summary
 
     if vocal_summary:
         audio_file_main = speak_with_edge_tts(r2, voice="en-US-AriaNeural", rate=0, pitch=0)
-        # audio_file_main is created but we do not rerun
         st.write("### 🎙️ Vocal Summary (Short Answer)")
         play_and_download_audio(audio_file_main)
 
@@ -261,16 +265,24 @@ def process_with_claude(text):
         st.session_state.chat_history.append({"user":text,"claude":ans})
     return ans
 
-def create_zip_of_files():
-    md_files = glob.glob("*.md")
-    mp3_files = glob.glob("*.mp3")
+def create_zip_of_files(md_files, mp3_files):
+    # Exclude README.md if present
+    md_files = [f for f in md_files if os.path.basename(f).lower() != 'readme.md']
+
     all_files = md_files + mp3_files
-    zip_name = "all_files.zip"
+    if not all_files:
+        return None
+    # Build a descriptive name from file stems
+    stems = [os.path.splitext(os.path.basename(f))[0] for f in all_files]
+    # Join them
+    joined = "_".join(stems)
+    # Truncate if too long
+    if len(joined) > 50:
+        joined = joined[:50] + "_etc"
+    zip_name = f"{joined}.zip"
     with zipfile.ZipFile(zip_name,'w') as z:
         for f in all_files:
             z.write(f)
-    # user action triggered, set rerun
-    st.session_state.should_rerun = True
     return zip_name
 
 def get_media_html(p,typ="video",w="100%"):
@@ -285,11 +297,12 @@ def load_files_for_sidebar():
     md_files = glob.glob("*.md")
     mp3_files = glob.glob("*.mp3")
 
+    # Exclude README.md from listings
+    md_files = [f for f in md_files if os.path.basename(f).lower() != 'readme.md']
+
     files_by_ext = defaultdict(list)
-    for f in md_files:
-        files_by_ext['md'].append(f)
-    for f in mp3_files:
-        files_by_ext['mp3'].append(f)
+    if md_files: files_by_ext['md'].extend(md_files)
+    if mp3_files: files_by_ext['mp3'].extend(mp3_files)
 
     # Sort each extension group by modification time descending
     for ext in files_by_ext:
@@ -302,72 +315,64 @@ def display_file_manager_sidebar(files_by_ext):
     md_files = files_by_ext.get('md', [])
     mp3_files = files_by_ext.get('mp3', [])
 
-    col_del = st.sidebar.columns(2)
+    # Buttons to delete all except README.md (already excluded)
+    col_del = st.sidebar.columns(3)
     with col_del[0]:
-        if st.button("🗑 Delete All MD"):
+        if st.button("🗑 Del All MD"):
             for f in md_files:
                 os.remove(f)
             st.session_state.should_rerun = True
     with col_del[1]:
-        if st.button("🗑 Delete All MP3"):
+        if st.button("🗑 Del All MP3"):
             for f in mp3_files:
                 os.remove(f)
             st.session_state.should_rerun = True
+    with col_del[2]:
+        if st.button("⬇️ Zip All"):
+            # create a zip of all md and mp3 except README.md
+            z = create_zip_of_files(md_files, mp3_files)
+            if z:
+                st.sidebar.markdown(get_download_link(z),unsafe_allow_html=True)
 
     ext_counts = {ext: len(files) for ext, files in files_by_ext.items()}
     sorted_ext = sorted(files_by_ext.keys(), key=lambda x: ext_counts[x], reverse=True)
 
+    # Display files with actions
     for ext in sorted_ext:
         emoji = FILE_EMOJIS.get(ext, "📦")
         count = len(files_by_ext[ext])
-        with st.sidebar.expander(f"{emoji} {ext.upper()} Files ({count})"):
+        with st.sidebar.expander(f"{emoji} {ext.upper()} Files ({count})", expanded=True):
             for f in files_by_ext[ext]:
                 fname = os.path.basename(f)
                 ctime = datetime.fromtimestamp(os.path.getmtime(f)).strftime("%Y-%m-%d %H:%M:%S")
-                col1, col2, col3, col4 = st.columns([2,1,1,1])
-                with col1:
-                    st.write(f"**{fname}** - {ctime}")
-                with col2:
-                    # View button
-                    if ext == "md":
-                        if st.button("👀", key="view_"+f):
-                            content = open(f,'r',encoding='utf-8').read()
-                            st.write("**Viewing file content:**")
-                            st.markdown(content)
-                    else: # mp3
-                        if st.button("👀", key="view_"+f):
-                            st.write(f"Playing: {fname}")
-                            st.audio(f)
-                with col3:
-                    # Edit button for MD
+                # Show filename and buttons in a row
+                st.write(f"**{fname}** - {ctime}")
+                file_buttons_col = st.columns([1,1,1])
+                with file_buttons_col[0]:
+                    if st.button("👀View", key="view_"+f):
+                        st.session_state.viewing_file = f
+                        st.session_state.viewing_file_type = ext
+                        # No rerun needed, just set state
+                with file_buttons_col[1]:
                     if ext == "md":
-                        if st.button("✏️", key="edit_"+f):
+                        if st.button("✏️Edit", key="edit_"+f):
                             st.session_state.editing_file = f
                             st.session_state.edit_new_name = fname.replace(".md","")
                             st.session_state.edit_new_content = open(f,'r',encoding='utf-8').read()
-                            # user triggered, set rerun
                             st.session_state.should_rerun = True
-                    else:
-                        pass
-                with col4:
-                    # Delete button
-                    if st.button("🗑", key="del_"+f):
+                with file_buttons_col[2]:
+                    if st.button("🗑Del", key="del_"+f):
                         os.remove(f)
-                        # user triggered, set rerun
                         st.session_state.should_rerun = True
 
-    if (len(md_files) > 0 or len(mp3_files) > 0) and st.sidebar.button("⬇️ Download All (.md and .mp3)"):
-        z = create_zip_of_files()
-        st.sidebar.markdown(get_download_link(z),unsafe_allow_html=True)
-
     # If editing an md file
     if st.session_state.editing_file and os.path.exists(st.session_state.editing_file):
         st.sidebar.subheader(f"Editing: {os.path.basename(st.session_state.editing_file)}")
-        st.session_state.edit_new_name = st.sidebar.text_input("New name (without extension):", value=st.session_state.edit_new_name)
+        st.session_state.edit_new_name = st.sidebar.text_input("New name (no extension):", value=st.session_state.edit_new_name)
         st.session_state.edit_new_content = st.sidebar.text_area("Content:", st.session_state.edit_new_content, height=200)
         c1,c2 = st.sidebar.columns(2)
         with c1:
-            if st.button("Save Changes"):
+            if st.button("Save"):
                 old_path = st.session_state.editing_file
                 new_path = st.session_state.edit_new_name + ".md"
                 if new_path != os.path.basename(old_path):
@@ -375,12 +380,10 @@ def display_file_manager_sidebar(files_by_ext):
                 with open(new_path,'w',encoding='utf-8') as f:
                     f.write(st.session_state.edit_new_content)
                 st.session_state.editing_file = None
-                # user triggered, set rerun
                 st.session_state.should_rerun = True
         with c2:
             if st.button("Cancel"):
                 st.session_state.editing_file = None
-                # user triggered, set rerun
                 st.session_state.should_rerun = True
 
 def main():
@@ -506,7 +509,6 @@ def main():
                 with open(st.session_state.current_file,'w',encoding='utf-8') as f:
                     f.write(new_text)
                 st.success("Updated!")
-                # user action: rename or save
                 st.session_state.should_rerun = True
         else:
             st.write("Select a file from the sidebar to edit.")
@@ -515,7 +517,23 @@ def main():
     files_by_ext = load_files_for_sidebar()
     display_file_manager_sidebar(files_by_ext)
 
-    # If user triggered changes require a rerun, do it now
+    # If viewing a file, show its content below (in the main area)
+    if st.session_state.viewing_file and os.path.exists(st.session_state.viewing_file):
+        st.write("---")
+        st.write(f"**Viewing File:** {os.path.basename(st.session_state.viewing_file)}")
+        if st.session_state.viewing_file_type == "md":
+            # show markdown
+            content = open(st.session_state.viewing_file,'r',encoding='utf-8').read()
+            st.markdown(content)
+        elif st.session_state.viewing_file_type == "mp3":
+            # show audio
+            st.audio(st.session_state.viewing_file)
+        # Optionally add a "Close View" button
+        if st.button("Close View"):
+            st.session_state.viewing_file = None
+            st.session_state.viewing_file_type = None
+
+    # If user-triggered changes happened, rerun once at the end
     if st.session_state.should_rerun:
         st.session_state.should_rerun = False
         st.rerun()