Update app.py

app.py

@@ -3,43 +3,77 @@
 # =============================================================================
 import base64
 import glob
+import hashlib
 import json
 import os
 import pandas as pd
 import pytz
+import random
 import re
 import shutil
 import streamlit as st
 import time
+import traceback
 import uuid
 import zipfile
+from PIL import Image
 from azure.cosmos import CosmosClient, PartitionKey, exceptions
 from datetime import datetime
+from git import Repo
+from github import Github
+from gradio_client import Client
+import tempfile
+import io
+import requests
+import numpy as np
+from urllib.parse import quote
+
+# =============================================================================
+# ───────────── EXTERNAL HELP LINKS ─────────────
+# =============================================================================
+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": "💻"},
+    {"title": "CosmosDB Index and Query Vectors", "url": "https://learn.microsoft.com/en-us/azure/cosmos-db/nosql/how-to-python-vector-index-query", "emoji": "💻"},
+    {"title": "CosmosDB NoSQL Materialized Views", "url": "https://learn.microsoft.com/en-us/azure/cosmos-db/nosql/materialized-views", "emoji": "💻"},
+    {"title": "LangChain Vector Store Guide", "url": "https://python.langchain.com/docs/integrations/vectorstores/azure_cosmos_db_no_sql/", "emoji": "💻"},
+    {"title": "Vector Database Prompt Engineering RAG for Python", "url": "https://learn.microsoft.com/en-us/azure/cosmos-db/vector-database?source=recommendations", "emoji": "💻"},
+    {"title": "MergeKit Official GitHub", "url": "https://github.com/arcee-ai/MergeKit", "emoji": "💻"},
+    {"title": "MergeKit Sample Usage", "url": "https://github.com/arcee-ai/MergeKit#examples", "emoji": "📚"},
+    {"title": "DistillKit Official GitHub", "url": "https://github.com/arcee-ai/DistillKit", "emoji": "💻"},
+    {"title": "DistillKit Sample Usage", "url": "https://github.com/arcee-ai/DistillKit#usage", "emoji": "📚"},
+    {"title": "arcee.ai Official Website", "url": "https://arcee.ai", "emoji": "🌐"},
+]
 
 # =============================================================================
 # ───────────── APP CONFIGURATION ─────────────
 # =============================================================================
 Site_Name = '🐙 GitCosmos'
 title = "🐙 GitCosmos"
+helpURL = 'https://huggingface.co/awacke1'
+bugURL = 'https://huggingface.co/spaces/awacke1/AzureCosmosDBUI/'
+icons = '🐙🌌💫'
 st.set_page_config(
     page_title=title,
-    page_icon='🐙🌌💫',
+    page_icon=icons,
     layout="wide",
-    initial_sidebar_state="auto"
+    initial_sidebar_state="auto",
+    menu_items={
+        'Get Help': helpURL,
+        'Report a bug': bugURL,
+        'About': title
+    }
 )
 
-# Cosmos DB Configuration
 ENDPOINT = "https://acae-afd.documents.azure.com:443/"
-Key = os.environ.get("Key")  # Ensure this is set in your environment
+DATABASE_NAME = os.environ.get("COSMOS_DATABASE_NAME")
+CONTAINER_NAME = os.environ.get("COSMOS_CONTAINER_NAME")
+Key = os.environ.get("Key")
+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 ─────────────
 # =============================================================================
-def generate_unique_id():
-    timestamp = datetime.utcnow().strftime('%Y%m%d%H%M%S%f')
-    unique_uuid = str(uuid.uuid4())
-    return f"{timestamp}-{unique_uuid}"
-
 def get_download_link(file_path):
     with open(file_path, "rb") as file:
         contents = file.read()
@@ -47,6 +81,40 @@ 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>'
 
