Update app.py

app.py

@@ -499,15 +499,16 @@ def display_saved_files_in_sidebar():
                 st.rerun()
 
 
-                
-
 
 
-# 🎭 Main function - Where the magic happens (and occasionally breaks)
+                
+# 🎭 Main function - "All the world's a stage, and all the code merely players" -Shakespeare, probably
 def main():
     st.title("🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent")
 
-    # 🚦 Initialize session state
+
+    
+    # 🎲 Session state vars - "Life is like a session state, you never know what you're gonna get"
     if 'logged_in' not in st.session_state:
         st.session_state.logged_in = False
     if 'selected_records' not in st.session_state:
@@ -525,19 +526,18 @@ def main():
     if 'cloned_doc' not in st.session_state:
         st.session_state.cloned_doc = None
 
-    # ⚙️ q= Run ArXiv search from query parameters
+
+    
+    # 🔍 Query processing - "To search or not to search, that is the query" 
     try:
         query_params = st.query_params
         query = query_params.get('q') or query_params.get('query') or ''
         if query:
-            # 🕵️‍♂️ We have a query! Let's process it!
             result, result2, result3, response2 = search_glossary(query)
-            # When saving results, pass the container
+            
+            # 💾 Save results - "Every file you save is a future you pave"
             try:
-                # 📝 Automatically save the AI-generated content
                 if st.button("Save AI Output"):
-
-                    # 🕵️‍♂️ Save our nice research summaries for later sleuthing
                     filename = create_file_from_content(result)
                     st.success(f"File saved: {filename}")                    
                     filename = create_file_from_content(result2)
@@ -547,14 +547,11 @@ def main():
                     filename = create_file_from_content(response2)
                     st.success(f"File saved: {filename}")
                     
-                    #st.rerun()  # Rerun the app to update the sidebar
-            
-                # 📁 Display saved markdown files in the sidebar
                 display_saved_files_in_sidebar()
             except Exception as e:
                 st.error(f"An unexpected error occurred: {str(e)} 😱")
 
-                
+            # 🌟 Cosmos DB operations - "In Cosmos DB we trust, but we still handle errors we must"
             try:
                 save_to_cosmos_db(st.session_state.cosmos_container, query, result, result)
                 save_to_cosmos_db(st.session_state.cosmos_container, query, result2, result2)
@@ -566,42 +563,52 @@ def main():
             except Exception as e:
                 st.error(f"An unexpected error occurred: {str(e)} 😱")
 
-            st.stop()  # Stop further execution
+            st.stop()
     except Exception as e:
         st.markdown(' ')
 
-    # 🔐 Automatic Login
+
+    
+    # 🔐 Auth check - "With great keys come great connectivity"
     if Key:
         st.session_state.primary_key = Key
         st.session_state.logged_in = True
     else:
         st.error("Cosmos DB Key is not set in environment variables. 🔑❌")
-        return  # Can't proceed without a key
+        return
+
 
+    
     if st.session_state.logged_in:
-        # 🌌 Initialize Cosmos DB client
+        # 🌌 DB initialization - "In the beginning, there was connection string..."
         try:
             if st.session_state.client is None:
                 st.session_state.client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key)
             
-            # 🗄️ Sidebar for database, container, and document selection
+            # 📚 Navigation setup - "Navigation is not about where you are, but where you're going"
             st.sidebar.title("🐙Git🌌Cosmos💫🗄️Navigator")
             
             databases = get_databases(st.session_state.client)
             selected_db = st.sidebar.selectbox("🗃️ Select Database", databases)
+
+
             
+            # 🔄 State management - "Change is the only constant in state management"
             if selected_db != st.session_state.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("📁 Select Container", containers)
                 
+                # 🔄 Container state handling - "Container changes, state arranges"
                 if selected_container != st.session_state.selected_container:
                     st.session_state.selected_container = selected_container
                     st.session_state.selected_document_id = None
