Update app.py

app.py

@@ -18,9 +18,10 @@ import pytz
 import re
 from PIL import Image
 import zipfile
+import hashlib
 import time
 
-# App Configuration
+# 🎭 App Configuration - Because every app needs a good costume!
 Site_Name = '🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent'
 title = "🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent"
 helpURL = 'https://huggingface.co/awacke1'
@@ -39,23 +40,25 @@ st.set_page_config(
     }
 )
 
-# Cosmos DB configuration
+# 🌌 Cosmos DB configuration - Where data goes to party!
 ENDPOINT = "https://acae-afd.documents.azure.com:443/"
 DATABASE_NAME = os.environ.get("COSMOS_DATABASE_NAME")
 CONTAINER_NAME = os.environ.get("COSMOS_CONTAINER_NAME")
 Key = os.environ.get("Key")
 
-# Your local app URL (Change this to your app's URL)
+# 🌐 Your local app URL - Home sweet home
 LOCAL_APP_URL = "https://huggingface.co/spaces/awacke1/AzureCosmosDBUI"
 
-# Anthropic configuration
+# 🤖 Anthropic configuration - Teaching machines to be more human (and funnier)
 client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
 
-# Initialize session state
+# 🧠 Initialize session state - Because even apps need a good memory
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 
-# Helper Functions
+# 🛠️ Helper Functions - The unsung heroes of our code
+
+# 📎 Get a file download link - Making file sharing as easy as stealing candy from a baby
 def get_download_link(file_path):
     with open(file_path, "rb") as file:
         contents = file.read()
@@ -63,30 +66,33 @@ def get_download_link(file_path):
         file_name = os.path.basename(file_path)
         return f'<a href="data:file/txt;base64,{b64}" download="{file_name}">Download {file_name}📂</a>'
 
+# 🎲 Generate a unique ID - Because being unique is important (just ask your mother)
 def generate_unique_id():
-    now = datetime.now()
-    date_time = now.strftime("%d%m%Y%H%M%S")
-    ms = now.microsecond // 1000
-    unique_id = f"{date_time}{ms:03d}"
-    return unique_id
+    timestamp = datetime.utcnow().strftime('%Y%m%d%H%M%S%f')
+    unique_uuid = str(uuid.uuid4())
+    return f"{timestamp}-{unique_uuid}"
 
+# 📝 Generate a filename - Naming files like a pro (or a very confused librarian)
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
     safe_prompt = re.sub(r'\W+', '', prompt)[:90]
     return f"{safe_date_time}{safe_prompt}.{file_type}"
 
+# 💾 Create and save a file - Because data hoarding is a legitimate hobby
 def create_file(filename, prompt, response, should_save=True):
     if not should_save:
         return
     with open(filename, 'w', encoding='utf-8') as file:
         file.write(prompt + "\n\n" + response)
 
+# 📖 Load file content - Bringing words back from the digital grave
 def load_file(file_name):
     with open(file_name, "r", encoding='utf-8') as file:
         content = file.read()
     return content
 
