Spaces:

awacke1
/

RescuerOfStolenBikes

Sleeping

App Files Files Community

awacke1 commited on Dec 16, 2024

Commit

18489bd

verified ·

1 Parent(s): f8d1729

Update app.py

Browse files

Files changed (1) hide show

app.py +55 -41

app.py CHANGED Viewed

@@ -73,7 +73,7 @@ def get_download_link(file):
 @st.cache_resource
 def speech_synthesis_html(result):
-    # This old function can remain as a fallback, but we won't use it after integrating EdgeTTS.
     html_code = f"""
     <html><body>
     <script>
@@ -177,7 +177,7 @@ def search_arxiv(query):
     st.markdown(r2)
     return f"{r1}\n\n{r2}"
-def perform_ai_lookup(q):
     start = time.time()
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     # Perform a RAG-based search
@@ -189,31 +189,40 @@ def perform_ai_lookup(q):
     st.markdown(result)
-    # Speak main result
-    audio_file_main = speak_with_edge_tts(r2, voice="en-US-AriaNeural", rate=0, pitch=0)
-    st.write("### Audio Output for Main Result")
-    play_and_download_audio(audio_file_main)
-    # Speak references summaries
-    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("### Audio Output for References Summaries")
-    play_and_download_audio(audio_file_refs)
-    # Extract titles from refs and speak them
-    titles = []
-    for line in refs.split('\n'):
-        m = re.search(r"\[([^\]]+)\]", line)
-        if m:
-            titles.append(m.group(1))
-    if titles:
-        titles_text = "Here are the titles of the papers: " + ", ".join(titles)
-        audio_file_titles = speak_with_edge_tts(titles_text, voice="en-US-AriaNeural", rate=0, pitch=0)
-        st.write("### Audio Output for Paper Titles")
-        play_and_download_audio(audio_file_titles)
     elapsed = time.time()-start
-    st.write(f"Elapsed: {elapsed:.2f} s")
     fn = generate_filename(q,"md")
     create_file(fn,q,result)
     return result
@@ -339,7 +348,7 @@ def main():
                 process_with_claude(user_input)
             elif model_choice == "Arxiv":
                 st.subheader("Arxiv Only Results:")
-                perform_ai_lookup(user_input)
             else:
                 col1,col2,col3=st.columns(3)
                 with col1:
@@ -353,12 +362,27 @@ def main():
                 with col3:
                     st.subheader("Arxiv + Mistral:")
                     try:
-                        r = perform_ai_lookup(user_input)
-                        st.markdown(r)
                     except:
                         st.write("Arxiv error")
-    if tab_main == "🎤 Voice Input":
         st.subheader("🎤 Voice Recognition")
         user_text = st.text_area("Message:", height=100)
         user_text = user_text.strip()
@@ -370,7 +394,7 @@ def main():
                     process_with_claude(user_text)
                 elif model_choice == "Arxiv":
                     st.subheader("Arxiv Only Results:")
-                    perform_ai_lookup(user_text)
                 else:
                     col1,col2,col3=st.columns(3)
                     with col1:
@@ -381,7 +405,7 @@ def main():
                         process_with_claude(user_text)
                     with col3:
                         st.subheader("Arxiv & Mistral:")
-                        res = perform_ai_lookup(user_text)
                         st.markdown(res)
         st.subheader("📜 Chat History")
         t1,t2=st.tabs(["Claude History","GPT-4o History"])
@@ -395,7 +419,6 @@ def main():
                     st.markdown(m["content"])
     elif tab_main == "📸 Media Gallery":
-        # Only show Images and Videos since Audio is now in sidebar
         st.header("🎬 Media Gallery - Images and Videos")
         tabs = st.tabs(["🖼️ Images", "🎥 Video"])
         with tabs[0]:
@@ -423,14 +446,6 @@ def main():
             else:
                 st.write("No videos found.")
-    elif tab_main == "🔍 Search ArXiv":
-        q=st.text_input("Research query:")
-        if q:
-            q = q.strip()
-            if q:
-                r=search_arxiv(q)
-                st.markdown(r)
     elif tab_main == "📝 File Editor":
         if getattr(st.session_state,'current_file',None):
             st.subheader(f"Editing: {st.session_state.current_file}")
@@ -444,6 +459,5 @@ def main():
     display_file_manager()
 if __name__=="__main__":
     main()

 @st.cache_resource
 def speech_synthesis_html(result):
+    # This old function remains for fallback, unused after integrating EdgeTTS.
     html_code = f"""
     <html><body>
     <script>
     st.markdown(r2)
     return f"{r1}\n\n{r2}"
