Update app.py

app.py

@@ -15,7 +15,7 @@ import traceback
 import uuid
 import zipfile
 from PIL import Image
-from azure.cosmos import CosmosClient, exceptions
+from azure.cosmos import CosmosClient, PartitionKey, exceptions
 from datetime import datetime
 from git import Repo
 from github import Github
@@ -187,42 +187,23 @@ def update_record(container, updated_record):
 
 # 🗑️ Delete a record from Cosmos DB using /id as partition key
 def delete_record(container, record):
-   """
-   Delete a record from Cosmos DB using its 'id' field as both the item ID and partition key.
-   
-   Args:
-       container: Cosmos DB container client
-       record: Dictionary containing at least 'id'
-   
-   Returns:
-       tuple: (success: bool, message: str)
-   """
    try:
-       # Ensure record has an ID
        if "id" not in record:
            return False, "Record must contain an 'id' field. 🛑"
-
        doc_id = record["id"]
-       
-       # Log the document being deleted (persistent in session state)
        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)}")
-       
-       # Use 'id' as the partition key since container is partitioned on /id
        partition_key_value = doc_id
        st.session_state.delete_log.append(f"Using ID and Partition Key: {partition_key_value}")
-
-       # Perform the deletion
        container.delete_item(item=doc_id, partition_key=partition_key_value)
        success_msg = f"Record {doc_id} successfully deleted from Cosmos DB. 🗑️"
        st.session_state.delete_log.append(success_msg)
        return True, success_msg
-
    except exceptions.CosmosResourceNotFoundError:
        success_msg = f"Record {doc_id} not found in Cosmos DB (already deleted or never existed). 🗑️"
        st.session_state.delete_log.append(success_msg)
-       return True, success_msg  # Treat as success since the goal is removal
+       return True, success_msg
    except exceptions.CosmosHttpResponseError as e:
        error_msg = f"HTTP error deleting {doc_id}: {str(e)}. 🚨"
        st.session_state.delete_log.append(error_msg)
@@ -279,23 +260,75 @@ def archive_current_container(database_name, container_name, client):
    except Exception as e:
        return f"Archive error: {str(e)} 😢"
 
+# %% ───────────── 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:
+        if analytical_storage_ttl is not None:
+            container = database.create_container(
+                id=container_id,
+                partition_key=PartitionKey(path=partition_key_path),
+                analytical_storage_ttl=analytical_storage_ttl,
+                indexing_policy=indexing_policy,
+                vector_embedding_policy=vector_embedding_policy
+            )
+        else:
+            container = database.create_container(
+                id=container_id,
+                partition_key=PartitionKey(path=partition_key_path),
+                indexing_policy=indexing_policy,
+                vector_embedding_policy=vector_embedding_policy
+            )
+    except exceptions.CosmosResourceExistsError:
+        container = database.get_container_client(container_id)
+    except exceptions.CosmosHttpResponseError as e:
+        st.error(f"Error creating container: {str(e)}")
+        return None
+    return container
+
+def advanced_insert_item(container, item):
+    try:
+        container.upsert_item(item)
+        return True, f"Item {item.get('id', '')} inserted. ➕"
+    except Exception as e:
+        return False, str(e)
+
+def advanced_update_item(container, item):
+    try:
+        container.upsert_item(item)
+        return True, f"Item {item.get('id', '')} updated. ✏️"
+    except Exception as e:
+        return False, str(e)
+
+def advanced_delete_item(container, item_id, partition_key_value):
+    try:
+        container.delete_item(item=item_id, partition_key=partition_key_value)
+        return True, f"Item {item_id} deleted. 🗑️"
+    except Exception as e:
+        return False, str(e)
+
+def vector_search(container, query_vector, vector_field, top=10, exact_search=False):
+    # Convert the vector list to a JSON string
+    query_vector_str = json.dumps(query_vector)
+    query = f"""SELECT TOP {top} c.id, VectorDistance(c.{vector_field}, {query_vector_str}, {str(exact_search).lower()}, 
+    {{'dataType':'float32','distanceFunction':'cosine'}}) AS SimilarityScore 
+    FROM c ORDER BY SimilarityScore"""
+    results = list(container.query_items(query=query, enable_cross_partition_query=True))
+    return results
+
 # %% ───────────── GITHUB FUNCTIONS ─────────────