@@ -611,15 +618,19 @@ def main():
                 if st.session_state.selected_container:
                     container = database.get_container_client(st.session_state.selected_container)
                     
-                    # 📦 Add Export button
+                    # 📦 Export functionality - "Pack it, zip it, ship it"
                     if st.button("📦 Export Container Data"):
-                        download_link = archive_current_container(st.session_state.selected_database, st.session_state.selected_container, st.session_state.client)
+                        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)
+
+
                     
-                    # Fetch documents
+                    # 📝 Document handling - "Document, document, on the wall, who's the most recent of them all?"
                     documents = get_documents(container)
                     total_docs = len(documents)
                     
@@ -631,17 +642,18 @@ def main():
                         st.info(f"Showing all {len(documents_to_display)} documents.")
                     
                     if documents_to_display:
-                        # 🎨 Add Viewer/Editor selection
-                        view_options = ['Show as Markdown', 'Show as Code Editor', 'Show as Edit and Save', 'Clone Document', 'New Record']
+                        # 🎨 View options - "Different strokes for different folks"
+                        view_options = ['Show as Markdown', 'Show as Code Editor', 
+                                      'Show as Edit and Save', 'Clone Document', 'New Record']
                         selected_view = st.selectbox("Select Viewer/Editor", view_options, index=2)
                         
+                        # 📄 Markdown view - "Mark it down, mark it up"
                         if selected_view == 'Show as Markdown':
-                            # 🖌️ Show each record as Markdown with navigation
                             total_docs = len(documents)
                             doc = documents[st.session_state.current_index]
                             st.markdown(f"#### Document ID: {doc.get('id', '')}")
 
-                            # 🕵️‍♂️ Let's extract values from the JSON that have at least one space
+                            # 🕵️ Value extraction - "Finding spaces in all the right places"
                             values_with_space = []
                             def extract_values(obj):
                                 if isinstance(obj, dict):
@@ -655,17 +667,16 @@ def main():
                                         values_with_space.append(obj)
 
                             extract_values(doc)
-
-                            # 🔗 Let's create a list of links for these values
                             st.markdown("#### 🔗 Links for Extracted Texts")
                             for term in values_with_space:
                                 display_glossary_entity(term)
 
-                            # Show the document content as markdown
                             content = json.dumps(doc, indent=2)
                             st.markdown(f"```json\n{content}\n```")
 
-                            # Navigation buttons
+
+                            
+                            # ⬅️➡️ Navigation - "Left and right, day and night"
                             col_prev, col_next = st.columns([1, 1])
                             with col_prev:
                                 if st.button("⬅️ Previous", key='prev_markdown'):
@@ -678,12 +689,21 @@ def main():
                                         st.session_state.current_index += 1
                                         st.rerun()
 
+
+                        
+                        # 💻 Code editor view - "Code is poetry, bugs are typos"
                         elif selected_view == 'Show as Code Editor':
-                            # 💻 Show each record in a code editor with navigation
                             total_docs = len(documents)
                             doc = documents[st.session_state.current_index]
                             st.markdown(f"#### Document ID: {doc.get('id', '')}")
-                            doc_str = st.text_area("Edit Document", value=json.dumps(doc, indent=2), height=300, key=f'code_editor_{st.session_state.current_index}')
+                            doc_str = st.text_area("Edit Document", 
+                                                 value=json.dumps(doc, indent=2), 
+                                                 height=300, 
+                                                 key=f'code_editor_{st.session_state.current_index}')
+
+
+                            
+                            # ⬅️➡️ Navigation and save controls
                             col_prev, col_next = st.columns([1, 1])
                             with col_prev:
                                 if st.button("⬅️ Previous", key='prev_code'):
@@ -695,6 +715,10 @@ def main():
                                     if st.session_state.current_index < total_docs - 1:
                                         st.session_state.current_index += 1
                                         st.rerun()
+
+
+                            
+                            # 💾 Save functionality - "Save early, save often"
                             if st.button("💾 Save Changes", key=f'save_button_{st.session_state.current_index}'):
                                 try:
                                     updated_doc = json.loads(doc_str)
