Update app.py

app.py

@@ -1,6 +1,4 @@
-# =============================================================================
-# ───────────── IMPORTS ─────────────
-# =============================================================================
+# 📚📚📚 IMPORTS 📚📚📚
 import base64
 import glob
 import hashlib
@@ -28,9 +26,7 @@ import requests
 import numpy as np
 from urllib.parse import quote
 
-# =============================================================================
-# ───────────── EXTERNAL HELP LINKS (Always visible in sidebar) ─────────────
-# =============================================================================
+# 🌐🌐🌐 EXTERNAL HELP LINKS (Always visible in sidebar) 🌐🌐🌐
 external_links = [
     {"title": "CosmosDB GenAI Full Text Search", "url": "https://learn.microsoft.com/en-us/azure/cosmos-db/gen-ai/full-text-search", "emoji": "💻"},
     {"title": "CosmosDB SQL API Client Library", "url": "https://learn.microsoft.com/en-us/python/api/overview/azure/cosmos-readme?view=azure-python", "emoji": "💻"},
@@ -45,9 +41,7 @@ external_links = [
     {"title": "arcee.ai Official Website", "url": "https://arcee.ai", "emoji": "🌐"},
 ]
 
-# =============================================================================
-# ───────────── APP CONFIGURATION ─────────────
-# =============================================================================
+# 🚀🚀🚀 APP CONFIGURATION 🚀🚀🚀
 Site_Name = '🐙 GitCosmos'
 title = "🐙 GitCosmos"
 helpURL = 'https://huggingface.co/awacke1'
@@ -65,7 +59,7 @@ st.set_page_config(
     }
 )
 
-# Cosmos DB & App URLs
+# 🌌🌌🌌 COSMOS DB & APP CONFIGURATION 🌌🌌🌌
 ENDPOINT = "https://acae-afd.documents.azure.com:443/"
 DATABASE_NAME = os.environ.get("COSMOS_DATABASE_NAME")
 CONTAINER_NAME = os.environ.get("COSMOS_CONTAINER_NAME")
@@ -73,9 +67,7 @@ Key = os.environ.get("Key")
 LOCAL_APP_URL = "https://huggingface.co/spaces/awacke1/AzureCosmosDBUI"
 CosmosDBUrl = 'https://portal.azure.com/#@AaronCWackergmail.onmicrosoft.com/resource/subscriptions/003fba60-5b3f-48f4-ab36-3ed11bc40816/resourceGroups/datasets/providers/Microsoft.DocumentDB/databaseAccounts/acae-afd/dataExplorer'
 
-# =============================================================================
-# ───────────── HELPER FUNCTIONS ─────────────
-# =============================================================================
+# 🔧🔧🔧 HELPER FUNCTIONS 🔧🔧🔧
 def get_download_link(file_path):
     with open(file_path, "rb") as file:
         contents = file.read()
@@ -154,9 +146,7 @@ def sanitize_json_text(text):
     text = text.replace("\n", "\\n").replace("\r", "\\r").replace("\t", "\\t")
     return text
 
-# =============================================================================
-# ───────────── COSMOS DB FUNCTIONS ─────────────
-# =============================================================================
+# 🌌🌌🌌 COSMOS DB FUNCTIONS 🌌🌌🌌
 def get_databases(client):
     return [db['id'] for db in client.list_databases()]
 
@@ -222,29 +212,19 @@ def archive_current_container(database_name, container_name, client):
     except Exception as e:
         return f"Archive error: {str(e)} 😢"
 
-# =============================================================================
-# ───────────── ADVANCED COSMOS FUNCTIONS ─────────────
+# 🌠🌠🌠 ADVANCED COSMOS FUNCTIONS (if needed) 🌠🌠🌠
 # (Include your advanced Cosmos functions here if needed)
-# =============================================================================
-# For example, you might have functions like:
-# def create_new_container(...):
-#     ...
-
-# =============================================================================
-# ───────────── NEW COSMOSDB DEMO FUNCTIONS ─────────────
-# Each function below corresponds to one of your provided code snippets.
-# =============================================================================
+# For example, you might have functions like create_new_container(), etc.
 
