Update app.py

app.py

@@ -145,14 +145,14 @@ def generate_html(local_files):
     html += "</ul>"
     return html
 
-
+#@st.cache_resource
 def search_arxiv(query):
     start_time = time.strftime("%Y-%m-%d %H:%M:%S")
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     search_query = query
     search_source = "Arxiv Search - Latest - (EXPERIMENTAL)"
     llm_model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
-    st.markdown('### 🔎 ' + query)
+    #st.markdown('### 🔎 ' + query)
 
     # Search 1 - Retrieve the Papers
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
@@ -165,18 +165,19 @@ def search_arxiv(query):
     )
     lastpart = ''
     totalparts = ''
-    results = response1[0]  # Format for markdown display with links
-    results2 = response1[1]  # format for subquery without links
-    st.markdown(results)
 
-    # Extract URLs from the response
-    urls = extract_urls(results)
+    Question = '### 🔎 ' + query + '\r\n'  # Format for markdown display with links
+    References =  response1[0]  
+    References2 = response1[1]  
 
-    # Download PDFs and generate HTML5 code for local links
-    # local_links_html = download_pdfs_and_generate_html(urls)
-    st.markdown(urls)
+    #st.markdown(results)
 
-    RunSecondQuery = False
+    # URLs from the response
+    ReferenceLinks = extract_urls(References)
+    #st.markdown(urls)
+    #results = results + urls
+
+    RunSecondQuery = True
     if RunSecondQuery:
         # Search 2 - Retrieve the Summary with Papers Context and Original Query
         response2 = client.predict(
@@ -185,11 +186,18 @@ def search_arxiv(query):
             True,
             api_name="/ask_llm"
         )
-        st.markdown(response2)
-        results = results + response2
+        #st.markdown(response2)
+        if len(response2) > 10:
+            Answer = response2
+            SpeechSynthesis(Answer)
+            # Restructure results to follow format of Question, Answer, References, ReferenceLinks
+            results = Question + Answer + References + ReferenceLinks
+            st.markdown(results)
 
     st.write('🔍Run of Multi-Agent System Paper Summary Spec is Complete')
     end_time = time.strftime("%Y-%m-%d %H:%M:%S")
+
+    # Output
     start_timestamp = time.mktime(time.strptime(start_time, "%Y-%m-%d %H:%M:%S"))
     end_timestamp = time.mktime(time.strptime(end_time, "%Y-%m-%d %H:%M:%S"))
     elapsed_seconds = end_timestamp - start_timestamp
@@ -197,7 +205,8 @@ def search_arxiv(query):
     st.write(f"Finish time: {end_time}")
     st.write(f"Elapsed time: {elapsed_seconds:.2f} seconds")
 
-    SpeechSynthesis(results)
+
+    #SpeechSynthesis(results)
     filename = generate_filename(query, "md")
     create_file(filename, query, results, should_save)
 
@@ -436,15 +445,15 @@ def display_glossary_entity(k):
 def display_glossary_grid(roleplaying_glossary):
     search_urls = {
         "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",  # this url plus query!
+        "🃏Analyst": lambda k: f"/?q={quote(k)}-{quote(PromptPrefix)}",  # this url plus query!
+        "📚PyCoder": lambda k: f"/?q={quote(k)}-{quote(PromptPrefix2)}",  # this url plus query!
+        "🔬JSCoder": lambda k: f"/?q={quote(k)}-{quote(PromptPrefix3)}",  # this url plus query!
         "📖Wiki": lambda k: f"https://en.wikipedia.org/wiki/{quote(k)}",
         "🔍Google": lambda k: f"https://www.google.com/search?q={quote(k)}",
         "▶️YouTube": lambda k: f"https://www.youtube.com/results?search_query={quote(k)}",
         "🔎Bing": lambda k: f"https://www.bing.com/search?q={quote(k)}",
         "🎥YouTube": lambda k: f"https://www.youtube.com/results?search_query={quote(k)}",
         "🐦Twitter": lambda k: f"https://twitter.com/search?q={quote(k)}",
-        "🃏Analyst": lambda k: f"https://huggingface.co/spaces/awacke1/{Site_Name}?q={quote(k)}-{quote(PromptPrefix)}",  # this url plus query!
-        "📚PyCoder": lambda k: f"https://huggingface.co/spaces/awacke1/{Site_Name}?q={quote(k)}-{quote(PromptPrefix2)}",  # this url plus query!
-        "🔬JSCoder": lambda k: f"https://huggingface.co/spaces/awacke1/{Site_Name}?q={quote(k)}-{quote(PromptPrefix3)}",  # this url plus query!
     }
 
     for category, details in roleplaying_glossary.items():