@@ -707,31 +731,39 @@ def main():
                                         st.error(message)
                                 except json.JSONDecodeError as e:
                                     st.error(f"Invalid JSON: {str(e)} 🚫")
+
+
                                     
+                        # ✏️ Edit and save view - "Edit with wisdom, save with precision"
                         elif selected_view == 'Show as Edit and Save':
-                            # ✏️ Show as Edit and Save in columns
                             st.markdown("#### Edit the document fields below:")
                             
-                            # Create columns for each document
                             num_cols = len(documents_to_display)
                             cols = st.columns(num_cols)
 
+
+                            
                             for idx, (col, doc) in enumerate(zip(cols, documents_to_display)):
                                 with col:
                                     st.markdown(f"##### Document ID: {doc.get('id', '')}")
-                                    editable_id = st.text_input("ID", value=doc.get('id', ''), key=f'edit_id_{idx}')
-                                    # Remove 'id' from the document for editing other fields
+                                    editable_id = st.text_input("ID", value=doc.get('id', ''), 
+                                                              key=f'edit_id_{idx}')
                                     editable_doc = doc.copy()
                                     editable_doc.pop('id', None)
-                                    doc_str = st.text_area("Document Content (in JSON format)", value=json.dumps(editable_doc, indent=2), height=300, key=f'doc_str_{idx}')
+                                    doc_str = st.text_area("Document Content (in JSON format)", 
+                                                         value=json.dumps(editable_doc, indent=2), 
+                                                         height=300, 
+                                                         key=f'doc_str_{idx}')
+
+
                                     
-                                    # Add the "Run With AI" button next to "Save Changes"
+                                    # 💾🤖 Save and AI operations
                                     col_save, col_ai = st.columns(2)
                                     with col_save:
                                         if st.button("💾 Save Changes", key=f'save_button_{idx}'):
                                             try:
                                                 updated_doc = json.loads(doc_str)
-                                                updated_doc['id'] = editable_id  # Include the possibly edited ID
+                                                updated_doc['id'] = editable_id
                                                 success, message = update_record(container, updated_doc)
                                                 if success:
                                                     st.success(f"Document {updated_doc['id']} saved successfully.")
@@ -741,28 +773,30 @@ def main():
                                                     st.error(message)
                                             except json.JSONDecodeError as e:
                                                 st.error(f"Invalid JSON: {str(e)} 🚫")
+
+                                                
                                     with col_ai:
                                         if st.button("🤖 Run With AI", key=f'run_with_ai_button_{idx}'):
-                                            # Use the entire document as input
                                             search_glossary(json.dumps(editable_doc, indent=2))
-                                            # Usage in your Streamlit app
                                             if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
                                                 with st.spinner("Cloning document..."):
-                                                    cloned_id = save_or_clone_to_cosmos_db(container, clone_id=doc['id'])
+                                                    cloned_id = save_or_clone_to_cosmos_db(container, 
+                                                                                         clone_id=doc['id'])
                                                     if cloned_id:
                                                         st.success(f"Document cloned successfully with new ID: {cloned_id}")
                                                         st.rerun()
                                                     else:
                                                         st.error("Failed to clone document. Please try again.")
 
+                                                        
+
+                        # 🧬 Clone functionality - "Copy wisely, paste precisely"
                         elif selected_view == 'Clone Document':
-                            # 🧬 Clone Document per record
                             st.markdown("#### Clone a document:")
                             for idx, doc in enumerate(documents_to_display):
                                 st.markdown(f"##### Document ID: {doc.get('id', '')}")
                                 if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
                                     cloned_doc = doc.copy()
-                                    # Generate a unique ID
                                     cloned_doc['id'] = generate_unique_id()
                                     st.session_state.cloned_doc = cloned_doc
                                     st.session_state.cloned_doc_str = json.dumps(cloned_doc, indent=2)