-# 📥 Clone a GitHub repository locally
 def download_github_repo(url, local_path):
    if os.path.exists(local_path):
        shutil.rmtree(local_path)
    Repo.clone_from(url, local_path)
 
-# 🗜️ Zip a directory
 def create_zip_file(source_dir, output_filename):
    shutil.make_archive(output_filename, 'zip', source_dir)
 
-# 🏗️ Create a new GitHub repo via API
 def create_repo(g, repo_name):
    user = g.get_user()
    return user.create_repo(repo_name)
 
-# 🚀 Push local repo changes to GitHub
 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)
@@ -315,7 +348,6 @@ def push_to_github(local_path, repo, github_token):
    origin.push(refspec=f'{current_branch}:{current_branch}')
 
 # %% ───────────── FILE & MEDIA MANAGEMENT FUNCTIONS ─────────────
-# 📂 List saved Markdown files in sidebar with actions
 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")
@@ -334,7 +366,6 @@ def display_saved_files_in_sidebar():
                os.remove(file)
                st.rerun()
 
-# 👀 Display file viewer in main area
 def display_file_viewer(file_path):
    content = load_file(file_path)
    if content:
@@ -346,7 +377,6 @@ def display_file_viewer(file_path):
        st.markdown(content)
        st.download_button("⬇️", data=content, file_name=os.path.basename(file_path), mime="text/markdown")
 
-# ✏️ Display file editor (Markdown & Code)
 def display_file_editor(file_path):
    if 'file_content' not in st.session_state:
        st.session_state.file_content = {}
@@ -374,7 +404,6 @@ def display_file_editor(file_path):
    with col2:
        st.download_button("⬇️", data=new_content, file_name=os.path.basename(file_path), mime="text/markdown")
 
-# 💾 Save content to a file (with error handling)
 def save_file_content(file_path, content):
    try:
        with open(file_path, 'w', encoding='utf-8') as file:
@@ -384,7 +413,6 @@ def save_file_content(file_path, content):
        st.error(f"Save error: {str(e)}")
        return False
 
-# 🗂️ Update file management UI section (view, edit, delete)
 def update_file_management_section():
    if 'file_view_mode' not in st.session_state:
        st.session_state.file_view_mode = None
@@ -449,7 +477,6 @@ def update_file_management_section():
            display_file_editor(st.session_state.current_file)
 
 # %% ───────────── VIDEO & AUDIO UI FUNCTIONS ─────────────
-# 🖼️ Validate and preprocess an image for video generation
 def validate_and_preprocess_image(file_data, target_size=(576, 1024)):
    try:
        st.write("Preprocessing image...")
@@ -484,7 +511,6 @@ def validate_and_preprocess_image(file_data, target_size=(576, 1024)):
        st.error(f"Image error: {str(e)}")
        return None
 
-# ▶️ Add video generation UI with Gradio client
 def add_video_generation_ui(container):
    st.markdown("### 🎥 Video Gen")
    col1, col2 = st.columns([2, 1])
@@ -564,7 +590,6 @@ def add_video_generation_ui(container):
            st.error(f"Upload error: {str(e)}")
 
 # %% ───────────── MAIN FUNCTION ─────────────
-# 🚀 Main app entry point
 def main():
    st.markdown("### 🐙 GitCosmos - Cosmos & Git Hub")
    if "chat_history" not in st.session_state:
@@ -666,11 +691,9 @@ def main():
                                            st.error(message)
                                    except Exception as e:
                                        st.error(f"Delete err: {str(e)}")
-                           
-                           # Display delete log persistently
                            if "delete_log" in st.session_state and st.session_state.delete_log:
                                st.subheader("Delete Log")
-                               for log_entry in st.session_state.delete_log[-5:]:  # Show last 5 entries
+                               for log_entry in st.session_state.delete_log[-5:]:
                                    st.write(log_entry)
                    elif selected_view == 'Run AI':
                        st.markdown("#### 🤖 Run AI (stub)")
