Update app.py

app.py

@@ -41,15 +41,24 @@ 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', "")
-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
+if 'transcript_history' not in st.session_state:
+    st.session_state['transcript_history'] = []
+if 'chat_history' not in st.session_state:
+    st.session_state['chat_history'] = []
+if 'openai_model' not in st.session_state:
+    st.session_state['openai_model'] = "gpt-4o-2024-05-13"
+if 'messages' not in st.session_state:
+    st.session_state['messages'] = []
+if 'last_voice_input' not in st.session_state:
+    st.session_state['last_voice_input'] = ""
+if 'editing_file' not in st.session_state:
+    st.session_state['editing_file'] = None
+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 'should_rerun' not in st.session_state:
+    st.session_state['should_rerun'] = False
 
 # 🎨 Minimal Custom CSS
 st.markdown("""
@@ -75,9 +84,9 @@ 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.
     with open(filename, 'w', encoding='utf-8') as f:
         f.write(prompt + "\n\n" + response)
-    st.session_state.should_rerun = True
 
 def get_download_link(file):
     with open(file, "rb") as f:
@@ -97,6 +106,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.
     if not text.strip():
         return None
     rate_str = f"{rate:+d}%"
@@ -104,7 +114,6 @@ 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.session_state.should_rerun = True
     return out_fn
 
 def speak_with_edge_tts(text, voice="en-US-AriaNeural", rate=0, pitch=0):
@@ -133,10 +142,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})
-    st.session_state.should_rerun = True
     return transcription.text
 
 def process_video(video_path, seconds_per_frame=1):
@@ -191,6 +200,7 @@ 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)
 
@@ -259,6 +269,7 @@ def create_zip_of_files():
     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
 
@@ -291,7 +302,6 @@ def display_file_manager_sidebar(files_by_ext):
     md_files = files_by_ext.get('md', [])
     mp3_files = files_by_ext.get('mp3', [])
 
-    # Delete all buttons
     col_del = st.sidebar.columns(2)
     with col_del[0]:
         if st.button("🗑 Delete All MD"):
@@ -304,11 +314,9 @@ def display_file_manager_sidebar(files_by_ext):
                 os.remove(f)
             st.session_state.should_rerun = True
 
-    # 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,6 +345,7 @@ def display_file_manager_sidebar(files_by_ext):
                             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
@@ -344,9 +353,9 @@ def display_file_manager_sidebar(files_by_ext):
                     # Delete button
                     if st.button("🗑", key="del_"+f):
                         os.remove(f)
+                        # user triggered, set 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)"):
         z = create_zip_of_files()
         st.sidebar.markdown(get_download_link(z),unsafe_allow_html=True)
@@ -366,13 +375,14 @@ 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():
     st.sidebar.markdown("### 🚲BikeAI🏆 Multi-Agent Research AI")
     tab_main = st.radio("Action:",["🎤 Voice Input","📸 Media Gallery","🔍 Search ArXiv","📝 File Editor"],horizontal=True)
@@ -422,8 +432,7 @@ def main():
         if q:
             q = q.strip()
             if q and st.button("Run ArXiv Query"):
-                r = perform_ai_lookup(q, vocal_summary=vocal_summary, extended_refs=extended_refs, titles_summary=titles_summary)
-                st.markdown(r)
+                perform_ai_lookup(q, vocal_summary=vocal_summary, extended_refs=extended_refs, titles_summary=titles_summary)
 
     elif tab_main == "🎤 Voice Input":
         st.subheader("🎤 Voice Recognition")
@@ -490,7 +499,6 @@ def main():
                 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)
@@ -498,15 +506,16 @@ 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.")
 
-    # After all main content is processed, load files and display in sidebar
+    # After main content, 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 user triggered changes require a rerun, do it now
     if st.session_state.should_rerun:
         st.session_state.should_rerun = False
         st.rerun()