@@ -770,7 +804,11 @@ def main():
                                     st.rerun()
                             if st.session_state.get('clone_mode', False):
                                 st.markdown("#### Edit Cloned Document and Name Your Clone:")
-                                cloned_doc_str = st.text_area("Cloned Document Content (in JSON format) - Update name please to make it unique before saving!", value=st.session_state.cloned_doc_str, height=300)
+                                cloned_doc_str = st.text_area("Cloned Document Content (in JSON format) - Update name please to make it unique before saving!", 
+                                                            value=st.session_state.cloned_doc_str, 
+                                                            height=300)
+
+                                
                                 if st.button("💾 Save Cloned Document"):
                                     try:
                                         new_doc = json.loads(cloned_doc_str)
@@ -778,47 +816,53 @@ def main():
                                         if success:
                                             st.success(f"Cloned document saved with id: {new_doc['id']} 🎉")
                                             st.session_state.selected_document_id = new_doc['id']
-                                            st.session_state.clone_mode = False
-                                            st.session_state.cloned_doc = None
-                                            st.session_state.cloned_doc_str = ''
-                                            st.rerun()
-                                        else:
-                                            st.error(message)
-                                    except json.JSONDecodeError as e:
-                                        st.error(f"Invalid JSON: {str(e)} 🚫")
-                                    
-                        elif selected_view == 'New Record':
-                            # 🆕 New Record
-                            st.markdown("#### Create a new document:")
-                            if st.button("🤖 Insert Auto-Generated Record"):
-                                success, message = save_or_clone_to_cosmos_db(container, query="Auto-generated", response="This is an auto-generated record.")
-                                if success:
-                                    st.success(message)
-                                    st.rerun()
+                                                st.session_state.clone_mode = False
+                                                st.session_state.cloned_doc = None
+                                                st.session_state.cloned_doc_str = ''
+                                                st.rerun()
+                                            else:
+                                                st.error(message)
+                                        except json.JSONDecodeError as e:
+                                            st.error(f"Invalid JSON: {str(e)} 🚫")
+
+
+                                            
+                            # 🆕 New Record view - "Every new record is a new beginning"
+                            elif selected_view == 'New Record':
+                                st.markdown("#### Create a new document:")
+                                if st.button("🤖 Insert Auto-Generated Record"):
+                                    success, message = save_or_clone_to_cosmos_db(container, 
+                                                                               query="Auto-generated", 
+                                                                               response="This is an auto-generated record.")
+                                    if success:
+                                        st.success(message)
+                                        st.rerun()
+                                    else:
+                                        st.error(message)
                                 else:
-                                    st.error(message)
-                            else:
-                                new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
-                                new_doc_str = st.text_area("Document Content (in JSON format)", value='{}', height=300)
-                                if st.button("➕ Create New Document"):
-                                    try:
-                                        new_doc = json.loads(new_doc_str)
-                                        new_doc['id'] = new_id  # Use the provided ID
-                                        success, message = insert_record(container, new_doc)
-                                        if success:
-                                            st.success(f"New document created with id: {new_doc['id']} 🎉")
-                                            st.session_state.selected_document_id = new_doc['id']
-                                            # Switch to 'Show as Edit and Save' mode
-                                            st.rerun()
-                                        else:
-                                            st.error(message)
-                                    except json.JSONDecodeError as e:
-                                        st.error(f"Invalid JSON: {str(e)} 🚫")
+                                    new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
+                                    new_doc_str = st.text_area("Document Content (in JSON format)", 
+                                                             value='{}', height=300)
+                                    if st.button("➕ Create New Document"):
+                                        try:
+                                            new_doc = json.loads(new_doc_str)
+                                            new_doc['id'] = new_id
+                                            success, message = insert_record(container, new_doc)
+                                            if success:
+                                                st.success(f"New document created with id: {new_doc['id']} 🎉")
+                                                st.session_state.selected_document_id = new_doc['id']
+                                                st.rerun()
+                                            else:
+                                                st.error(message)
+                                        except json.JSONDecodeError as e:
+                                            st.error(f"Invalid JSON: {str(e)} 🚫")
                                     
                     else:
                         st.sidebar.info("No documents found in this container. 📭")
