Spaces:

awacke1
/

RescuerOfStolenBikes

Sleeping

App Files Files Community

awacke1 commited on Dec 16, 2024

Commit

6cf3eb8

verified ·

1 Parent(s): 18489bd

Update app.py

Browse files

Files changed (1) hide show

app.py +54 -45

app.py CHANGED Viewed

@@ -73,7 +73,6 @@ def get_download_link(file):
 @st.cache_resource
 def speech_synthesis_html(result):
-    # This old function remains for fallback, unused after integrating EdgeTTS.
     html_code = f"""
     <html><body>
     <script>
@@ -87,9 +86,6 @@ def speech_synthesis_html(result):
 #------------add EdgeTTS
 # --- NEW FUNCTIONS FOR EDGE TTS ---
 async def edge_tts_generate_audio(text, voice="en-US-AriaNeural", rate=0, pitch=0):
-    """
-    Generate audio from text using Edge TTS and return the path to the MP3 file.
-    """
     if not text.strip():
         return None
     rate_str = f"{rate:+d}%"
@@ -100,15 +96,9 @@ async def edge_tts_generate_audio(text, voice="en-US-AriaNeural", rate=0, pitch=
     return out_fn
 def speak_with_edge_tts(text, voice="en-US-AriaNeural", rate=0, pitch=0):
-    """
-    Synchronous wrapper to call the async TTS generation and return the file path.
-    """
     return asyncio.run(edge_tts_generate_audio(text, voice, rate, pitch))
 def play_and_download_audio(file_path):
-    """
-    Display an audio player and a download link for the generated MP3 file.
-    """
     if file_path and os.path.exists(file_path):
         st.audio(file_path)
         st.markdown(get_download_link(file_path), unsafe_allow_html=True)
@@ -180,10 +170,8 @@ def search_arxiv(query):
 def perform_ai_lookup(q, vocal_summary=True, extended_refs=False, titles_summary=True):
     start = time.time()
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
-    # Perform a RAG-based search
     r = client.predict(q,20,"Semantic Search","mistralai/Mixtral-8x7B-Instruct-v0.1",api_name="/update_with_rag_md")
     refs = r[0]
-    # Ask model for answer
     r2 = client.predict(q,"mistralai/Mixtral-8x7B-Instruct-v0.1",True,api_name="/ask_llm")
     result = f"### 🔎 {q}\n\n{r2}\n\n{refs}"
@@ -261,7 +249,6 @@ def process_with_claude(text):
         return ans
 def create_zip_of_files():
-    # Include all .md and .mp3 files in the zip
     md_files = glob.glob("*.md")
     mp3_files = glob.glob("*.mp3")
     all_files = md_files + mp3_files
@@ -286,17 +273,32 @@ def display_file_manager():
     mp3_files = sorted(glob.glob("*.mp3"), reverse=True)
     if mp3_files:
         st.sidebar.subheader("MP3 Files:")
         for a in mp3_files:
-            with st.sidebar.expander(f"{os.path.basename(a)}"):
                 # Show audio player
-                st.sidebar.markdown(get_media_html(a,"audio"),unsafe_allow_html=True)
-                # Download link for the MP3 file
-                st.sidebar.markdown(get_download_link(a), unsafe_allow_html=True)
-                # Button to transcribe this file
-                if st.sidebar.button(f"Transcribe {os.path.basename(a)}"):
-                    t = process_audio(a)
-                    st.sidebar.write("Transcription:")
-                    st.sidebar.write(t)
     else:
         st.sidebar.write("No MP3 files found.")
@@ -304,31 +306,38 @@ def display_file_manager():
     st.sidebar.subheader("MD Files:")
     files = sorted(glob.glob("*.md"), reverse=True)
     if st.sidebar.button("🗑 Delete All MD"):
-        for f in files: os.remove(f)
         st.experimental_rerun()
     # Download all as zip (including .mp3 and .md)
-    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)
-    for f in files:
-        col1, col2, col3, col4 = st.sidebar.columns([1,3,1,1])
-        with col1:
-            if st.sidebar.button("🌐", key="v"+f):
-                st.session_state.current_file = f
-                c = open(f,'r',encoding='utf-8').read()
-                st.write("**Viewing file content:**")
-                st.write(c)
-        with col2:
-            st.sidebar.markdown(get_download_link(f),unsafe_allow_html=True)
-        with col3:
-            if st.sidebar.button("📂", key="e"+f):
-                st.session_state.current_file = f
-                st.session_state.file_content = open(f,'r',encoding='utf-8').read()
-        with col4:
-            if st.sidebar.button("🗑", key="d"+f):
-                os.remove(f)
-                st.experimental_rerun()
 def main():
     st.sidebar.markdown("### 🚲BikeAI🏆 Multi-Agent Research AI")

 @st.cache_resource
 def speech_synthesis_html(result):
     html_code = f"""
     <html><body>
     <script>
 #------------add EdgeTTS
 # --- NEW FUNCTIONS FOR EDGE TTS ---
 async def edge_tts_generate_audio(text, voice="en-US-AriaNeural", rate=0, pitch=0):
     if not text.strip():
         return None
     rate_str = f"{rate:+d}%"
     return out_fn
 def speak_with_edge_tts(text, voice="en-US-AriaNeural", rate=0, pitch=0):
     return asyncio.run(edge_tts_generate_audio(text, voice, rate, pitch))
 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 perform_ai_lookup(q, vocal_summary=True, extended_refs=False, titles_summary=True):
     start = time.time()
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     r = client.predict(q,20,"Semantic Search","mistralai/Mixtral-8x7B-Instruct-v0.1",api_name="/update_with_rag_md")
     refs = r[0]
     r2 = client.predict(q,"mistralai/Mixtral-8x7B-Instruct-v0.1",True,api_name="/ask_llm")
     result = f"### 🔎 {q}\n\n{r2}\n\n{refs}"
         return ans
 def create_zip_of_files():
     md_files = glob.glob("*.md")
     mp3_files = glob.glob("*.mp3")
     all_files = md_files + mp3_files
     mp3_files = sorted(glob.glob("*.mp3"), reverse=True)
     if mp3_files:
         st.sidebar.subheader("MP3 Files:")
