Update app.py

app.py

@@ -5,7 +5,7 @@ import streamlit.components.v1 as components
 from datetime import datetime
 from audio_recorder_streamlit import audio_recorder
 from bs4 import BeautifulSoup
-from collections import deque
+from collections import defaultdict, deque
 from dotenv import load_dotenv
 from gradio_client import Client
 from huggingface_hub import InferenceClient
@@ -51,6 +51,9 @@ st.markdown("""
 <style>
     .main { background: linear-gradient(to right, #1a1a1a, #2d2d2d); color: #fff; }
     .stMarkdown { font-family: 'Helvetica Neue', sans-serif; }
+    .stButton>button {
+        margin-right: 0.5rem;
+    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -65,6 +68,7 @@ def generate_filename(prompt, file_type="md"):
 def create_file(filename, prompt, response):
     with open(filename, 'w', encoding='utf-8') as f:
         f.write(prompt + "\n\n" + response)
+    st.experimental_rerun()
 
 def get_download_link(file):
     with open(file, "rb") as f:
@@ -93,6 +97,7 @@ async def edge_tts_generate_audio(text, voice="en-US-AriaNeural", rate=0, pitch=
     communicate = edge_tts.Communicate(text, voice, rate=rate_str, pitch=pitch_str)
     out_fn = generate_filename(text,"mp3")
     await communicate.save(out_fn)
+    st.experimental_rerun()
     return out_fn
 
 def speak_with_edge_tts(text, voice="en-US-AriaNeural", rate=0, pitch=0):
@@ -102,7 +107,6 @@ 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)
-#---------------------------
 
 def process_image(image_path, user_prompt):
     with open(image_path, "rb") as imgf:
@@ -125,6 +129,7 @@ def process_audio(audio_path):
     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})
+    st.experimental_rerun()
     return transcription.text
 
 def process_video(video_path, seconds_per_frame=1):
@@ -230,6 +235,7 @@ def process_with_gpt(text):
         st.write("GPT-4o: " + ans)
         create_file(generate_filename(text,"md"),text,ans)
         st.session_state.messages.append({"role":"assistant","content":ans})
+        st.experimental_rerun()
         return ans
 
 def process_with_claude(text):
@@ -246,16 +252,18 @@ def process_with_claude(text):
         st.write("Claude: " + ans)
         create_file(generate_filename(text,"md"),text,ans)
         st.session_state.chat_history.append({"user":text,"claude":ans})
+        st.experimental_rerun()
         return ans
 
 def create_zip_of_files():
-    md_files = glob.glob("*.md")
-    mp3_files = glob.glob("*.mp3")
+    md_files = glob.glob("Media/*.md")
+    mp3_files = glob.glob("Media/*.mp3")
     all_files = md_files + mp3_files
     zip_name = "all_files.zip"
     with zipfile.ZipFile(zip_name,'w') as z:
         for f in all_files:
             z.write(f)
+    st.experimental_rerun()
     return zip_name
 
 def get_media_html(p,typ="video",w="100%"):
@@ -265,81 +273,109 @@ def get_media_html(p,typ="video",w="100%"):
     else:
         return f'<audio controls style="width:{w};"><source src="data:audio/mpeg;base64,{d}" type="audio/mpeg"></audio>'
 
-# CHANGES START HERE:
-# Define file emojis and the directory to scan
-FILE_EMOJIS = {
-    "cards": "🗃️",
-    "csv": "📊",
-    "heic": "🖼️",
-    "ico": "🪧",
-    "jpeg": "🖼️",
-    "json": "🔧",
-    "md": "📝",
-    "mid": "🎼",
-    "mov": "🎥",
-    "mp3": "🎵",
-    "mp4": "🎞️",
-    "png": "🖼️",
-    "svg": "🖌️",
-    "txt": "📄",
-    "wav": "🎶",
-    "webm": "📽️",
-    "webp": "🏞️",
-    "zip": "📦",
-}
 MEDIA_DIR = "Media"
 
-def get_file_ext(filename):
-    return os.path.splitext(filename)[1].lower().strip('.')
-
-def load_files():
-    all_files = []
-    for root, dirs, files in os.walk(MEDIA_DIR):
-        for f in files:
-            fp = os.path.join(root, f)
-            if os.path.isfile(fp):
-                ext = get_file_ext(fp)
-                mod_time = os.path.getmtime(fp)
-                all_files.append((fp, ext, mod_time))
-    return all_files
+def load_md_mp3_pairs():
+    # This function groups .md and .mp3 files by their filename stem
+    files = glob.glob(os.path.join(MEDIA_DIR,"*.md")) + glob.glob(os.path.join(MEDIA_DIR,"*.mp3"))
+    grouped = defaultdict(dict)
+    for f in files:
+        base = os.path.basename(f)
+        stem, ext = os.path.splitext(base)
+        ext = ext.lower()
+        if ext == '.md':
+            grouped[stem]['md'] = f
+        elif ext == '.mp3':
+            grouped[stem]['mp3'] = f
+    return grouped
 
 def display_files_sidebar():
-    st.sidebar.title("📂 Media Files")
-
-    all_files = load_files()
-    from collections import defaultdict
-    ext_map = defaultdict(list)
-    for fp, ext, mod_time in all_files:
-        ext_map[ext].append((fp, mod_time))
-
-    # Sort files in each extension group by modification time descending
-    for ext in ext_map:
-        ext_map[ext].sort(key=lambda x: x[1], reverse=True)
-
-    # Sort extensions by number of files descending
-    sorted_ext = sorted(ext_map.keys(), key=lambda x: len(ext_map[x]), reverse=True)
-
-    for ext in sorted_ext:
-        emoji = FILE_EMOJIS.get(ext, "📁")
-        count = len(ext_map[ext])
-        with st.sidebar.expander(f"{emoji} {ext.upper()} ({count})"):
-            for fp, mod_time in ext_map[ext]:
-                basename = os.path.basename(fp)
-                last_mod = datetime.fromtimestamp(mod_time).strftime("%Y-%m-%d %H:%M:%S")
-                col1, col2 = st.columns([3,1])
-                with col1:
-                    st.write(f"**{basename}** - Modified: {last_mod}")
-                with col2:
-                    if ext == "mp3":
-                        # For MP3, load download link only after user clicks an expander
-                        mp3_exp = st.expander("Load MP3 Download Link")
-                        with mp3_exp:
-                            st.markdown(get_download_link(fp), unsafe_allow_html=True)
-                            # If desired, add an on-demand audio player here.
-                    else:
-                        # Direct download link for other files
-                        st.markdown(get_download_link(fp), unsafe_allow_html=True)
-# CHANGES END HERE
+    st.sidebar.title("📂 Files")
+    pairs = load_md_mp3_pairs()
+
+    # Sort by modification time of the MD if exists, else mp3. Descending by latest mod time.
+    def mod_time(pair):
+        # Return the newest mod time of available files in the pair
+        times = []
+        for f in pair.values():
+            times.append(os.path.getmtime(f))
+        return max(times)
+    sorted_pairs = sorted(pairs.items(), key=lambda x: mod_time(x[1]), reverse=True)
+
+    for stem, files_dict in sorted_pairs:
+        with st.sidebar.expander(f"**{stem}**"):
+            # Display action buttons per file type
+            # If MD file exists:
+            if 'md' in files_dict:
+                md_file = files_dict['md']
+                c1, c2, c3, c4 = st.columns([2,1,1,1])
+                with c1:
+                    st.write("**Markdown File**")
+                with c2:
+                    if st.button("👀 View", key="view_md_"+stem):
+                        content = open(md_file,'r',encoding='utf-8').read()
+                        st.markdown("**MD File Content:**")
+                        st.markdown(content)
+                with c3:
+                    # Edit name/content
+                    if st.button("✏️ Edit", key="edit_md_"+stem):
+                        st.session_state.editing_md = stem
+                        st.experimental_rerun()
+                with c4:
+                    if st.button("🗑 Delete", key="del_md_"+stem):
+                        os.remove(md_file)
+                        st.experimental_rerun()
+            else:
+                st.write("No .md file for this stem.")
+
+            # If MP3 file exists:
+            if 'mp3' in files_dict:
+                mp3_file = files_dict['mp3']
+                c1, c2, c3 = st.columns([2,1,1])
+                with c1:
+                    st.write("**Audio File**")
+                with c2:
+                    if st.button("👀 View", key="view_mp3_"+stem):
+                        st.audio(mp3_file)
+                with c3:
+                    if st.button("🗑 Delete", key="del_mp3_"+stem):
+                        os.remove(mp3_file)
+                        st.experimental_rerun()
+            else:
+                st.write("No .mp3 file for this stem.")
+
+    # Button to create a zip of all files
+    if len(pairs) > 0:
+        if 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 'editing_md' in st.session_state:
+        stem = st.session_state.editing_md
+        pairs = load_md_mp3_pairs()
+        files_dict = pairs.get(stem, {})
+        if 'md' in files_dict:
+            md_file = files_dict['md']
+            content = open(md_file,'r',encoding='utf-8').read()
+            st.sidebar.subheader(f"Editing: {stem}.md")
+            new_stem = st.sidebar.text_input("New stem (filename without extension):", value=stem)
+            new_content = st.sidebar.text_area("Content:", content, height=200)
+            if st.sidebar.button("Save Changes"):
+                # If name changed, rename the file
+                if new_stem != stem:
+                    new_path = os.path.join(MEDIA_DIR, new_stem+".md")
+                    os.rename(md_file, new_path)
+                    md_file = new_path
+                # Update content
+                with open(md_file,'w',encoding='utf-8') as f:
+                    f.write(new_content)
+                del st.session_state.editing_md
+                st.experimental_rerun()
+            if st.sidebar.button("Cancel"):
+                del st.session_state.editing_md
+                st.experimental_rerun()
+
 
 def main():
     st.sidebar.markdown("### 🚲BikeAI🏆 Multi-Agent Research AI")
@@ -433,42 +469,35 @@ def main():
         st.header("🎬 Media Gallery - Images and Videos")
         tabs = st.tabs(["🖼️ Images", "🎥 Video"])
         with tabs[0]:
-            imgs = glob.glob("*.png")+glob.glob("*.jpg")
+            imgs = glob.glob("Media/*.png")+glob.glob("Media/*.jpg")
             if imgs:
                 c = st.slider("Cols",1,5,3)
                 cols = st.columns(c)
                 for i,f in enumerate(imgs):
                     with cols[i%c]:
                         st.image(Image.open(f),use_container_width=True)
-                        if st.button(f"👀 Analyze {os.path.basename(f)}"):
+                        if st.button(f"👀 Analyze {os.path.basename(f)}", key=f"analyze_{f}"):
                             a = process_image(f,"Describe this image.")
                             st.markdown(a)
             else:
                 st.write("No images found.")
         with tabs[1]:
-            vids = glob.glob("*.mp4")
+            vids = glob.glob("Media/*.mp4")
             if vids:
                 for v in vids:
                     with st.expander(f"🎥 {os.path.basename(v)}"):
                         st.markdown(get_media_html(v,"video"),unsafe_allow_html=True)
-                        if st.button(f"Analyze {os.path.basename(v)}"):
+                        if st.button(f"Analyze {os.path.basename(v)}", key=f"analyze_{v}"):
                             a = process_video_with_gpt(v,"Describe video.")
                             st.markdown(a)
             else:
                 st.write("No videos found.")
 
     elif tab_main == "📝 File Editor":
-        if getattr(st.session_state,'current_file',None):
-            st.subheader(f"Editing: {st.session_state.current_file}")
-            new_text = st.text_area("Content:", st.session_state.file_content, height=300)
-            if st.button("Save"):
-                with open(st.session_state.current_file,'w',encoding='utf-8') as f:
-                    f.write(new_text)
-                st.success("Updated!")
-        else:
-            st.write("Select a file from the sidebar to edit.")
-
-    # Use the new sidebar display function
+        st.write("Use the sidebar to edit .md files by clicking the ✏️ button on the desired file.")
+
+
+    # Display file list last to ensure updates
     display_files_sidebar()
 
 if __name__=="__main__":