Update app.py

app.py

@@ -1,4 +1,7 @@
-# %% ───────────── IMPORTS ─────────────
+# app.py
+# =============================================================================
+# ───────────── IMPORTS ─────────────
+# =============================================================================
 import base64
 import glob
 import hashlib
@@ -26,7 +29,9 @@ import requests
 import numpy as np
 from urllib.parse import quote
 
-# External help links for AI tools
+# =============================================================================
+# ───────────── EXTERNAL HELP LINKS (Always visible in sidebar) ─────────────
+# =============================================================================
 external_links = [
     {"title": "MergeKit Official GitHub", "url": "https://github.com/arcee-ai/MergeKit", "emoji": "💻"},
     {"title": "MergeKit arXiv Paper", "url": "https://arxiv.org/abs/xxxx.xxxxx", "emoji": "📘"},
@@ -40,7 +45,9 @@ 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'
@@ -66,7 +73,9 @@ 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()
@@ -140,7 +149,9 @@ def preprocess_text(text):
     text = re.sub(r'[^\x00-\x7F]+', '', text)
     return text.strip()
 
-# %% ───────────── COSMOS DB FUNCTIONS ─────────────
+# =============================================================================
+# ───────────── COSMOS DB FUNCTIONS ─────────────
+# =============================================================================
 def get_databases(client):
     return [db['id'] for db in client.list_databases()]
 
@@ -178,7 +189,7 @@ def delete_record(container, record):
         if "delete_log" not in st.session_state:
             st.session_state.delete_log = []
         st.session_state.delete_log.append(f"Attempting to delete document: {json.dumps(record, indent=2)}")
-        partition_key_value = doc_id
+        partition_key_value = record.get("pk", doc_id)
         st.session_state.delete_log.append(f"Using ID and Partition Key: {partition_key_value}")
         container.delete_item(item=doc_id, partition_key=partition_key_value)
         success_msg = f"Record {doc_id} successfully deleted from Cosmos DB. 🗑️"
@@ -205,6 +216,7 @@ def save_to_cosmos_db(container, query, response1, response2):
             new_id = f"{timestamp}-{unique_uuid}"
             record = {
                 "id": new_id,
+                "pk": new_id,
                 "name": new_id,
                 "query": query,
                 "response1": response1,
@@ -242,11 +254,12 @@ 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 ─────────────
+# =============================================================================
 def create_new_container(database, container_id, partition_key_path,
                          analytical_storage_ttl=None, indexing_policy=None, vector_embedding_policy=None):
     try:
-        # Try creating with analytical_storage_ttl if provided.
         if analytical_storage_ttl is not None:
             container = database.create_container(
                 id=container_id,
@@ -263,7 +276,6 @@ def create_new_container(database, container_id, partition_key_path,
                 vector_embedding_policy=vector_embedding_policy
             )
     except exceptions.CosmosHttpResponseError as e:
-        # If the error is due to the analytical_storage_ttl property not being valid, try again without it.
         if analytical_storage_ttl is not None and "analyticalStorageTtl" in str(e):
             try:
                 container = database.create_container(
@@ -311,7 +323,9 @@ def vector_search(container, query_vector, vector_field, top=10, exact_search=Fa
     results = list(container.query_items(query=query, enable_cross_partition_query=True))
     return results
 
-# %% ───────────── GITHUB FUNCTIONS ─────────────
+# =============================================================================
+# ───────────── GITHUB FUNCTIONS ─────────────
+# =============================================================================
 def download_github_repo(url, local_path):
     if os.path.exists(local_path):
         shutil.rmtree(local_path)
@@ -342,7 +356,9 @@ def push_to_github(local_path, repo, github_token):
         local_repo.git.commit('-m', 'Initial commit')
     origin.push(refspec=f'{current_branch}:{current_branch}')
 
-# %% ───────────── 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")
@@ -470,7 +486,9 @@ def update_file_management_section():
         elif st.session_state.file_view_mode == 'edit':
             display_file_editor(st.session_state.current_file)
 
-# %% ───────────── VIDEO & AUDIO UI FUNCTIONS ─────────────
+# =============================================================================
+# ───────────── VIDEO & AUDIO UI FUNCTIONS ─────────────
+# =============================================================================
 def validate_and_preprocess_image(file_data, target_size=(576, 1024)):
     try:
         st.write("Preprocessing image...")
@@ -556,6 +574,7 @@ def add_video_generation_ui(container):
                                     if container:
                                         video_record = {
                                             "id": generate_unique_id(),
+                                            "pk": generate_unique_id(),
                                             "type": "generated_video",
                                             "filename": video_filename,
                                             "seed": seed if seed is not None else "random",
@@ -583,7 +602,63 @@ def add_video_generation_ui(container):
         except Exception as e:
             st.error(f"Upload error: {str(e)}")
 
-# %% ───────────── MAIN FUNCTION ─────────────
+# =============================================================================
+# ───────────── AI SAMPLES SIDEBAR (Processed as a Python List) ─────────────
+# =============================================================================
+def display_ai_samples():
+    # Define a list of sample queries
+    ai_samples = [
+        {
+            "name": "FullTextContains",
+            "description": "Query using FullTextContains",
+            "query": 'SELECT TOP 10 * FROM c WHERE FullTextContains(c.text, "bicycle")'
+        },
+        {
+            "name": "FullTextContainsAll",
+            "description": "Query using FullTextContainsAll",
+            "query": 'SELECT TOP 10 * FROM c WHERE FullTextContainsAll(c.text, "red", "bicycle")'
+        },
+        {
+            "name": "FullTextContainsAny",
+            "description": "Query using FullTextContainsAny",
+            "query": 'SELECT TOP 10 * FROM c WHERE FullTextContains(c.text, "red") AND FullTextContainsAny(c.text, "bicycle", "skateboard")'
+        },
+        {
+            "name": "FullTextScore",
+            "description": "Query using FullTextScore (order by relevance)",
+            "query": 'SELECT TOP 10 * FROM c ORDER BY RANK FullTextScore(c.text, ["bicycle", "mountain"])'
+        },
+        {
+            "name": "Vector Search with Score",
+            "description": "Example vector search snippet",
+            "query": 'results = vector_search.similarity_search_with_score(query="Your query", k=5)\nfor result, score in results:\n    print(result.json(), score)'
+        },
+        {
+            "name": "Vector Search with Filtering",
+            "description": "Example vector search with a filter",
+            "query": 'pre_filter = {"conditions": [{"property": "metadata.page", "operator": "$eq", "value": 0}]}\nresults = vector_search.similarity_search_with_score(query="Your query", k=5, pre_filter=pre_filter)'
+        },
+        {
+            "name": "Hybrid Search",
+            "description": "Example hybrid search snippet",
+            "query": 'results = vector_search.similarity_search_with_score(query="Your query", k=5, query_type=CosmosDBQueryType.HYBRID)'
+        }
+    ]
+    st.sidebar.markdown("### 🤖 AI Samples")
+    st.sidebar.info("🚀 Get started with our AI samples! Time free access to get started today.")
+    # Provide a dropdown to select one sample
+    sample_names = [sample["name"] for sample in ai_samples]
+    selected_sample_name = st.sidebar.selectbox("Select an AI Sample", sample_names)
+    selected_sample = next((s for s in ai_samples if s["name"] == selected_sample_name), None)
+    if selected_sample:
+        st.sidebar.markdown(f"**{selected_sample['name']}**: {selected_sample['description']}")
+        # Use language 'sql' for queries containing FullText, else python
+        lang = "sql" if "FullText" in selected_sample["name"] else "python"
+        st.sidebar.code(selected_sample["query"], language=lang)
+
+# =============================================================================
+# ───────────── MAIN FUNCTION ─────────────
+# =============================================================================
 def main():
     st.markdown("### 🐙 GitCosmos - Cosmos & Git Hub")
     if "chat_history" not in st.session_state:
@@ -619,9 +694,9 @@ def main():
                     st.session_state.show_new_container_form = True
                 if st.session_state.show_new_container_form:
                     with st.sidebar.form("new_container_form"):
-                        new_container_id = st.text_input("Container ID", value="newContainer")
-                        new_partition_key = st.text_input("Partition Key", value="/id")
-                        new_analytical = st.checkbox("Enable Analytical Store", value=False)
+                        new_container_id = st.text_input("Container ID", value="aiml-container")
+                        new_partition_key = st.text_input("Partition Key", value="/pk")
+                        new_analytical = st.checkbox("Enable Analytical Store", value=True)
                         submitted = st.form_submit_button("Create Container")
                         if submitted:
                             analytical_ttl = -1 if new_analytical else None
@@ -633,6 +708,21 @@ def main():
                             )
                             if new_container:
                                 st.success(f"Container '{new_container_id}' created.")
+                                # Insert a default templated item into the new container
+                                default_id = generate_unique_id()
+                                default_item = {
+                                    "id": default_id,
+                                    "pk": default_id,
+                                    "name": "Default Image Prompt",
+                                    "prompt": "Enter your image prompt here",
+                                    "timestamp": datetime.now().isoformat(),
+                                    "type": "image_prompt"
+                                }
+                                insert_success, insert_message = insert_record(new_container, default_item)
+                                if insert_success:
+                                    st.info("Default templated item created in new container.")
+                                else:
+                                    st.error(f"Default item insertion error: {insert_message}")
                                 st.session_state.show_new_container_form = False
                                 st.session_state.new_container_created = new_container_id
                                 st.rerun()
@@ -728,7 +818,7 @@ def main():
                             st.markdown(f"Original ID: {doc.get('id', '')}")
                             new_id = st.text_input("New ID", value=generate_unique_id(), key='new_clone_id')
                             new_name = st.text_input("New Name", value=f"Clone_{new_id[:8]}", key='new_clone_name')
-                            new_doc = {'id': new_id, 'name': new_name, **{k: v for k, v in doc.items() if k not in ['id', 'name', '_rid', '_self', '_etag', '_attachments', '_ts']}}
+                            new_doc = {'id': new_id, 'pk': new_id, 'name': new_name, **{k: v for k, v in doc.items() if k not in ['id', 'name', 'pk', '_rid', '_self', '_etag', '_attachments', '_ts']}}
                             doc_str = st.text_area("Edit JSON", value=json.dumps(new_doc, indent=2), height=300, key='clone_preview')
                             col1, col2 = st.columns(2)
                             with col1:
@@ -761,6 +851,7 @@ def main():
                         if st.button("🤖 Auto-Gen"):
                             auto_doc = {
                                 "id": generate_unique_id(),
+                                "pk": generate_unique_id(),
                                 "name": f"Auto {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
                                 "content": "Auto-generated record.",
                                 "timestamp": datetime.now().isoformat()
@@ -775,6 +866,7 @@ def main():
                             new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
                             default_doc = {
                                 "id": new_id,
+                                "pk": new_id,
                                 "name": "New Doc",
                                 "content": "",
                                 "timestamp": datetime.now().isoformat()
@@ -785,6 +877,7 @@ def main():
                                     cleaned = preprocess_text(new_doc_str)
                                     new_doc = json.loads(cleaned)
                                     new_doc['id'] = new_id
+                                    new_doc['pk'] = new_id
                                     success, message = insert_record(container, new_doc)
                                     if success:
                                         st.success(f"Created {new_doc['id']}")
@@ -842,6 +935,8 @@ def main():
                                         shutil.rmtree(local_path)
                             else:
                                 st.error("Missing token or URL 🔑❓")
+                    # Display AI Samples sidebar UI (processed from Python list)
+                    display_ai_samples()
                     update_file_management_section()
         except exceptions.CosmosHttpResponseError as e:
             st.error(f"Cosmos error: {str(e)} 🚨")