Update app.py

app.py

@@ -131,7 +131,6 @@ def process_text(text_input):
             filename = generate_filename(text_input, "md")
             create_and_save_file(response, "md", text_input, should_save=should_save)
             st.session_state.messages.append({"role": "assistant", "content": response})
-            st.rerun()
 
 # Image Processing
 def process_image(image_input, user_prompt):
@@ -154,7 +153,6 @@ def process_image(image_input, user_prompt):
     image_response = response.choices[0].message.content
     filename = generate_filename(user_prompt, "md", original_name)
     create_and_save_file(image_response, "md", user_prompt, original_name, should_save=should_save)
-    st.rerun()
     return image_response
 
 # Audio Processing
@@ -188,7 +186,6 @@ def process_audio(audio_input, text_input=''):
                     filename = generate_filename(transcription, "md")
                     create_and_save_file(response, "md", text_input, should_save=should_save)
                     st.session_state.messages.append({"role": "assistant", "content": response})
-                    st.rerun()
             except openai.BadRequestError as e:
                 st.error(f"Audio processing error: {str(e)}")
 
@@ -257,7 +254,6 @@ def process_audio_and_video(video_input):
                         st.markdown(result)
                         filename = generate_filename(transcript, "md")
                         create_and_save_file(result, "md", "Video summary", should_save=should_save)
-                        st.rerun()
                 except openai.BadRequestError as e:
                     st.error(f"Video audio processing error: {str(e)}")
         else:
@@ -277,7 +273,6 @@ def search_arxiv(query):
     filename = generate_filename(query, "md")
     create_and_save_file(result, "md", query, should_save=should_save)
     st.session_state.messages.append({"role": "assistant", "content": result})
-    st.rerun()
     return result
 
 # RAG PDF Gallery
@@ -471,8 +466,16 @@ def main():
     
     if option == "Text":
         default_text = "emojis in markdown. Maybe a buckeyball feature rating comparing them against each other in markdown emoji outline or tables."
-        text_input = st.text_input("Enter your text:", value=default_text)
-        if text_input:
+        col1, col2 = st.columns([1, 5])
+        with col1:
+            if st.button("📝 MD", key="md_button"):
+                st.session_state["text_input"] = default_text
+                with st.spinner("Processing..."):
+                    process_text(default_text)
+                st.rerun()
+        with col2:
+            text_input = st.text_input("Enter your text:", value=st.session_state.get("text_input", ""), key="text_input_field")
+        if text_input and text_input != st.session_state.get("text_input", ""):  # Only process if changed
             with st.spinner("Processing..."):
                 process_text(text_input)
     
@@ -494,6 +497,7 @@ def main():
                     image_response = process_image(image_input, text_input)
                     with st.chat_message("ai", avatar="🦖"):
                         st.markdown(image_response)
+                st.rerun()
     
     elif option == "Audio":
         text_input = st.text_input("Audio Prompt:", value="Summarize this audio transcription in Markdown.")
@@ -502,10 +506,13 @@ def main():
         if audio_bytes:
             with open("recorded_audio.wav", "wb") as f:
                 f.write(audio_bytes)
-            process_audio(audio_bytes, text_input)
+            with st.spinner("Processing..."):
+                process_audio(audio_bytes, text_input)
+            st.rerun()
         elif audio_input and text_input:
             with st.spinner("Processing..."):
                 process_audio(audio_input, text_input)
+            st.rerun()
     
     elif option == "Video":
         text_input = st.text_input("Video Prompt:", value="Summarize this video and its transcription in Markdown.")
@@ -516,6 +523,7 @@ def main():
             else:
                 with st.spinner("Processing..."):
                     process_audio_and_video(video_input)
+                st.rerun()
     
     elif option == "ArXiv Search":
         query = st.text_input("AI Search ArXiv Scholarly Articles:")
@@ -523,6 +531,7 @@ def main():
             with st.spinner("Searching ArXiv..."):
                 result = search_arxiv(query)
                 st.markdown(result)
+            st.rerun()
     
     elif option == "RAG PDF Gallery":
         rag_pdf_gallery()
@@ -533,7 +542,9 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 if prompt := st.chat_input("GPT-4o Multimodal ChatBot - What can I help you with?"):
-    process_text(prompt)
+    with st.spinner("Processing..."):
+        process_text(prompt)
+    st.rerun()
 
 FileSidebar()
 main()