Update app.py

app.py

@@ -40,15 +40,16 @@ openai_client = OpenAI(api_key=openai.api_key, organization=os.getenv('OPENAI_OR
 HF_KEY = os.getenv('HF_KEY')
 API_URL = os.getenv('API_URL')
 
+# Session states
 st.session_state.setdefault('transcript_history', [])
 st.session_state.setdefault('chat_history', [])
 st.session_state.setdefault('openai_model', "gpt-4o-2024-05-13")
 st.session_state.setdefault('messages', [])
 st.session_state.setdefault('last_voice_input', "")
-# For editing .md files
 st.session_state.setdefault('editing_file', None)
 st.session_state.setdefault('edit_new_name', "")
 st.session_state.setdefault('edit_new_content', "")
+st.session_state.setdefault('should_rerun', False)  # Flag to indicate we need to rerun after operations
 
 # 🎨 Minimal Custom CSS
 st.markdown("""
@@ -76,7 +77,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.rerun()
+    st.session_state.should_rerun = True
 
 def get_download_link(file):
     with open(file, "rb") as f:
@@ -103,7 +104,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.rerun()
+    st.session_state.should_rerun = True
     return out_fn
 
 def speak_with_edge_tts(text, voice="en-US-AriaNeural", rate=0, pitch=0):
@@ -135,7 +136,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.rerun()
+    st.session_state.should_rerun = True
     return transcription.text
 
 def process_video(video_path, seconds_per_frame=1):
@@ -188,20 +189,17 @@ def perform_ai_lookup(q, vocal_summary=True, extended_refs=False, titles_summary
 
     st.markdown(result)
 
-    # Vocal summary
     if vocal_summary:
         audio_file_main = speak_with_edge_tts(r2, voice="en-US-AriaNeural", rate=0, pitch=0)
         st.write("### 🎙️ Vocal Summary (Short Answer)")
         play_and_download_audio(audio_file_main)
 
-    # Extended refs
     if extended_refs:
         summaries_text = "Here are the summaries from the references: " + refs.replace('"','')
         audio_file_refs = speak_with_edge_tts(summaries_text, voice="en-US-AriaNeural", rate=0, pitch=0)
         st.write("### 📜 Extended References & Summaries")
         play_and_download_audio(audio_file_refs)
 
-    # Titles only
     if titles_summary:
         titles = []
         for line in refs.split('\n'):
@@ -261,7 +259,7 @@ def create_zip_of_files():
     with zipfile.ZipFile(zip_name,'w') as z:
         for f in all_files:
             z.write(f)
-    st.rerun()
+    st.session_state.should_rerun = True
     return zip_name
 
 def get_media_html(p,typ="video",w="100%"):
@@ -271,44 +269,46 @@ 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>'
 
-def display_file_manager():
-    st.sidebar.title("🎵 Audio & Document Manager")
-    st.sidebar.markdown("Lists .mp3 and .md files with emojis and sorted by file type count and mod time.")
-
+def load_files_for_sidebar():
     # Gather all md and mp3 files
     md_files = glob.glob("*.md")
     mp3_files = glob.glob("*.mp3")
 
-    # Group by extension
     files_by_ext = defaultdict(list)
     for f in md_files:
-        ext = "md"
-        files_by_ext[ext].append(f)
+        files_by_ext['md'].append(f)
     for f in mp3_files:
-        ext = "mp3"
-        files_by_ext[ext].append(f)
+        files_by_ext['mp3'].append(f)
 
     # Sort each extension group by modification time descending
     for ext in files_by_ext:
         files_by_ext[ext].sort(key=lambda x: os.path.getmtime(x), reverse=True)
+    return files_by_ext
 
-    # Sort extensions by number of files descending
-    sorted_ext = sorted(files_by_ext.keys(), key=lambda x: len(files_by_ext[x]), reverse=True)
+def display_file_manager_sidebar(files_by_ext):
+    st.sidebar.title("🎵 Audio & Document Manager")
+
+    md_files = files_by_ext.get('md', [])
+    mp3_files = files_by_ext.get('mp3', [])
 
     # Delete all buttons
-    del_col = st.sidebar.columns(2)
-    with del_col[0]:
+    col_del = st.sidebar.columns(2)
+    with col_del[0]:
         if st.button("🗑 Delete All MD"):
             for f in md_files:
                 os.remove(f)
-            st.rerun()
-    with del_col[1]:
+            st.session_state.should_rerun = True
+    with col_del[1]:
         if st.button("🗑 Delete All MP3"):
             for f in mp3_files:
                 os.remove(f)
-            st.rerun()
+            st.session_state.should_rerun = True
 
-    # Show groups
+    # Sort extensions by number of files descending
+    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 groups
     for ext in sorted_ext:
         emoji = FILE_EMOJIS.get(ext, "📦")
         count = len(files_by_ext[ext])
@@ -337,15 +337,14 @@ def display_file_manager():
                             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()
-                            st.rerun()
+                            st.session_state.should_rerun = True
                     else:
-                        # No edit for mp3
                         pass
                 with col4:
                     # Delete button
                     if st.button("🗑", key="del_"+f):
                         os.remove(f)
-                        st.rerun()
+                        st.session_state.should_rerun = True
 
     # Download all as zip
     if (len(md_files) > 0 or len(mp3_files) > 0) and st.sidebar.button("⬇️ Download All (.md and .mp3)"):
@@ -362,26 +361,25 @@ def display_file_manager():
             if st.button("Save Changes"):
                 old_path = st.session_state.editing_file
                 new_path = st.session_state.edit_new_name + ".md"
-                # Rename file if name changed
                 if new_path != os.path.basename(old_path):
                     os.rename(old_path, new_path)
-                # Update content
                 with open(new_path,'w',encoding='utf-8') as f:
                     f.write(st.session_state.edit_new_content)
                 st.session_state.editing_file = None
-                st.experimental_rerun()
+                st.session_state.should_rerun = True
         with c2:
             if st.button("Cancel"):
                 st.session_state.editing_file = None
-                st.rerun()
+                st.session_state.should_rerun = True
+
 
 def main():
     st.sidebar.markdown("### 🚲BikeAI🏆 Multi-Agent Research AI")
     tab_main = st.radio("Action:",["🎤 Voice Input","📸 Media Gallery","🔍 Search ArXiv","📝 File Editor"],horizontal=True)
-    
+
     model_choice = st.sidebar.radio("AI Model:", ["Arxiv","GPT-4o","Claude-3","GPT+Claude+Arxiv"], index=0)
 
-    # Declare the component
+    # Main Input Component
     mycomponent = components.declare_component("mycomponent", path="mycomponent")
     val = mycomponent(my_input_value="Hello")
     if val:
@@ -474,7 +472,7 @@ def main():
                 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:
@@ -485,13 +483,14 @@ def main():
                 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":
+        # Existing code for inline editing if needed
         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)
@@ -499,10 +498,18 @@ def main():
                 with open(st.session_state.current_file,'w',encoding='utf-8') as f:
                     f.write(new_text)
                 st.success("Updated!")
+                st.session_state.should_rerun = True
         else:
             st.write("Select a file from the sidebar to edit.")
 
-    display_file_manager()
+    # After all main content is processed, load files and display in sidebar
+    files_by_ext = load_files_for_sidebar()
+    display_file_manager_sidebar(files_by_ext)
+
+    # If we performed an operation, rerun now at the end
+    if st.session_state.should_rerun:
+        st.session_state.should_rerun = False
+        st.rerun()
 
 if __name__=="__main__":
     main()