+# 🎉🎉🎉 NEW COSMOSDB DEMO FUNCTIONS (Using ENDPOINT and Key) 🎉🎉🎉
 def demo_create_database():
     st.markdown("### Demo: Create Database")
     database_name = st.text_input("Enter Database Name", value="testDatabase", key="demo_db_name")
     if st.button("Create Database", key="btn_create_db"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.create_database(database_name)
             st.success(f"Database '{database_name}' created.")
@@ -262,12 +242,10 @@ def demo_create_container():
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_name")
     partition_key = st.text_input("Enter Partition Key Path", value="/productName", key="demo_partition_key")
     if st.button("Create Container", key="btn_create_container"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.create_container(id=container_name, partition_key=PartitionKey(path=partition_key))
@@ -288,12 +266,10 @@ def demo_create_analytical_container():
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_analytical")
     partition_key = st.text_input("Enter Partition Key Path", value="/productName", key="demo_partition_key_analytical")
     if st.button("Create Analytical Container", key="btn_create_analytical"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.create_container(
@@ -317,12 +293,10 @@ def demo_get_existing_container():
     database_name = st.text_input("Enter Database Name", value="testDatabase", key="demo_db_get")
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_get")
     if st.button("Get Container", key="btn_get_container"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.get_container_client(container_name)
@@ -337,12 +311,10 @@ def demo_insert_data():
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_insert")
     num_items = st.number_input("Number of items to insert", min_value=1, max_value=20, value=9, key="demo_num_items")
     if st.button("Insert Data", key="btn_insert_data"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.get_container_client(container_name)
@@ -362,12 +334,10 @@ def demo_delete_data():
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_delete")
     query_model = st.text_input("Product Model to delete", value="Model 2", key="demo_query_model")
     if st.button("Delete Data", key="btn_delete_data"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.get_container_client(container_name)
@@ -389,12 +359,10 @@ def demo_query_database():
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_query")
     query_str = st.text_area("Enter SQL Query", value='SELECT * FROM mycontainer r WHERE r.id="item3"', key="demo_query_str")
     if st.button("Run Query", key="btn_query_database"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.get_container_client(container_name)
@@ -413,12 +381,10 @@ def demo_parameterized_query():
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_param")
     model_value = st.text_input("Enter productModel value", value="Model 7", key="demo_model_value")
     if st.button("Run Parameterized Query", key="btn_param_query"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.get_container_client(container_name)
@@ -436,12 +402,10 @@ def demo_get_db_properties():
     st.markdown("### Demo: Get Database Properties")
     database_name = st.text_input("Enter Database Name", value="testDatabase", key="demo_db_props")
     if st.button("Get Properties", key="btn_db_props"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             props = database.read()
@@ -454,12 +418,10 @@ def demo_get_throughput():
     database_name = st.text_input("Enter Database Name", value="testDatabase", key="demo_db_throughput")
     container_name = st.text_input("Enter Container Name (for container throughput)", value="testContainer", key="demo_container_throughput")
     if st.button("Get Throughput", key="btn_get_throughput"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             db_offer = database.get_throughput()
@@ -479,16 +441,14 @@ def demo_modify_container_properties():
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_modify")
     new_ttl = st.number_input("Enter new default TTL (seconds)", min_value=0, value=10, key="demo_new_ttl")
     if st.button("Modify Container", key="btn_modify_container"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.get_container_client(container_name)
-            # Note: replace_container is used here as per your snippet.
+            # Using replace_container as in your snippet.
             database.replace_container(
                 container,
                 partition_key=PartitionKey(path="/productName"),
@@ -504,12 +464,10 @@ def demo_item_response_headers():
     database_name = st.text_input("Enter Database Name", value="testDatabase", key="demo_db_headers")
     container_name = st.text_input("Enter Container Name", value="products", key="demo_container_headers")
     if st.button("Create Item & Show Headers", key="btn_item_headers"):
-        url = os.environ.get("ACCOUNT_URI")
-        key_env = os.environ.get("ACCOUNT_KEY")
-        if not url or not key_env:
-            st.error("Missing ACCOUNT_URI or ACCOUNT_KEY environment variables.")
+        if not ENDPOINT or not Key:
+            st.error("Missing Cosmos configuration: ENDPOINT or Key")
             return
-        client_demo = CosmosClient(url, credential=key_env)
+        client_demo = CosmosClient(ENDPOINT, credential=Key)
         try:
             database = client_demo.get_database_client(database_name)
             container = database.get_container_client(container_name)
@@ -520,9 +478,7 @@ def demo_item_response_headers():
         except Exception as e:
             st.error(f"Error: {str(e)}")
 
-# =============================================================================
-# ───────────── FILE & MEDIA MANAGEMENT FUNCTIONS ─────────────
-# =============================================================================
+# 📁📁📁 FILE & MEDIA MANAGEMENT FUNCTIONS 📁📁📁
 def display_saved_files_in_sidebar():
     all_files = sorted([f for f in glob.glob("*.md") if not f.lower().startswith('readme')], reverse=True)
     st.sidebar.markdown("## 📁 Files")
@@ -654,9 +610,7 @@ def update_file_management_section():
         elif st.session_state.file_view_mode == 'edit':
             display_file_editor(st.session_state.current_file)
 
-# =============================================================================
-# ───────────── SIDEBAR DATA GRID (Records with formatted timestamps) ─────────────
-# =============================================================================
+# 📊📊📊 SIDEBAR DATA GRID (Records with formatted timestamps) 📊📊📊
 def show_sidebar_data_grid():
     if st.session_state.get("current_container"):
         try:
@@ -682,9 +636,7 @@ def show_sidebar_data_grid():
     else:
         st.sidebar.info("No container selected for data grid.")
 
-# =============================================================================
-# ───────────── DOCUMENTS LIST VIEW (Editable List with Sorting) ─────────────
-# =============================================================================
+# 📝📝📝 DOCUMENTS LIST VIEW (Editable List with Sorting) 📝📝📝
 def edit_documents_list(container):
     records = get_documents(container)
     sort_option = st.selectbox("Sort by", ["Timestamp", "Name"], key="sort_option")
@@ -724,9 +676,7 @@ def edit_documents_list(container):
                         st.error(f"Error updating {doc_id}: {message}")
         st.rerun()
 
-# =============================================================================
-# ───────────── VIDEO & AUDIO UI FUNCTIONS ─────────────
-# =============================================================================
+# 🎥🎵🎥 VIDEO & AUDIO UI FUNCTIONS 🎥🎵🎥
 def validate_and_preprocess_image(file_data, target_size=(576, 1024)):
     try:
         st.write("Preprocessing image...")
@@ -840,9 +790,7 @@ def add_video_generation_ui(container):
         except Exception as e:
             st.error(f"Upload error: {str(e)}")
 
-# =============================================================================
-# ───────────── NEW ITEM & FIELD FUNCTIONS ─────────────
-# =============================================================================
+# ➕➕➕ NEW ITEM & FIELD FUNCTIONS ➕➕➕
 def new_item_default(container):
     new_id = generate_unique_id()
     default_doc = {
@@ -887,9 +835,7 @@ def add_field_to_doc():
     except Exception as e:
          st.error(f"Error adding field: {str(e)}")
 
-# =============================================================================
-# ───────────── VECTOR SEARCH INTERFACE ─────────────
-# =============================================================================
+# 🔍🔍🔍 VECTOR SEARCH INTERFACE 🔍🔍🔍
 def vector_keyword_search(keyword, container):
     try:
         query = f"SELECT * FROM c WHERE CONTAINS(c.content, '{keyword}')"
@@ -916,9 +862,7 @@ def display_search_results(keyword, container):
                 except Exception as e:
                     st.error(f"Error saving {doc_id}: {str(e)}")
 
-# =============================================================================
-# ───────────── NEW AI MODALITY RECORD TEMPLATES ─────────────
-# =============================================================================
+# 🤖🤖🤖 NEW AI MODALITY RECORD TEMPLATES 🤖🤖🤖
 def new_ai_record(container):
     new_id = generate_unique_id()
     default_doc = {
@@ -960,9 +904,7 @@ def new_links_record(container):
          st.error("Error creating links record: " + message)
          return None
 
-# =============================================================================
-# ───────────── SIDEBAR DATA GRID (Editable Names Grid) ─────────────
-# =============================================================================
+# 🏷️🏷️🏷️ SIDEBAR DATA GRID (Editable Names Grid) 🏷️🏷️🏷️
 def edit_names_grid(container):
     records = get_documents(container)
     data = []
@@ -995,10 +937,7 @@ def edit_names_grid(container):
                         st.sidebar.error(f"Update error for {doc_id}: {message}")
         st.rerun()
 
-# =============================================================================
-# ───────────── SEARCH DOCUMENTS UI ─────────────
-# This function was missing and is now defined.
-# =============================================================================
+# 🔎🔎🔎 SEARCH DOCUMENTS UI 🔎🔎🔎
 def search_documents_ui(container):
     with st.sidebar.form("search_form"):
         keyword = st.text_input("Search Keyword", key="search_keyword")
@@ -1006,9 +945,7 @@ def search_documents_ui(container):
         if submitted and keyword:
             display_search_results(keyword, container)
 
-# =============================================================================
-# ───────────── MAIN FUNCTION ─────────────
-# =============================================================================
+# 🛠️🛠️🛠️ MAIN FUNCTION 🛠️🛠️🛠️
 def main():
     st.markdown("### 🐙 GitCosmos - Cosmos & Git Hub")
     st.markdown(f"[🔗 Portal]({CosmosDBUrl})")
@@ -1022,7 +959,7 @@ def main():
         st.error("Missing Cosmos Key 🔑❌")
         return
 
-    # ── New: CosmosDB Demo Features Section in Sidebar ──
+    # ── CosmosDB Demo Features in Sidebar ──
     st.sidebar.markdown("## CosmosDB Demo Features")
     demo_feature = st.sidebar.selectbox("Select a Demo", 
                                         ["Select", "Create Database", "Create Container", "Create Analytical Container",
@@ -1056,7 +993,7 @@ def main():
         elif demo_feature == "Item Response Headers":
             demo_item_response_headers()
 
-    # ── Existing Sidebar Items (Item Management, File Management, etc.)
+    # ── Existing Sidebar Items (Item Management, File Management, etc.) ──
     st.sidebar.markdown("## 🛠️ Item Management")
     if st.sidebar.button("New Item"):
         if st.session_state.get("current_container"):
@@ -1086,7 +1023,7 @@ def main():
     search_documents_ui(st.session_state.get("current_container"))
     show_sidebar_data_grid()
 
-    # (Your remaining navigation, file management, and document editing UI follow)
+    # Continue with your other UI sections (navigation, file management, document editing, etc.)
     try:
         if st.session_state.get("client") is None:
             st.session_state.client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key)
@@ -1113,7 +1050,7 @@ def main():
                     submitted = st.form_submit_button("Create Container")
                     if submitted:
                         analytical_ttl = -1 if new_analytical else None
-                        # Assuming create_new_container is defined in your advanced functions:
+                        # Using create_container method directly (adjust if you have an advanced function)
                         new_container = database.create_container(
                             id=new_container_id,
                             partition_key=PartitionKey(path=new_partition_key),
@@ -1223,7 +1160,6 @@ def main():
                     ai_query = st.text_area("Enter your query for ArXiv search:", key="arxiv_query", height=100)
                     if st.button("Send"):
                         st.session_state.last_query = ai_query
-                        # perform_ai_lookup would be your AI function call.
                         st.info("Performing AI lookup (function not implemented in this demo)...")
                 elif selected_view == 'Clone':
                     st.markdown("#### 📄 Clone")