+        if st.sidebar.button("🗑 Delete All MP3"):
+            for a in mp3_files:
+                os.remove(a)
+            st.experimental_rerun()
         for a in mp3_files:
+            exp = st.sidebar.expander(os.path.basename(a))
+            with exp:
                 # Show audio player
+                st.markdown(get_media_html(a,"audio"),unsafe_allow_html=True)
+                # Actions row
+                c1, c2, c3 = st.columns([3,3,1])
+                with c1:
+                    # Download link for the MP3 file
+                    st.markdown(get_download_link(a), unsafe_allow_html=True)
+                with c2:
+                    # Button to transcribe this file
+                    if st.button(f"Transcribe {os.path.basename(a)}", key="transcribe_"+a):
+                        t = process_audio(a)
+                        st.write("Transcription:")
+                        st.write(t)
+                with c3:
+                    # Delete button for mp3
+                    if st.button("🗑", key="del_mp3_"+a):
+                        os.remove(a)
+                        st.experimental_rerun()
     else:
         st.sidebar.write("No MP3 files found.")
     st.sidebar.subheader("MD Files:")
     files = sorted(glob.glob("*.md"), reverse=True)
     if st.sidebar.button("🗑 Delete All MD"):
+        for f in files:
+            os.remove(f)
         st.experimental_rerun()
+    if files:
+        for f in files:
+            col1, col2, col3, col4 = st.sidebar.columns([1,3,1,1])
+            with col1:
+                if st.button("🌐", key="v"+f):
+                    st.session_state.current_file = f
+                    c = open(f,'r',encoding='utf-8').read()
+                    st.write("**Viewing file content:**")
+                    st.write(c)
+            with col2:
+                st.markdown(get_download_link(f),unsafe_allow_html=True)
+            with col3:
+                if st.button("📂", key="e"+f):
+                    st.session_state.current_file = f
+                    st.session_state.file_content = open(f,'r',encoding='utf-8').read()
+            with col4:
+                if st.button("🗑", key="d"+f):
+                    os.remove(f)
+                    st.experimental_rerun()
+    else:
+        st.sidebar.write("No MD files found.")
     # Download all as zip (including .mp3 and .md)
+    if len(files) > 0 or len(mp3_files) > 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)
 def main():
     st.sidebar.markdown("### 🚲BikeAI🏆 Multi-Agent Research AI")