+def generate_unique_id():
+    timestamp = datetime.utcnow().strftime('%Y%m%d%H%M%S%f')
+    unique_uuid = str(uuid.uuid4())
+    return f"{timestamp}-{unique_uuid}"
+
+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}"
+
+def create_file(filename, prompt, response, should_save=True):
+    if should_save:
+        with open(filename, 'w', encoding='utf-8') as file:
+            file.write(prompt + "\n\n" + response)
+
+def load_file(file_name):
+    with open(file_name, "r", encoding='utf-8') as file:
+        return file.read()
+
+def create_zip_of_files(files):
+    zip_name = "all_files.zip"
+    with zipfile.ZipFile(zip_name, 'w') as zipf:
+        for file in files:
+            zipf.write(file)
+    return zip_name
+
+def preprocess_text(text):
+    text = text.replace('\r\n', '\\n').replace('\r', '\\n').replace('\n', '\\n')
+    text = text.replace('"', '\\"')
+    text = re.sub(r'[\t]', ' ', text)
+    text = re.sub(r'[^\x00-\x7F]+', '', text)
+    return text.strip()
+
 def sanitize_json_text(text):
     text = re.sub(r'[\x00-\x08\x0B\x0C\x0E-\x1F]', '', text)
     text = text.replace("\n", "\\n").replace("\r", "\\r").replace("\t", "\\t")
@@ -97,6 +165,27 @@ def delete_record(container, record):
     except Exception as e:
         return False, f"Error: {str(e)} 😱"
 
+def save_to_cosmos_db(container, query, response1, response2):
+    try:
+        timestamp = datetime.utcnow().strftime('%Y%m%d%H%M%S%f')
+        unique_uuid = str(uuid.uuid4())
+        new_id = f"{timestamp}-{unique_uuid}"
+        record = {
+            "id": new_id,
+            "pk": new_id,
+            "name": new_id,
+            "query": query,
+            "response1": response1,
+            "response2": response2,
+            "timestamp": datetime.utcnow().isoformat(),
+            "type": "ai_response",
+            "version": "1.0"
+        }
+        container.create_item(body=record)
+        st.success(f"Saved: {record['id']}")
+    except Exception as e:
+        st.error(f"Save error: {str(e)}")
+
 def archive_current_container(database_name, container_name, client):
     try:
         base_dir = "./cosmos_archive"
@@ -131,6 +220,129 @@ def create_new_container(database, container_id, partition_key_path):
         st.error(f"Error creating container: {str(e)}")
         return None
 
+# =============================================================================
+# ───────────── 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}')
+
+# =============================================================================
+# ───────────── FILE & MEDIA MANAGEMENT FUNCTIONS ─────────────
+# =============================================================================
+def display_file_viewer(file_path):
+    content = load_file(file_path)
+    if content:
+        st.markdown("### 📄 File Viewer")
+        st.markdown(f"**{file_path}**")
+        file_stats = os.stat(file_path)
+        st.markdown(f"**Mod:** {datetime.fromtimestamp(file_stats.st_mtime).strftime('%Y-%m-%d %H:%M:%S')} | **Size:** {file_stats.st_size} bytes")
+        st.markdown("---")
+        st.markdown(content)
+        st.download_button("⬇️", data=content, file_name=os.path.basename(file_path), mime="text/markdown")
+
+def display_file_editor(file_path):
+    if 'file_content' not in st.session_state:
+        st.session_state.file_content = {}
+    if file_path not in st.session_state.file_content:
+        content = load_file(file_path)
+        if content:
+            st.session_state.file_content[file_path] = content
+    st.markdown("### ✏️ Edit File")
+    st.markdown(f"**Editing:** {file_path}")
+    new_content = st.text_area("Edit:", value=st.session_state.file_content.get(file_path, ""), height=400, key=f"editor_{hash(file_path)}")
+    col1, col2 = st.columns([1, 5])
+    with col1:
+        if st.button("💾 Save"):
+            sanitized = sanitize_json_text(new_content)
+            try:
+                json.loads(sanitized)
+                with open(file_path, 'w', encoding='utf-8') as file:
+                    file.write(sanitized)
+                st.session_state.file_content[file_path] = sanitized
+                st.success("Saved! 🎉")
+                time.sleep(1)
+                st.rerun()
+            except Exception as e:
+                st.error(f"Save error: {str(e)}")
+    with col2:
+        st.download_button("⬇️", data=new_content, file_name=os.path.basename(file_path), mime="text/markdown")
+
+def update_file_management_section():
+    if 'file_view_mode' not in st.session_state:
+        st.session_state.file_view_mode = None
+    if 'current_file' not in st.session_state:
+        st.session_state.current_file = None
+    all_files = sorted(glob.glob("*.md"), reverse=True)
+    st.sidebar.subheader("📁 Files")
+    if st.sidebar.button("🗑 Delete All"):
+        for file in all_files:
+            os.remove(file)
+        st.session_state.file_content = {}
+        st.session_state.current_file = None
+        st.session_state.file_view_mode = None
+        st.rerun()
+    if st.sidebar.button("⬇️ Download All"):
+        zip_file = create_zip_of_files(all_files)
+        st.sidebar.markdown(get_download_link(zip_file), unsafe_allow_html=True)
+    for file in all_files:
+        col1, col2, col3, col4 = st.sidebar.columns([1, 3, 1, 1])
+        with col1:
+            if st.button("🌐", key=f"view_{file}"):
+                st.session_state.current_file = file
+                st.session_state.file_view_mode = 'view'
+                st.rerun()
+        with col2:
+            st.markdown(get_download_link(file), unsafe_allow_html=True)
+        with col3:
+            if st.button("📂", key=f"edit_{file}"):
+                st.session_state.current_file = file
+                st.session_state.file_view_mode = 'edit'
+                st.rerun()
+        with col4:
+            if st.button("🗑", key=f"delete_{file}"):
+                os.remove(file)
+                if file in st.session_state.file_content:
+                    del st.session_state.file_content[file]
+                if st.session_state.current_file == file:
+                    st.session_state.current_file = None
+                    st.session_state.file_view_mode = None
+                st.rerun()
+    st.sidebar.subheader("External Help Links")
+    for link in external_links:
+        st.sidebar.markdown(f"{link['emoji']} [{link['title']}]({link['url']})", unsafe_allow_html=True)
+    if st.session_state.current_file:
+        if st.session_state.file_view_mode == 'view':
+            display_file_viewer(st.session_state.current_file)
+        elif st.session_state.file_view_mode == 'edit':
+            display_file_editor(st.session_state.current_file)
+
 # =============================================================================
 # ───────────── UI FUNCTIONS ─────────────
 # =============================================================================
