Update app.py

app.py

@@ -588,326 +588,89 @@ def add_video_generation_ui(container):
                                st.warning(f"Cleanup error: {str(e)}")
        except Exception as e:
            st.error(f"Upload error: {str(e)}")
-
-# %% ───────────── MAIN FUNCTION ─────────────
 def main():
-   st.markdown("### 🐙 GitCosmos - Cosmos & Git Hub")
-   if "chat_history" not in st.session_state:
-       st.session_state.chat_history = []
-   # Auth & Cosmos client initialization
-   if Key:
-       st.session_state.primary_key = Key
-       st.session_state.logged_in = True
-   else:
-       st.error("Missing Cosmos Key 🔑❌")
-       return
-   if st.session_state.logged_in:
-       try:
-           if st.session_state.get("client") is None:
-               st.session_state.client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key)
-           st.sidebar.title("🐙 Navigator")
-           databases = get_databases(st.session_state.client)
-           selected_db = st.sidebar.selectbox("🗃️ DB", databases)
-           st.markdown(CosmosDBUrl)
-           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.selected_document_id = None
-               st.session_state.current_index = 0
-               st.rerun()
-           if st.session_state.selected_database:
-               database = st.session_state.client.get_database_client(st.session_state.selected_database)
-               containers = get_containers(database)
-               selected_container = st.sidebar.selectbox("📁 Container", containers)
-               if selected_container != st.session_state.get("selected_container"):
-                   st.session_state.selected_container = selected_container
-                   st.session_state.selected_document_id = None
-                   st.session_state.current_index = 0
-                   st.rerun()
-               if st.session_state.selected_container:
-                   container = database.get_container_client(st.session_state.selected_container)
-                   if st.sidebar.button("📦 Export"):
-                       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.markdown(download_link, unsafe_allow_html=True)
-                       else:
-                           st.error(download_link)
-                   documents = get_documents(container)
-                   total_docs = len(documents)
-                   num_docs = st.slider("Docs", 1, 20, 1)
-                   documents_to_display = documents[:num_docs] if total_docs > num_docs else documents
-                   st.sidebar.info(f"Showing {len(documents_to_display)} docs")
-                   # --- Document Viewer / Editor ---
-                   view_options = ['Markdown', 'Code', 'Run AI', 'Clone', 'New']
-                   selected_view = st.sidebar.selectbox("View", view_options, index=1)
-                   if selected_view == 'Markdown':
-                       st.markdown("#### 📄 Markdown")
-                       if documents:
-                           doc = documents[st.session_state.current_index]
-                           content = json.dumps(doc, indent=2)
-                           st.markdown(f"```json\n{content}\n```")
-                           col_prev, col_next = st.columns(2)
-                           with col_prev:
-                               if st.button("⬅️") and st.session_state.current_index > 0:
-                                   st.session_state.current_index -= 1
-                                   st.rerun()
-                           with col_next:
-                               if st.button("➡️") and st.session_state.current_index < total_docs - 1:
-                                   st.session_state.current_index += 1
-                                   st.rerun()
-                   elif selected_view == 'Code':
-                       st.markdown("#### 💻 Code Editor")
-                       if documents:
-                           doc = documents[st.session_state.current_index]
-                           doc_str = st.text_area("Edit JSON", value=json.dumps(doc, indent=2), height=300, key=f'code_{st.session_state.current_index}')
-                           col_prev, col_next = st.columns(2)
-                           with col_prev:
-                               if st.button("⬅️") and st.session_state.current_index > 0:
-                                   st.session_state.current_index -= 1
-                                   st.rerun()
-                           with col_next:
-                               if st.button("➡️") and st.session_state.current_index < total_docs - 1:
-                                   st.session_state.current_index += 1
-                                   st.rerun()
-                           col_save, col_delete = st.columns(2)
-                           with col_save:
-                               if st.button("💾 Save", key=f'save_{st.session_state.current_index}'):
-                                   try:
-                                       updated_doc = json.loads(doc_str)
-                                       container.upsert_item(body=updated_doc)
-                                       st.success(f"Saved {updated_doc['id']}")
-                                       st.rerun()
-                                   except Exception as e:
-                                       st.error(f"Save err: {str(e)}")
-                           with col_delete:
-                               if st.button("🗑️ Delete", key=f'delete_{st.session_state.current_index}'):
-                                   try:
-                                       current_doc = json.loads(doc_str)
-                                       success, message = delete_record(container, current_doc)
-                                       if success:
-                                           st.success(message)
-                                           st.rerun()
-                                       else:
-                                           st.error(message)
-                                   except Exception as e:
-                                       st.error(f"Delete err: {str(e)}")
-                           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:]:
-                                   st.write(log_entry)
-                   elif selected_view == 'Run AI':
-                       st.markdown("#### 🤖 Run AI (stub)")
-                       st.info("AI functionality not implemented.")
-                   elif selected_view == 'Clone':
-                       st.markdown("#### 📄 Clone")
-                       if documents:
-                           doc = documents[st.session_state.current_index]
-                           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']}}
-                           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:
-                               if st.button("🔄 Regenerate"):
-                                   new_id = generate_unique_id()
-                                   st.session_state.new_clone_id = new_id
-                                   st.rerun()
-                           with col2:
-                               if st.button("💾 Save Clone"):
-                                   try:
-                                       final_doc = json.loads(doc_str)
-                                       for field in ['_rid', '_self', '_etag', '_attachments', '_ts']:
-                                           final_doc.pop(field, None)
-                                       container.create_item(body=final_doc)
-                                       st.success(f"Cloned {final_doc['id']}")
-                                       st.rerun()
-                                   except Exception as e:
-                                       st.error(f"Clone err: {str(e)}")
-                           col_prev, col_next = st.columns(2)
-                           with col_prev:
-                               if st.button("⬅️") and st.session_state.current_index > 0:
-                                   st.session_state.current_index -= 1
-                                   st.rerun()
-                           with col_next:
-                               if st.button("➡️") and st.session_state.current_index < total_docs - 1:
-                                   st.session_state.current_index += 1
-                                   st.rerun()
-                   elif selected_view == 'New':
-                       st.markdown("#### ➕ New Doc")
-                       if st.button("🤖 Auto-Gen"):
-                           auto_doc = {
-                               "id": generate_unique_id(),
-                               "name": f"Auto {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
-                               "content": "Auto-generated record.",
-                               "timestamp": datetime.now().isoformat()
-                           }
-                           success, message = insert_record(container, auto_doc)
-                           if success:
-                               st.success(message)
-                               st.rerun()
-                           else:
-                               st.error(message)
-                       else:
-                           new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
-                           default_doc = {
-                               "id": new_id,
-                               "name": "New Doc",
-                               "content": "",
-                               "timestamp": datetime.now().isoformat()
-                           }
-                           new_doc_str = st.text_area("JSON", value=json.dumps(default_doc, indent=2), height=300)
-                           if st.button("➕ Create"):
-                               try:
-                                   cleaned = preprocess_text(new_doc_str)
-                                   new_doc = json.loads(cleaned)
-                                   new_doc['id'] = new_id
-                                   success, message = insert_record(container, new_doc)
-                                   if success:
-                                       st.success(f"Created {new_doc['id']}")
-                                       st.rerun()
-                                   else:
-                                       st.error(message)
-                               except Exception as e:
-                                   st.error(f"Create err: {str(e)}")
-                       st.subheader(f"📊 {st.session_state.selected_container}")
-                       if documents_to_display:
-                           df = pd.DataFrame(documents_to_display)
-                           st.dataframe(df)
-                       else:
-                           st.info("No docs.")
-                   
-                   # ───── 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")
-                   new_repo_name = st.text_input("New Repo Name", value=f"Clone-{datetime.now().strftime('%Y%m%d_%H%M%S')}")
-                   col_g1, col_g2 = st.columns(2)
-                   with col_g1:
-                       if st.button("📥 Clone Repo"):
-                           if github_token and source_repo:
-                               try:
-                                   local_path = f"./temp_repo_{datetime.now().strftime('%Y%m%d%H%M%S')}"
-                                   download_github_repo(source_repo, local_path)
-                                   zip_filename = f"{new_repo_name}.zip"
-                                   create_zip_file(local_path, zip_filename[:-4])
-                                   st.markdown(get_download_link(zip_filename), unsafe_allow_html=True)
-                                   st.success("Cloned! 🎉")
-                               except Exception as e:
-                                   st.error(f"Clone err: {str(e)}")
-                               finally:
-                                   if os.path.exists(local_path):
-                                       shutil.rmtree(local_path)
-                                   if os.path.exists(zip_filename):
-                                       os.remove(zip_filename)
-                           else:
-                               st.error("Missing token or URL 🔑❓")
-                   with col_g2:
-                       if st.button("📤 Push Repo"):
-                           if github_token and source_repo:
-                               try:
-                                   g = Github(github_token)
-                                   new_repo = create_repo(g, new_repo_name)
-                                   local_path = f"./temp_repo_{datetime.now().strftime('%Y%m%d%H%M%S')}"
-                                   download_github_repo(source_repo, local_path)
-                                   push_to_github(local_path, new_repo, github_token)
-                                   st.success(f"Pushed to {new_repo.html_url} 🚀")
-                               except Exception as e:
-                                   st.error(f"Push err: {str(e)}")
-                               finally:
-                                   if os.path.exists(local_path):
-                                       shutil.rmtree(local_path)
-                           else:
-                               st.error("Missing token or URL 🔑❓")
-                   update_file_management_section()
-       except exceptions.CosmosHttpResponseError as e:
-           st.error(f"Cosmos error: {str(e)} 🚨")
-       except Exception as e:
-           st.error(f"Error: {str(e)} 😱")
-   if st.session_state.logged_in and st.sidebar.button("🚪 Logout"):
-       st.markdown("#### 🚪 Logout")
-       st.session_state.logged_in = False
-       st.session_state.selected_records = []
-       st.session_state.client = None
-       st.session_state.selected_database = None
-       st.session_state.selected_container = None
-       st.session_state.selected_document_id = None
-       st.session_state.current_index = 0
-       st.rerun()
+    st.markdown("### 🐙 GitCosmos - Cosmos & Git Hub")
+    if "chat_history" not in st.session_state:
+        st.session_state.chat_history = []
+    # Auth & Cosmos client initialization
+    if Key:
+        st.session_state.primary_key = Key
+        st.session_state.logged_in = True
+    else:
+        st.error("Missing Cosmos Key 🔑❌")
+        return
+    if st.session_state.logged_in:
+        try:
+            if st.session_state.get("client") is None:
+                st.session_state.client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key)
+            st.sidebar.title("🐙 Navigator")
+            databases = get_databases(st.session_state.client)
+            selected_db = st.sidebar.selectbox("🗃️ DB", databases)
+            st.markdown(CosmosDBUrl)
+            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.selected_document_id = None
+                st.session_state.current_index = 0
+                st.rerun()
+            if st.session_state.selected_database:
+                database = st.session_state.client.get_database_client(st.session_state.selected_database)
+                
+                # ── New Container Button & Form Under DB Menu ──
+                if "show_new_container_form" not in st.session_state:
+                    st.session_state.show_new_container_form = False
+                if st.sidebar.button("🆕 New Container"):
+                    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)
+                        submitted = st.form_submit_button("Create Container")
+                        if submitted:
+                            analytical_ttl = -1 if new_analytical else None
+                            new_container = create_new_container(
+                                database,
+                                new_container_id,
+                                new_partition_key,
+                                analytical_storage_ttl=analytical_ttl
+                            )
+                            if new_container:
+                                st.success(f"Container '{new_container_id}' created.")
+                                st.session_state.show_new_container_form = False
+                                st.session_state.new_container_created = new_container_id
+                                st.rerun()
+                
+                # ── Get and Update Container List ──
+                containers = get_containers(database)
+                if "new_container_created" in st.session_state and st.session_state.new_container_created not in containers:
+                    containers.append(st.session_state.new_container_created)
+                selected_container = st.sidebar.selectbox("📁 Container", containers)
+                if selected_container != st.session_state.get("selected_container"):
+                    st.session_state.selected_container = selected_container
+                    st.session_state.selected_document_id = None
+                    st.session_state.current_index = 0
+                    st.rerun()
+                
+                if st.session_state.selected_container:
+                    container = database.get_container_client(st.session_state.selected_container)
+                    # ... rest of your container operations code follows here ...
+                    
+        except exceptions.CosmosHttpResponseError as e:
+            st.error(f"Cosmos error: {str(e)} 🚨")
+        except Exception as e:
+            st.error(f"Error: {str(e)} 😱")
+    if st.session_state.logged_in and st.sidebar.button("🚪 Logout"):
+        st.markdown("#### 🚪 Logout")
+        st.session_state.logged_in = False
+        st.session_state.selected_records = []
+        st.session_state.client = None
+        st.session_state.selected_database = None
+        st.session_state.selected_container = None
+        st.session_state.selected_document_id = None
+        st.session_state.current_index = 0
+        st.rerun()
 
 if __name__ == "__main__":
-   main()
+    main()
+