Update app.py

app.py

@@ -155,8 +155,89 @@ def archive_current_container(database_name, container_name, client):
     except Exception as e:
         return f"An error occurred while archiving data: {str(e)} 😢"
 
-# 🔍 Search Glossary function
+
+# Helper to extract hyperlinks
+def extract_hyperlinks(responses):
+    hyperlinks = []
+    for response in responses:
+        parsed_response = json.loads(response)
+        links = [value for key, value in parsed_response.items() if isinstance(value, str) and value.startswith("http")]
+        hyperlinks.extend(links)
+    return hyperlinks
+
+# Helper to format text with line numbers
+def format_with_line_numbers(text):
+    lines = text.splitlines()
+    formatted_text = '\n'.join(f"{i+1}: {line}" for i, line in enumerate(lines))
+    return formatted_text
+
+# Save responses to Cosmos DB
+def save_to_cosmos_db(query, response1, response2):
+    cosmos_container = st.session_state.get("cosmos_container")
+    if cosmos_container:
+        record = {
+            "query": query,
+            "response1": json.loads(response1),
+            "response2": json.loads(response2)
+        }
+        try:
+            cosmos_container.create_item(body=record)
+            st.success(f"Record saved successfully with ID: {record['id']}")
+        except exceptions.CosmosHttpResponseError as e:
+            st.error(f"Error saving record to Cosmos DB: {e}")
+    else:
+        st.error("Cosmos DB is not initialized.")
+
+
+# Add dropdowns for model and database choices
 def search_glossary(query):
+    st.markdown(f"### 🔍 Search Glossary for: `{query}`")
+    
+    # Dropdown for model selection
+    model_options = ['mistralai/Mixtral-8x7B-Instruct-v0.1', 'mistralai/Mistral-7B-Instruct-v0.2', 'google/gemma-7b-it', 'None']
+    model_choice = st.selectbox('🧠 Select LLM Model', options=model_options, index=1)
+    
+    # Dropdown for database selection
+    database_options = ['Semantic Search', 'Arxiv Search - Latest - (EXPERIMENTAL)']
+    database_choice = st.selectbox('📚 Select Database', options=database_options, index=0)
+    
+    # Run Button with Emoji
+    if st.button("🚀 Run"):
+        # ArXiv searcher - Paper Summary & Ask LLM
+        client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
+        
+        response1 = client.predict(
+            message=query,
+            llm_results_use=5,
+            database_choice=database_choice,
+            llm_model_picked=model_choice,
+            api_name="/update_with_rag_md"
+        )
+        
+        response2 = client.predict(
+            prompt=query,
+            llm_model_picked=model_choice,
+            stream_outputs=True,
+            api_name="/ask_llm"
+        )
+
+        # Aggregate hyperlinks and show with emojis
+        hyperlinks = extract_hyperlinks([response1, response2])
+        st.markdown("### 🔗 Aggregated Hyperlinks")
+        for link in hyperlinks:
+            st.markdown(f"🔗 [{link}]({link})")
+
+        # Show responses in a code format with line numbers
+        st.markdown("### 📜 Response Outputs with Line Numbers")
+        st.code(f"Response 1: \n{format_with_line_numbers(response1)}\n\nResponse 2: \n{format_with_line_numbers(response2)}", language="json")
+
+        # Save both responses to Cosmos DB
+        save_to_cosmos_db(query, response1, response2)
+
+    
+
+# 🔍 Search Glossary function
+def search_glossaryv1(query):
     # 🕵️‍♂️ Searching the glossary for: query
     all_results = ""
     st.markdown(f"- {query}")
@@ -176,7 +257,6 @@ def search_glossary(query):
 
     #llm_model_picked Literal['mistralai/Mixtral-8x7B-Instruct-v0.1', 'mistralai/Mistral-7B-Instruct-v0.2', 'google/gemma-7b-it', 'None'] Default: "mistralai/Mistral-7B-Instruct-v0.2"
 
-
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     result = client.predict(
 		prompt=query,
@@ -184,7 +264,6 @@ def search_glossary(query):
 		stream_outputs=True,
 		api_name="/ask_llm"
     )
-
     st.write('🔍 Run of Multi-Agent System Paper Summary Spec is Complete')
     st.markdown(response2)
     
@@ -197,9 +276,12 @@ def search_glossary(query):
         "mistralai/Mixtral-8x7B-Instruct-v0.1",	# LLM Model Dropdown component
         api_name="/update_with_rag_md"
     )
-    st.write('🔍 Run of Multi-Agent System Paper References is Complete')
-    responseall = response2 + response1[0] + response1[1]
-    st.markdown(responseall)
+
+
+    
+    #st.write('🔍 Run of Multi-Agent System Paper References is Complete')
+    #responseall = response2 + response1[0] + response1[1]
+    #st.markdown(responseall)
     return responseall
 
 # 📝 Function to process text input
@@ -465,6 +547,7 @@ def main():
                                         st.error(message)
                                 except json.JSONDecodeError as e:
                                     st.error(f"Invalid JSON: {str(e)} 🚫")
+                                    
                         elif selected_view == 'Show as Edit and Save':
                             # ✏️ Show as Edit and Save in columns
                             st.markdown("#### Edit the document fields below:")
@@ -494,6 +577,7 @@ def main():
                                                 st.error(message)
                                         except json.JSONDecodeError as e:
                                             st.error(f"Invalid JSON: {str(e)} 🚫")
+                                            
                         elif selected_view == 'Clone Document':
                             # 🧬 Clone Document per record
                             st.markdown("#### Clone a document:")
@@ -525,6 +609,7 @@ def main():
                                             st.error(message)
                                     except json.JSONDecodeError as e:
                                         st.error(f"Invalid JSON: {str(e)} 🚫")
+                                        
                         elif selected_view == 'New Record':
                             # 🆕 New Record
                             st.markdown("#### Create a new document:")
@@ -544,6 +629,7 @@ def main():
                                         st.error(message)
                                 except json.JSONDecodeError as e:
                                     st.error(f"Invalid JSON: {str(e)} 🚫")
+                                    
                     else:
                         st.sidebar.info("No documents found in this container. 📭")