@@ -753,7 +776,82 @@ def main():
                            st.dataframe(df)
                        else:
                            st.info("No docs.")
-                   # --- End of Document UI ---
+                   
+                   # ───── Advanced Cosmos Ops ─────
+                   with st.sidebar.expander("Advanced Cosmos Ops"):
+                       st.markdown("### Advanced Cosmos Operations")
+                       # Create Container
+                       with st.form("create_container_form"):
+                           container_id = st.text_input("Container ID", value="newContainer")
+                           partition_key_path = st.text_input("Partition Key Path", value="/id")
+                           analytical = st.checkbox("Enable Analytical Store", value=False)
+                           submitted = st.form_submit_button("🆕 Create Container")
+                           if submitted:
+                               analytical_ttl = -1 if analytical else None
+                               new_container = create_new_container(database, container_id, partition_key_path,
+                                                                    analytical_storage_ttl=analytical_ttl)
+                               if new_container:
+                                   st.success(f"Container '{container_id}' created.")
+                               else:
+                                   st.error("Container creation failed.")
+                       
+                       # Insert Item
+                       with st.form("insert_item_form"):
+                           item_json = st.text_area("Item JSON", value='{"id": "itemX", "productName": "Widget", "productModel": "Model X"}')
+                           submitted = st.form_submit_button("➕ Insert Item")
+                           if submitted:
+                               try:
+                                   item = json.loads(item_json)
+                                   success, msg = advanced_insert_item(container, item)
+                                   if success:
+                                       st.success(msg)
+                                   else:
+                                       st.error(msg)
+                               except Exception as e:
+                                   st.error(str(e))
+                       
+                       # Update Item
+                       with st.form("update_item_form"):
+                           update_json = st.text_area("Update Item JSON", value='{"id": "itemX", "productName": "Widget", "productModel": "Updated Model"}')
+                           submitted = st.form_submit_button("✏️ Update Item")
+                           if submitted:
+                               try:
+                                   item = json.loads(update_json)
+                                   success, msg = advanced_update_item(container, item)
+                                   if success:
+                                       st.success(msg)
+                                   else:
+                                       st.error(msg)
+                               except Exception as e:
+                                   st.error(str(e))
+                       
+                       # Delete Item
+                       with st.form("delete_item_form"):
+                           del_item_id = st.text_input("Item ID to delete", value="itemX")
+                           del_partition = st.text_input("Partition Key Value", value="Widget")
+                           submitted = st.form_submit_button("🗑️ Delete Item")
+                           if submitted:
+                               success, msg = advanced_delete_item(container, del_item_id, del_partition)
+                               if success:
+                                   st.success(msg)
+                               else:
+                                   st.error(msg)
+                       
+                       # Vector Search
+                       with st.form("vector_search_form"):
+                           vector_str = st.text_input("Query Vector (comma separated)", value="0.1, 0.2, 0.3, 0.4")
+                           top_n = st.number_input("Top N Results", min_value=1, max_value=100, value=10)
+                           vector_field = st.text_input("Vector Field", value="vector")
+                           submitted = st.form_submit_button("🔎 Vector Search")
+                           if submitted:
+                               try:
+                                   query_vector = [float(x.strip()) for x in vector_str.split(",") if x.strip()]
+                                   results = vector_search(container, query_vector, vector_field, top=top_n)
+                                   st.write(results)
+                               except Exception as e:
+                                   st.error(str(e))
+                       
+                   # ───── End Advanced Ops ─────
                    st.subheader("🐙 GitHub Ops")
                    github_token = os.environ.get("GITHUB")
                    source_repo = st.text_input("Source Repo URL", value="https://github.com/AaronCWacker/AIExamples-8-24-Streamlit")
@@ -795,7 +893,6 @@ def main():
                                        shutil.rmtree(local_path)
                            else:
                                st.error("Missing token or URL 🔑❓")
-                   # --- File Management Section ---
                    update_file_management_section()
        except exceptions.CosmosHttpResponseError as e:
            st.error(f"Cosmos error: {str(e)} 🚨")