@@ -150,31 +362,33 @@ def edit_all_documents(container):
             if doc_key not in st.session_state:
                 st.session_state[doc_key] = json.dumps(doc, indent=2)
             edited_content = st.text_area("Edit JSON", value=st.session_state[doc_key], height=300, key=doc_key)
-            if st.button("💾 Save", key=f"save_{doc['id']}"):
-                try:
-                    updated_doc = json.loads(sanitize_json_text(edited_content))
-                    # Preserve system fields and identity
-                    updated_doc['id'] = doc['id']
-                    updated_doc['pk'] = doc.get('pk', doc['id'])
-                    for field in ['_ts', '_rid', '_self', '_etag', '_attachments']:
-                        updated_doc.pop(field, None)  # Remove system fields before upsert
-                    success, message = update_record(container, updated_doc)
+            col_save, col_delete = st.columns(2)
+            with col_save:
+                if st.button("💾 Save", key=f"save_{doc['id']}"):
+                    try:
+                        updated_doc = json.loads(sanitize_json_text(edited_content))
+                        updated_doc['id'] = doc['id']
+                        updated_doc['pk'] = doc.get('pk', doc['id'])
+                        for field in ['_ts', '_rid', '_self', '_etag', '_attachments']:
+                            updated_doc.pop(field, None)
+                        success, message = update_record(container, updated_doc)
+                        if success:
+                            st.success(f"Saved {doc['id']}")
+                            st.session_state[doc_key] = json.dumps(updated_doc, indent=2)
+                        else:
+                            st.error(message)
+                    except json.JSONDecodeError:
+                        st.error("Invalid JSON format.")
+                    except Exception as e:
+                        st.error(f"Save error: {str(e)}")
+            with col_delete:
+                if st.button("🗑️ Delete", key=f"delete_{doc['id']}"):
+                    success, message = delete_record(container, doc)
                     if success:
-                        st.success(f"Saved {doc['id']}")
-                        st.session_state[doc_key] = json.dumps(updated_doc, indent=2)
+                        st.success(message)
+                        st.rerun()
                     else:
                         st.error(message)
-                except json.JSONDecodeError:
-                    st.error("Invalid JSON format.")
-                except Exception as e:
-                    st.error(f"Save error: {str(e)}")
-            if st.button("🗑️ Delete", key=f"delete_{doc['id']}"):
-                success, message = delete_record(container, doc)
-                if success:
-                    st.success(message)
-                    st.rerun()
-                else:
-                    st.error(message)
 
 def new_item_default(container):
     new_id = generate_unique_id()
