Update app.py

app.py

@@ -184,6 +184,190 @@ def save_to_cosmos_db(container, query, response1, response2):
     except Exception as e:
         st.error(f"An unexpected error occurred: {str(e)}")
 
+
+
+# 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"):
+    
+    # 🕵️‍♂️ Searching the glossary for: query
+    all_results = ""
+    st.markdown(f"- {query}")
+
+    # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM
+    #database_choice Literal['Semantic Search', 'Arxiv Search - Latest - (EXPERIMENTAL)'] Default: "Semantic Search"
+    #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")
+
+    
+    # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
+    result = client.predict(
+        prompt=query,
+        llm_model_picked="mistralai/Mixtral-8x7B-Instruct-v0.1",
+        stream_outputs=True,
+        api_name="/ask_llm"
+    )
+    st.markdown(result)
+    st.code(result, language="python", line_numbers=True)
+    
+    # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
+    result2 = client.predict(
+        prompt=query,
+        llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2",
+        stream_outputs=True,
+        api_name="/ask_llm"
+    )
+    st.markdown(result2)
+    st.code(result2, language="python", line_numbers=True)
+
+    # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
+    result3 = client.predict(
+        prompt=query,
+        llm_model_picked="google/gemma-7b-it",
+        stream_outputs=True,
+        api_name="/ask_llm"
+    )
+    st.markdown(result3)
+    st.code(result3, language="python", line_numbers=True)
+
+    
+    # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /update_with_rag_md
+    response2 = client.predict(
+        message=query,	# str  in 'parameter_13' Textbox component
+        llm_results_use=10,
+        database_choice="Semantic Search",
+        llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2",
+        api_name="/update_with_rag_md"
+    )        # update_with_rag_md     Returns tuple of 2 elements   [0] str   The output value that appears in the "value_14" Markdown component.   [1] str
+
+    st.markdown(response2[0])
+    st.code(response2[0], language="python", line_numbers=True, wrap_lines=True)  
+    
+    st.markdown(response2[1])
+    st.code(response2[1], language="python", line_numbers=True, wrap_lines=True)
+
+    # When saving results, pass the container
+    try:
+        save_to_cosmos_db(st.session_state.cosmos_container, query, result, result)
+        save_to_cosmos_db(st.session_state.cosmos_container, query, result2, result2)
+        save_to_cosmos_db(st.session_state.cosmos_container, query, result3, result3)
+        save_to_cosmos_db(st.session_state.cosmos_container, query, response2[0], response2[0])
+        save_to_cosmos_db(st.session_state.cosmos_container, query, response2[1], response2[1])
+    except exceptions.CosmosHttpResponseError as e:
+        return False, f"HTTP error occurred: {str(e)} 🚨"
+    except Exception as e:
+        return False, f"An unexpected error occurred: {str(e)} 😱"
+
+
+    try:    
+        # 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")
+    except exceptions.CosmosHttpResponseError as e:
+        return False, f"HTTP error occurred: {str(e)} 🚨"
+    except Exception as e:
+        return False, f"An unexpected error occurred: {str(e)} 😱"
+
+
+
+# 🎤 Function to process text input
+def process_text(text_input):
+    # 🎤 Processing text input—translating human words into cosmic signals! 📡
+    if text_input:
+        if 'messages' not in st.session_state:
+            st.session_state.messages = []
+        
+        st.session_state.messages.append({"role": "user", "content": text_input})
+
+        with st.chat_message("user"):
+            st.markdown(text_input)
+            
+        with st.chat_message("assistant"):
+            search_glossary(text_input)
+
+# 📝 Function to generate a filename
+def generate_filename(text, file_type):
+    # 📝 Generate a filename based on the text input
+    safe_text = "".join(c if c.isalnum() or c in (' ', '.', '_') else '_' for c in text)
+    safe_text = "_".join(safe_text.strip().split())
+    filename = f"{safe_text}.{file_type}"
+    return filename
+
+# 🕵️‍♀️ Function to extract markdown title
+def extract_markdown_title(content):
+    # 🕵️‍♀️ Extracting markdown title—finding the headline in the cosmic news! 📰
+    lines = content.splitlines()
+    for line in lines:
+        if line.startswith('#'):
+            return line.lstrip('#').strip()
+    return None
+
+# 💾 Function to create and save a file
+def create_and_save_file(content, file_type="md", prompt=None, is_image=False, should_save=True):
+    # 💾 Creating and saving a file—capturing cosmic wisdom! 📝
+    if not should_save:
+        return None
+
+    # Step 1: Generate filename based on the prompt or content
+    filename = generate_filename(prompt if prompt else content, file_type)
+
+    # Step 2: If it's a markdown file, check if it has a title
+    if file_type == "md":
+        title_from_content = extract_markdown_title(content)
+        if title_from_content:
+            filename = generate_filename(title_from_content, file_type)
+
+    # Step 3: Save the file
+    with open(filename, "w", encoding="utf-8") as f:
+        if is_image:
+            f.write(content)
+        else:
+            f.write(prompt + "\n\n" + content)
+
+    return filename
+
+# 🤖 Function to insert an auto-generated record
+def insert_auto_generated_record(container):
+    # 🤖 Automatically generating a record and inserting it into Cosmos DB!
+    try:
+        # Generate a unique id
+        new_id = generate_unique_id()
+        # Create a sample JSON document
+        new_doc = {
+            'id': new_id,
+            'name': f'Sample Name {new_id[:8]}',
+            'description': 'This is a sample auto-generated description.',
+            'timestamp': datetime.utcnow().isoformat()
+        }
+        # Insert the document
+        container.create_item(body=new_doc)
+        return True, f"Record inserted successfully with id: {new_id} 🎉"
+    except exceptions.CosmosHttpResponseError as e:
+        return False, f"HTTP error occurred: {str(e)} 🚨"
+    except Exception as e:
+        return False, f"An unexpected error occurred: {str(e)} 😱"
+
+
+
 # GitHub functions
 def download_github_repo(url, local_path):
     if os.path.exists(local_path):
@@ -242,6 +426,17 @@ def main():
     if 'cloned_doc' not in st.session_state:
         st.session_state.cloned_doc = None
 
+    # ⚙️ q= Run ArXiv search from query parameters
+    try:
+        query_params = st.query_params
+        query = query_params.get('q') or query_params.get('query') or ''
+        if query:
+            # 🕵️‍♂️ We have a query! Let's process it!
+            process_text(query)
+            st.stop()  # Stop further execution
+    except Exception as e:
+        st.markdown(' ')
+
     # Automatic Login
     if Key:
         st.session_state.primary_key = Key