+# 🔍 Display glossary entity - Making search fun again (as if it ever was)
 def display_glossary_entity(k):
     search_urls = {
         "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",
@@ -97,6 +103,7 @@ def display_glossary_entity(k):
     links_md = ' '.join([f"<a href='{url(k)}' target='_blank'>{emoji}</a>" for emoji, url in search_urls.items()])
     st.markdown(f"{k} {links_md}", unsafe_allow_html=True)
 
+# 🗜️ Create zip of files - Squeezing files together like sardines in a can
 def create_zip_of_files(files):
     zip_name = "all_files.zip"
     with zipfile.ZipFile(zip_name, 'w') as zipf:
@@ -104,6 +111,7 @@ def create_zip_of_files(files):
             zipf.write(file)
     return zip_name
 
+# 🎬 Get video HTML - Making videos play nice (or at least trying to)
 def get_video_html(video_path, width="100%"):
     video_url = f"data:video/mp4;base64,{base64.b64encode(open(video_path, 'rb').read()).decode()}"
     return f'''
@@ -113,6 +121,7 @@ def get_video_html(video_path, width="100%"):
     </video>
     '''
 
+# 🎵 Get audio HTML - Let the music play (and hope it's not Baby Shark)
 def get_audio_html(audio_path, width="100%"):
     audio_url = f"data:audio/mpeg;base64,{base64.b64encode(open(audio_path, 'rb').read()).decode()}"
     return f'''
@@ -122,18 +131,23 @@ def get_audio_html(audio_path, width="100%"):
     </audio>
     '''
 
-# Cosmos DB functions
+# 🌌 Cosmos DB functions - Where data goes to live its best life
+
+# 📚 Get databases - Collecting databases like Pokemon cards
 def get_databases(client):
     return [db['id'] for db in client.list_databases()]
 
+# 📦 Get containers - Finding where all the good stuff is hidden
 def get_containers(database):
     return [container['id'] for container in database.list_containers()]
 
+# 📄 Get documents - Retrieving the sacred texts (or just some JSON)
 def get_documents(container, limit=None):
     query = "SELECT * FROM c ORDER BY c._ts DESC"
     items = list(container.query_items(query=query, enable_cross_partition_query=True, max_item_count=limit))
     return items
 
+# 📥 Insert record - Adding new data (and crossing fingers it doesn't break anything)
 def insert_record(container, record):
     try:
         container.create_item(body=record)
@@ -143,6 +157,7 @@ def insert_record(container, record):
     except Exception as e:
         return False, f"An unexpected error occurred: {str(e)} 😱"
 
+# 🔄 Update record - Giving data a makeover
 def update_record(container, updated_record):
     try:
         container.upsert_item(body=updated_record)
@@ -152,6 +167,7 @@ def update_record(container, updated_record):
     except Exception as e:
         return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
+# 🗑️ Delete record - Saying goodbye to data (it's not you, it's me)
 def delete_record(container, name, id):
     try:
         container.delete_item(item=id, partition_key=id)
@@ -163,11 +179,71 @@ def delete_record(container, name, id):
     except Exception as e:
         return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
-import uuid
+# 💾 Save to Cosmos DB - Preserving data for future generations (or just until the next update)
+def save_to_cosmos_db(container, query, response1, response2):
+    try:
+        if container:
+            record = {
+                "id": generate_unique_id(),
+                "query": query,
+                "response1": response1,
+                "response2": response2,
+                "timestamp": datetime.utcnow().isoformat()
+            }
+            try:
+                container.create_item(body=record)
+                st.success(f"Record saved successfully with ID: {record['id']}")
+                # Refresh the documents display
+                st.session_state.documents = get_documents(container)
+            except exceptions.CosmosHttpResponseError as e:
+                st.error(f"Error saving record to Cosmos DB: {e}")
+        else:
+            st.error("Cosmos DB container is not initialized.")
+    except Exception as e:
+        st.error(f"An unexpected error occurred: {str(e)}")
 
-def generate_unique_id():
-    return str(uuid.uuid4())
-    
+# 🐙 GitHub functions - Where code goes to socialize
+
+# 📥 Download GitHub repo - Cloning repos like it's going out of style
+def download_github_repo(url, local_path):
+    if os.path.exists(local_path):
+        shutil.rmtree(local_path)
+    Repo.clone_from(url, local_path)
+
+# 🗜️ Create zip file - Squeezing files tighter than your budget
+def create_zip_file(source_dir, output_filename):
+    shutil.make_archive(output_filename, 'zip', source_dir)
+
+# 🏗️ Create repo - Building digital homes for lonely code
+def create_repo(g, repo_name):
+    user = g.get_user()
+    return user.create_repo(repo_name)
+
+# 🚀 Push to GitHub - Sending code to the cloud (hopefully not the rainy kind)
+def push_to_github(local_path, repo, github_token):
+    repo_url = f"https://{github_token}@github.com/{repo.full_name}.git"
+    local_repo = Repo(local_path)
+
+    if 'origin' in [remote.name for remote in local_repo.remotes]:
+        origin = local_repo.remote('origin')
+        origin.set_url(repo_url)
+    else:
+        origin = local_repo.create_remote('origin', repo_url)
+
+    if not local_repo.heads:
+        local_repo.git.checkout('-b', 'main')
+        current_branch = 'main'
+    else:
+        current_branch = local_repo.active_branch.name
+
+    local_repo.git.add(A=True)
+
+    if local_repo.is_dirty():
+        local_repo.git.commit('-m', 'Initial commit')
+
+    origin.push(refspec=f'{current_branch}:{current_branch}')
+
+# 💾 Save or clone to Cosmos DB - Because every document deserves a twin
 def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=None):
     try:
         if not container:
@@ -216,34 +292,52 @@ def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=No
     except Exception as e:
         st.error(f"An unexpected error occurred: {str(e)}")
 
+# 📦 Archive current container - Packing up data like you're moving to a new digital house
+def archive_current_container(database_name, container_name, client):
+    try:
+        base_dir = "./cosmos_archive_current_container"
+        if os.path.exists(base_dir):
+            shutil.rmtree(base_dir)
+        os.makedirs(base_dir)
+
+        db_client = client.get_database_client(database_name)
+        container_client = db_client.get_container_client(container_name)
+        items = list(container_client.read_all_items())
+        
+        container_dir = os.path.join(base_dir, container_name)
+        os.makedirs(container_dir)
+        
+        for item in items:
+            item_id = item.get('id', f"unknown_{datetime.now().strftime('%Y%m%d%H%M%S')}")
+            with open(os.path.join(container_dir, f"{item_id}.json"), 'w') as f:
+                json.dump(item, f, indent=2)
+        
+        archive_name = f"{container_name}_archive_{datetime.now().strftime('%Y%m%d%H%M%S')}"
+        shutil.make_archive(archive_name, 'zip', base_dir)
+        
+        return get_download_link(f"{archive_name}.zip")
+    except Exception as e:
+        return f"An error occurred while archiving data: {str(e)} 😢"
 
-# Add dropdowns for model and database choices
+# 🔍 Search glossary - Finding needles in digital haystacks
 def search_glossary(query):
-    st.markdown(f"### 🔍 Search Glossary for: `{query}`")
-    
+    st.markdown(f"### 🔍 SearchGlossary 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,
@@ -253,7 +347,7 @@ def search_glossary(query):
     )
     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,
@@ -274,7 +368,6 @@ def search_glossary(query):
     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
@@ -282,173 +375,25 @@ def search_glossary(query):
         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])
-
-        save_or_clone_to_cosmos_db(container=st.session_state.cosmos_container, query=query, response=result, clone_id=None)
-        save_or_clone_to_cosmos_db(container=st.session_state.cosmos_container, query=query, response=result2, clone_id=None)
-        save_or_clone_to_cosmos_db(container=st.session_state.cosmos_container, query=query, response=result3, clone_id=None)
-        save_or_clone_to_cosmos_db(container=st.session_state.cosmos_container, query=query, response=response2[0], clone_id=None)
-        save_or_clone_to_cosmos_db(container=st.session_state.cosmos_container, query=query, response=response2[1], clone_id=None)
-
-
-        
-    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)
+    return result, result2, result3, response2
 
-    # 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):
-        shutil.rmtree(local_path)
-    Repo.clone_from(url, local_path)
-
-def create_zip_file(source_dir, output_filename):
-    shutil.make_archive(output_filename, 'zip', source_dir)
-
-def create_repo(g, repo_name):
-    user = g.get_user()
-    return user.create_repo(repo_name)
-
-def push_to_github(local_path, repo, github_token):
-    repo_url = f"https://{github_token}@github.com/{repo.full_name}.git"
-    local_repo = Repo(local_path)
-
-    if 'origin' in [remote.name for remote in local_repo.remotes]:
-        origin = local_repo.remote('origin')
-        origin.set_url(repo_url)
-    else:
-        origin = local_repo.create_remote('origin', repo_url)
-
-    if not local_repo.heads:
-        local_repo.git.checkout('-b', 'main')
-        current_branch = 'main'
-    else:
-        current_branch = local_repo.active_branch.name
-
-    local_repo.git.add(A=True)
-
-    if local_repo.is_dirty():
-        local_repo.git.commit('-m', 'Initial commit')
-
-    origin.push(refspec=f'{current_branch}:{current_branch}')
-
-# Main function
+# 🎭 Main function - Where the magic happens (and occasionally breaks)
 def main():
     st.title("🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent")
 
-    # Initialize session state
+    # 🚦 Initialize session state
+    if 'logged_in' not in st.session_state:
+        st.session_state.logged_in = False
+    if 'selected_records' not in st.session_state:
+        st.session_state.selected_records = []
     if 'client' not in st.session_state:
         st.session_state.client = None
     if 'selected_database' not in st.session_state:
@@ -459,27 +404,47 @@ def main():
         st.session_state.selected_document_id = None
     if 'current_index' not in st.session_state:
         st.session_state.current_index = 0
-    if 'documents' not in st.session_state:
-        st.session_state.documents = []
-
-
+    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!
+            result, result2, result3, response2 = search_glossary(query)
+            # 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:
+                st.error(f"HTTP error occurred: {str(e)} 🚨")
+            except Exception as e:
+                st.error(f"An unexpected error occurred: {str(e)} 😱")
+
+            st.stop()  # Stop further execution
+    except Exception as e:
+        st.markdown(' ')
 
-    # Automatic Login
+    # 🔐 Automatic Login
     if Key:
         st.session_state.primary_key = Key
         st.session_state.logged_in = True
     else:
         st.error("Cosmos DB Key is not set in environment variables. 🔑❌")
-        return
+        return  # Can't proceed without a key
 
     if st.session_state.logged_in:
-        # Initialize Cosmos DB client
+        # 🌌 Initialize Cosmos DB client
         try:
             if st.session_state.client is None:
                 st.session_state.client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key)
             
-            # Sidebar for database, container, and document selection
+            # 🗄️ Sidebar for database, container, and document selection
             st.sidebar.title("🐙Git🌌Cosmos💫🗄️Navigator")
             
             databases = get_databases(st.session_state.client)
@@ -505,24 +470,8 @@ def main():
                 
                 if st.session_state.selected_container:
                     container = database.get_container_client(st.session_state.selected_container)
-
-                if selected_db != st.session_state.selected_database:
-                    st.session_state.selected_database = selected_db
-                    st.session_state.selected_container = None
-                    st.session_state.selected_document_id = None
-                    st.session_state.current_index = 0
-                    st.rerun()
-
-
-
-                    
-                                
-
-
-
-
                     
-                    # Add Export button
+                    # 📦 Add Export button
                     if st.button("📦 Export Container Data"):
                         download_link = archive_current_container(st.session_state.selected_database, st.session_state.selected_container, st.session_state.client)
                         if download_link.startswith('<a'):
@@ -533,28 +482,6 @@ def main():
                     # Fetch documents
                     documents = get_documents(container)
                     total_docs = len(documents)
-        
-                    # Add buttons for saving new document and cloning
-                    if st.button("💾 Save New Document"):
-                        user_query = st.session_state.get('user_query', '')
-                        ai_response = st.session_state.get('ai_response', '')
-                        save_or_clone_to_cosmos_db(container, query=user_query, response=ai_response)
-        
-                    if st.button("🔄 Clone Document"):
-                        doc_to_clone = st.selectbox("Select document to clone", [doc['id'] for doc in st.session_state.documents])
-                        if doc_to_clone:
-                            save_or_clone_to_cosmos_db(container, clone_id=doc_to_clone)
-        
-                    # Display documents
-                    if st.session_state.documents:
-                        for doc in st.session_state.documents:
-                            st.write(f"ID: {doc['id']}")
-                            st.write(f"Query: {doc.get('query', 'N/A')}")
-                            st.write(f"Response: {doc.get('response', 'N/A')}")
-                            st.write("---")
-        
-
-
                     
                     if total_docs > 5:
                         documents_to_display = documents[:5]
@@ -564,17 +491,17 @@ def main():
                         st.info(f"Showing all {len(documents_to_display)} documents.")
                     
                     if documents_to_display:
-                        # Add Viewer/Editor selection
+                        # 🎨 Add Viewer/Editor selection
                         view_options = ['Show as Markdown', 'Show as Code Editor', 'Show as Edit and Save', 'Clone Document', 'New Record']
                         selected_view = st.selectbox("Select Viewer/Editor", view_options, index=2)
                         
                         if selected_view == 'Show as Markdown':
-                            # Show each record as Markdown with navigation
+                            # 🖌️ Show each record as Markdown with navigation
                             total_docs = len(documents)
                             doc = documents[st.session_state.current_index]
                             st.markdown(f"#### Document ID: {doc.get('id', '')}")
 
-                            # Extract values from the JSON that have at least one space
+                            # 🕵️‍♂️ Let's extract values from the JSON that have at least one space
                             values_with_space = []
                             def extract_values(obj):
                                 if isinstance(obj, dict):