+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
     st.markdown(result)
+    # Main Vocal Summary (Short Answer)
+    if vocal_summary:
+        start_main_part = time.time()
+        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)
+        st.write(f"**Elapsed (Short Answer):** {time.time() - start_main_part:.2f} s")
+    # Extended References & Summaries (optional)
+    if extended_refs:
+        start_refs_part = time.time()
+        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)
+        st.write(f"**Elapsed (Extended References):** {time.time() - start_refs_part:.2f} s")
+    # Paper Titles Only (short)
+    if titles_summary:
+        start_titles_part = time.time()
+        titles = []
+        for line in refs.split('\n'):
+            m = re.search(r"\[([^\]]+)\]", line)
+            if m:
+                titles.append(m.group(1))
+        if titles:
+            titles_text = "Here are the titles of the papers: " + ", ".join(titles)
+            audio_file_titles = speak_with_edge_tts(titles_text, voice="en-US-AriaNeural", rate=0, pitch=0)
+            st.write("### 🔖 Paper Titles")
+            play_and_download_audio(audio_file_titles)
+            st.write(f"**Elapsed (Titles):** {time.time() - start_titles_part:.2f} s")
     elapsed = time.time()-start
+    st.write(f"**Total Elapsed:** {elapsed:.2f} s")
     fn = generate_filename(q,"md")
     create_file(fn,q,result)
     return result
                 process_with_claude(user_input)
             elif model_choice == "Arxiv":
                 st.subheader("Arxiv Only Results:")
+                perform_ai_lookup(user_input, vocal_summary=True, extended_refs=False, titles_summary=True)
             else:
                 col1,col2,col3=st.columns(3)
                 with col1:
                 with col3:
                     st.subheader("Arxiv + Mistral:")
                     try:
+                        perform_ai_lookup(user_input, vocal_summary=True, extended_refs=False, titles_summary=True)
                     except:
                         st.write("Arxiv error")
+    if tab_main == "🔍 Search ArXiv":
+        st.subheader("🔍 Search ArXiv")
+        q=st.text_input("Research query:")
+        # 🎛️ Audio Generation Options
+        st.markdown("### 🎛️ Audio Generation Options")
+        vocal_summary = st.checkbox("🎙️ Vocal Summary (Short Answer)", value=True)
+        extended_refs = st.checkbox("📜 Extended References & Summaries (Long)", value=False)
+        titles_summary = st.checkbox("🔖 Paper Titles Only", value=True)
+        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)
+    elif tab_main == "🎤 Voice Input":
         st.subheader("🎤 Voice Recognition")
         user_text = st.text_area("Message:", height=100)
         user_text = user_text.strip()
                     process_with_claude(user_text)
                 elif model_choice == "Arxiv":
                     st.subheader("Arxiv Only Results:")
+                    perform_ai_lookup(user_text, vocal_summary=True, extended_refs=False, titles_summary=True)
                 else:
                     col1,col2,col3=st.columns(3)
                     with col1:
                         process_with_claude(user_text)
                     with col3:
                         st.subheader("Arxiv & Mistral:")
+                        res = perform_ai_lookup(user_text, vocal_summary=True, extended_refs=False, titles_summary=True)
                         st.markdown(res)
         st.subheader("📜 Chat History")
         t1,t2=st.tabs(["Claude History","GPT-4o History"])
                     st.markdown(m["content"])
     elif tab_main == "📸 Media Gallery":
         st.header("🎬 Media Gallery - Images and Videos")
         tabs = st.tabs(["🖼️ Images", "🎥 Video"])
         with tabs[0]:
             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}")
     display_file_manager()
 if __name__=="__main__":
     main()