-        
-            # 🎉 Main content area
+
+
+            
+            # 📊 Data display - "Data tells tales that words cannot"
             st.subheader(f"📊 Container: {st.session_state.selected_container}")
             if st.session_state.selected_container:
                 if documents_to_display:
@@ -826,13 +870,19 @@ def main():
                     st.dataframe(df)
                 else:
                     st.info("No documents to display. 🧐")
+
+
                 
-            # 🐙 GitHub section
+            # 🐙 GitHub integration - "Git happens"
             st.subheader("🐙 GitHub Operations")
-            github_token = os.environ.get("GITHUB")  # Read GitHub token from environment variable
-            source_repo = st.text_input("Source GitHub Repository URL", value="https://github.com/AaronCWacker/AIExamples-8-24-Streamlit")
-            new_repo_name = st.text_input("New Repository Name (for cloning)", value=f"AIExample-Clone-{datetime.now().strftime('%Y%m%d_%H%M%S')}")
+            github_token = os.environ.get("GITHUB")
+            source_repo = st.text_input("Source GitHub Repository URL", 
+                                      value="https://github.com/AaronCWacker/AIExamples-8-24-Streamlit")
+            new_repo_name = st.text_input("New Repository Name (for cloning)", 
+                                        value=f"AIExample-Clone-{datetime.now().strftime('%Y%m%d_%H%M%S')}")
+
 
+            
             col1, col2 = st.columns(2)
             with col1:
                 if st.button("📥 Clone Repository"):
@@ -853,7 +903,9 @@ def main():
                                 os.remove(zip_filename)
                     else:
                         st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
-    
+
+
+            
             with col2:
                 if st.button("📤 Push to New Repository"):
                     if github_token and source_repo:
@@ -872,7 +924,9 @@ def main():
                     else:
                         st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
 
-            # 💬 Chat with Claude
+
+            
+            # 💬 Chat functionality - "Every chat is a chance to learn"
             st.subheader("💬 Chat with Claude")
             user_input = st.text_area("Message 📨:", height=100)
 
@@ -896,14 +950,18 @@ def main():
                     # Save to Cosmos DB
                     save_to_cosmos_db(container, user_input, response.content[0].text, "")
 
-            # Display Chat History
+
+            
+            # 📜 Chat history display - "History repeats itself, first as chat, then as wisdom"
             st.subheader("Past Conversations 📜")
             for chat in st.session_state.chat_history:
                 st.text_area("You said 💬:", chat["user"], height=100, disabled=True)
                 st.text_area("Claude replied 🤖:", chat["claude"], height=200, disabled=True)
                 st.markdown("---")
 
-            # File Editor
+
+            
+            # 📝 File editor - "Edit with care, save with flair"
             if hasattr(st.session_state, 'current_file'):
                 st.subheader(f"Editing: {st.session_state.current_file} 🛠")
                 new_content = st.text_area("File Content ✏️:", st.session_state.file_content, height=300)
@@ -912,7 +970,9 @@ def main():
                         file.write(new_content)
                     st.success("File updated successfully! 🎉")
 
-            # File Management
+
+            
+            # 📂 File management - "Manage many, maintain order"
             st.sidebar.title("📁 File Management")
             
             all_files = glob.glob("*.md")
@@ -944,12 +1004,16 @@ def main():
                         os.remove(file)
                         st.rerun()
 
+                        
+
         except exceptions.CosmosHttpResponseError as e:
             st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)} 🚨")
         except Exception as e:
             st.error(f"An unexpected error occurred: {str(e)} 😱")
 
-    # 🚪 Logout button
+
+    
+    # 🚪 Logout - "All good things must come to an end"
     if st.session_state.logged_in and st.sidebar.button("🚪 Logout"):
         st.session_state.logged_in = False
         st.session_state.selected_records.clear()