@@ -589,7 +516,7 @@ def main():
 
                             extract_values(doc)
 
-                            # Create a list of links for these values
+                            # 🔗 Let's create a list of links for these values
                             st.markdown("#### 🔗 Links for Extracted Texts")
                             for term in values_with_space:
                                 display_glossary_entity(term)
@@ -612,7 +539,7 @@ def main():
                                         st.rerun()
 
                         elif selected_view == 'Show as Code Editor':
-                            # Show each record in a code editor with navigation
+                            # 💻 Show each record in a code editor with navigation
                             total_docs = len(documents)
                             doc = documents[st.session_state.current_index]
                             st.markdown(f"#### Document ID: {doc.get('id', '')}")
@@ -642,7 +569,7 @@ def main():
                                     st.error(f"Invalid JSON: {str(e)} 🚫")
                                     
                         elif selected_view == 'Show as Edit and Save':
-                            # Show as Edit and Save in columns
+                            # ✏️ Show as Edit and Save in columns
                             st.markdown("#### Edit the document fields below:")
                             
                             # Create columns for each document
@@ -680,8 +607,8 @@ def main():
                                             search_glossary(json.dumps(editable_doc, indent=2))
                         
                         elif selected_view == 'Clone Document':
-                            # Clone Document per record
-                            st.markdown("#### Clone a document:")
+                            # 🧬 Clone Document per record
+                            st.markdownst.markdown("#### Clone a document:")
                             for idx, doc in enumerate(documents_to_display):
                                 st.markdown(f"##### Document ID: {doc.get('id', '')}")
                                 if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
@@ -712,10 +639,10 @@ def main():
                                         st.error(f"Invalid JSON: {str(e)} 🚫")
                                     
                         elif selected_view == 'New Record':
-                            # New Record
+                            # 🆕 New Record
                             st.markdown("#### Create a new document:")
                             if st.button("🤖 Insert Auto-Generated Record"):
-                                success, message = insert_auto_generated_record(container)
+                                success, message = save_or_clone_to_cosmos_db(container, query="Auto-generated", response="This is an auto-generated record.")
                                 if success:
                                     st.success(message)
                                     st.rerun()
@@ -742,7 +669,7 @@ def main():
                     else:
                         st.sidebar.info("No documents found in this container. 📭")
         
-            # Main content area
+            # 🎉 Main content area
             st.subheader(f"📊 Container: {st.session_state.selected_container}")
             if st.session_state.selected_container:
                 if documents_to_display:
@@ -751,7 +678,7 @@ def main():
                 else:
                     st.info("No documents to display. 🧐")
                 
-            # GitHub section
+            # 🐙 GitHub section
             st.subheader("🐙 GitHub Operations")
             github_token = os.environ.get("GITHUB")  # Read GitHub token from environment variable
             source_repo = st.text_input("Source GitHub Repository URL", value="https://github.com/AaronCWacker/AIExamples-8-24-Streamlit")
@@ -777,7 +704,7 @@ def main():
                                 os.remove(zip_filename)
                     else:
                         st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
-        
+    
             with col2:
                 if st.button("📤 Push to New Repository"):
                     if github_token and source_repo:
@@ -796,7 +723,7 @@ def main():
                     else:
                         st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
 
-            # Chat with Claude
+            # 💬 Chat with Claude
             st.subheader("💬 Chat with Claude")
             user_input = st.text_area("Message 📨:", height=100)
 
@@ -818,7 +745,7 @@ def main():
                     st.session_state.chat_history.append({"user": user_input, "claude": response.content[0].text})
 
                     # Save to Cosmos DB
-                    save_or_clone_to_cosmos_db(container=st.session_state.cosmos_container, query=user_input, response=response.content[0].text, clone_id=None)
+                    save_to_cosmos_db(container, user_input, response.content[0].text, "")
 
             # Display Chat History
             st.subheader("Past Conversations 📜")
@@ -873,20 +800,7 @@ def main():
         except Exception as e:
             st.error(f"An unexpected error occurred: {str(e)} 😱")
 
-
-    # ⚙️ 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(' ')
-
-        
-    # Logout button
+    # 🚪 Logout button
     if st.session_state.logged_in and st.sidebar.button("🚪 Logout"):
         st.session_state.logged_in = False
         st.session_state.selected_records.clear()