@@ -193,37 +407,228 @@ def new_item_default(container):
     else:
         st.error(f"Error creating new item: {message}")
 
+def add_field_to_doc():
+    key = st.session_state.new_field_key
+    value = st.session_state.new_field_value
+    try:
+        doc = json.loads(st.session_state.doc_editor)
+        doc[key] = value
+        st.session_state.doc_editor = json.dumps(doc, indent=2)
+        container = st.session_state.current_container
+        success, message = update_record(container, doc)
+        if success:
+            st.success(f"Added field {key} 👍")
+        else:
+            st.error(message)
+    except Exception as e:
+        st.error(f"Error adding field: {str(e)}")
+
+def new_ai_record(container):
+    new_id = generate_unique_id()
+    default_doc = {
+        "id": new_id,
+        "pk": new_id,
+        "name": "AI Modality Record",
+        "function_url": "https://example.com/function",
+        "input_text": "### Input (markdown)\n\nType your input here.",
+        "output_text": "### Output (markdown)\n\nResult will appear here.",
+        "timestamp": datetime.now().isoformat(),
+        "type": "ai_modality"
+    }
+    success, message = insert_record(container, default_doc)
+    if success:
+        st.success("New AI modality record created! 💡")
+        st.rerun()
+    else:
+        st.error(f"Error creating AI record: {message}")
+
+def new_links_record(container):
+    new_id = generate_unique_id()
+    links_md = "\n".join([f"- {link['emoji']} [{link['title']}]({link['url']})" for link in external_links])
+    default_doc = {
+        "id": new_id,
+        "pk": new_id,
+        "name": "Portal Links Record",
+        "function_url": "",
+        "input_text": links_md,
+        "output_text": "",
+        "timestamp": datetime.now().isoformat(),
+        "type": "ai_modality"
+    }
+    success, message = insert_record(container, default_doc)
+    if success:
+        st.success("New Portal Links record created! 🔗")
+        st.rerun()
+    else:
+        st.error(f"Error creating links record: {message}")
+
+def vector_keyword_search(keyword, container):
+    try:
+        query = f"SELECT * FROM c WHERE CONTAINS(c.content, '{keyword}')"
+        results = list(container.query_items(query=query, enable_cross_partition_query=True))
+        return results
+    except Exception as e:
+        st.error(f"Search error: {str(e)}")
+        return []
+
+def display_search_results(keyword, container):
+    results = vector_keyword_search(keyword, container)
+    st.markdown("### 🔍 Search Results")
+    for res in results:
+        doc_id = res.get("id", "")
+        with st.expander(f"Result {doc_id}"):
+            edited = st.text_area("Edit Document", value=json.dumps(res, indent=2), key=f"search_{doc_id}")
+            if st.button(f"💾 Save changes", key=f"save_search_{doc_id}"):
+                try:
+                    updated_doc = json.loads(edited)
+                    success, message = update_record(container, updated_doc)
+                    if success:
+                        st.success(f"Updated {doc_id}!")
+                        st.rerun()
+                    else:
+                        st.error(message)
+                except Exception as e:
+                    st.error(f"Error saving: {str(e)}")
+
+def search_documents_ui(container):
+    with st.sidebar.form("search_form"):
+        keyword = st.text_input("Search Keyword", key="search_keyword")
+        if st.form_submit_button("🔍 Search") and keyword:
+            display_search_results(keyword, container)
+
+def validate_and_preprocess_image(file_data, target_size=(576, 1024)):
+    try:
+        if isinstance(file_data, bytes):
+            img = Image.open(io.BytesIO(file_data))
+        elif hasattr(file_data, 'read'):
+            if hasattr(file_data, 'seek'):
+                file_data.seek(0)
+            img = Image.open(file_data)
+        else:
+            raise ValueError(f"Unsupported input: {type(file_data)}")
+        if img.mode != 'RGB':
+            img = img.convert('RGB')
+        aspect_ratio = img.size[0] / img.size[1]
+        if aspect_ratio > target_size[0] / target_size[1]:
+            new_width = target_size[0]
+            new_height = int(new_width / aspect_ratio)
+        else:
+            new_height = target_size[1]
+            new_width = int(new_height * aspect_ratio)
+        new_width = (new_width // 2) * 2
+        new_height = (new_height // 2) * 2
+        resized_img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
+        final_img = Image.new('RGB', target_size, (255, 255, 255))
+        paste_x = (target_size[0] - new_width) // 2
+        paste_y = (target_size[1] - new_height) // 2
+        final_img.paste(resized_img, (paste_x, paste_y))
+        return final_img
+    except Exception as e:
+        st.error(f"Image error: {str(e)}")
+        return None
+
+def add_video_generation_ui(container):
+    st.markdown("### 🎥 Video Generation")
+    col1, col2 = st.columns([2, 1])
+    with col1:
+        uploaded_file = st.file_uploader("Upload Image 🖼️", type=['png', 'jpg', 'jpeg'])
+    with col2:
+        st.markdown("#### Parameters")
+        motion = st.slider("🌊 Motion", 1, 255, 127)
+        fps = st.slider("🎬 FPS", 1, 30, 6)
+        with st.expander("Advanced"):
+            use_custom = st.checkbox("Custom Seed")
+            seed = st.number_input("Seed", value=int(time.time() * 1000)) if use_custom else None
+    if uploaded_file:
+        file_data = uploaded_file.read()
+        preview1, preview2 = st.columns(2)
+        with preview1:
+            st.write("Original")
+            st.image(Image.open(io.BytesIO(file_data)), use_column_width=True)
+        with preview2:
+            proc_img = validate_and_preprocess_image(io.BytesIO(file_data))
+            if proc_img:
+                st.write("Processed")
+                st.image(proc_img, use_column_width=True)
+            else:
+                return
+        if st.button("🎥 Generate"):
+            with st.spinner("Generating video..."):
+                with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as temp_file:
+                    proc_img.save(temp_file.name, format='PNG')
+                    try:
+                        client = Client("awacke1/stable-video-diffusion", hf_token=os.environ.get("HUGGINGFACE_TOKEN"))
+                        result = client.predict(
+                            image=temp_file.name,
+                            seed=seed if seed is not None else int(time.time() * 1000),
+                            randomize_seed=seed is None,
+                            motion_bucket_id=motion,
+                            fps_id=fps,
+                            api_name="/video"
+                        )
+                        video_path = result[0].get('video') if isinstance(result[0], dict) else None
+                        if video_path and os.path.exists(video_path):
+                            video_filename = f"generated_video_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
+                            shutil.copy(video_path, video_filename)
+                            st.success(f"Video generated! 🎉")
+                            st.video(video_filename)
+                            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",
+                                    "motion": motion,
+                                    "fps": fps,
+                                    "timestamp": datetime.now().isoformat()
+                                }
+                                success, message = insert_record(container, video_record)
+                                if success:
+                                    st.success("DB record saved!")
+                                else:
+                                    st.error(f"DB error: {message}")
+                    except Exception as e:
+                        st.error(f"Video gen error: {str(e)}")
+                    finally:
+                        os.unlink(temp_file.name)
+
 # =============================================================================
 # ───────────── MAIN FUNCTION ─────────────
 # =============================================================================
 def main():
-    st.title("🐙 GitCosmos - CosmosDB Manager")
-
-    # Initialize session state
-    if "client" not in st.session_state:
-        if not Key:
-            st.error("Missing CosmosDB Key 🔑❌")
-            return
-        st.session_state.client = CosmosClient(ENDPOINT, credential=Key)
-        st.session_state.selected_database = None
-        st.session_state.selected_container = None
+    st.markdown("### 🐙 GitCosmos - Cosmos & Git Hub")
+    st.markdown(f"[🔗 Portal]({CosmosDBUrl})")
+    
+    if "chat_history" not in st.session_state:
+        st.session_state.chat_history = []
+    if "current_container" not in st.session_state:
+        st.session_state.current_container = None
+    if not Key:
+        st.error("Missing Cosmos Key 🔑❌")
+        return
+    st.session_state.primary_key = Key
+    st.session_state.logged_in = True
 
     # Sidebar: Hierarchical Navigation
     st.sidebar.title("🐙 Navigator")
-
+    
     # Databases Section
-    st.sidebar.subheader("Databases")
+    st.sidebar.subheader("🗃️ Databases")
+    if "client" not in st.session_state:
+        st.session_state.client = CosmosClient(ENDPOINT, credential=Key)
     databases = get_databases(st.session_state.client)
-    selected_db = st.sidebar.selectbox("🗃️ Select Database", databases, key="db_select")
-    if selected_db != st.session_state.selected_database:
+    selected_db = st.sidebar.selectbox("Select Database", databases, key="db_select")
+    if selected_db != st.session_state.get("selected_database"):
         st.session_state.selected_database = selected_db
         st.session_state.selected_container = None
+        st.session_state.current_container = None
         st.rerun()
 
     # Containers Section
     if st.session_state.selected_database:
         database = st.session_state.client.get_database_client(st.session_state.selected_database)
-        st.sidebar.subheader("Containers")
+        st.sidebar.subheader("📁 Containers")
         if st.sidebar.button("🆕 New Container"):
             with st.sidebar.form("new_container_form"):
                 container_id = st.text_input("Container ID", "new-container")
@@ -234,33 +639,52 @@ def main():
                         st.success(f"Container '{container_id}' created!")
                         st.rerun()
         containers = get_containers(database)
-        selected_container = st.sidebar.selectbox("📁 Select Container", containers, key="container_select")
-        if selected_container != st.session_state.selected_container:
+        selected_container = st.sidebar.selectbox("Select Container", containers, key="container_select")
+        if selected_container != st.session_state.get("selected_container"):
             st.session_state.selected_container = selected_container
+            st.session_state.current_container = database.get_container_client(selected_container)
             st.rerun()
 
         # Actions Section
-        st.sidebar.subheader("Actions")
-        if st.session_state.selected_container and st.sidebar.button("📦 Export Container"):
-            download_link = archive_current_container(st.session_state.selected_database, st.session_state.selected_container, st.session_state.client)
-            if download_link.startswith('<a'):
-                st.sidebar.markdown(download_link, unsafe_allow_html=True)
-            else:
-                st.sidebar.error(download_link)
+        st.sidebar.subheader("⚙️ Actions")
+        if st.session_state.current_container:
+            if st.sidebar.button("📦 Export Container"):
+                download_link = archive_current_container(st.session_state.selected_database, st.session_state.selected_container, st.session_state.client)
+                st.sidebar.markdown(download_link, unsafe_allow_html=True) if download_link.startswith('<a') else st.sidebar.error(download_link)
 
         # Items Section
-        st.sidebar.subheader("Items")
-        if st.session_state.selected_container:
+        st.sidebar.subheader("📑 Items")
+        if st.session_state.current_container:
             if st.sidebar.button("➕ New Item"):
-                container = database.get_container_client(st.session_state.selected_container)
-                new_item_default(container)
+                new_item_default(st.session_state.current_container)
+            st.sidebar.text_input("New Field Key", key="new_field_key")
+            st.sidebar.text_input("New Field Value", key="new_field_value")
+            if st.sidebar.button("➕ Add Field") and "doc_editor" in st.session_state:
+                add_field_to_doc()
+            if st.sidebar.button("🤖 New AI Record"):
+                new_ai_record(st.session_state.current_container)
+            if st.sidebar.button("🔗 New Links Record"):
+                new_links_record(st.session_state.current_container)
+            search_documents_ui(st.session_state.current_container)
 
-    # Central Area: Display and Edit All Documents
-    if st.session_state.selected_container:
-        container = database.get_container_client(st.session_state.selected_container)
-        edit_all_documents(container)
+    # Central Area: Editable Documents
+    if st.session_state.current_container:
+        edit_all_documents(st.session_state.current_container)
     else:
         st.info("Select a database and container to view and edit documents.")
 
+    # Additional Features
+    update_file_management_section()
+    add_video_generation_ui(st.session_state.current_container if st.session_state.current_container else None)
+
+    # Logout
+    if st.session_state.logged_in and st.sidebar.button("🚪 Logout"):
+        st.session_state.logged_in = False
+        st.session_state.client = None
+        st.session_state.selected_database = None
+        st.session_state.selected_container = None
+        st.session_state.current_container = None
+        st.rerun()
+
 if __name__ == "__main__":
     main()