Update app.py

app.py

@@ -12,29 +12,34 @@ from datetime import datetime
 import base64
 import json
 
+# 🎉 Welcome to our fun-filled Cosmos DB and GitHub Integration app!
 st.set_page_config(layout="wide")
 
-# Cosmos DB configuration
+# 🌌 Cosmos DB configuration
 ENDPOINT = "https://acae-afd.documents.azure.com:443/"
 SUBSCRIPTION_ID = "003fba60-5b3f-48f4-ab36-3ed11bc40816"
 DATABASE_NAME = os.environ.get("COSMOS_DATABASE_NAME")
 CONTAINER_NAME = os.environ.get("COSMOS_CONTAINER_NAME")
-Key = os.environ.get("Key")
+Key = os.environ.get("Key")  # 🔑 Don't forget your key!
 
-# GitHub configuration
+# 🐙 GitHub configuration
 def download_github_repo(url, local_path):
+    # 🚚 Let's download that GitHub repo!
     if os.path.exists(local_path):
         shutil.rmtree(local_path)
     Repo.clone_from(url, local_path)
 
 def create_zip_file(source_dir, output_filename):
+    # 📦 Zipping up files like a pro!
     shutil.make_archive(output_filename, 'zip', source_dir)
 
 def create_repo(g, repo_name):
+    # 🛠️ Creating a new GitHub repo. Magic!
     user = g.get_user()
     return user.create_repo(repo_name)
 
 def push_to_github(local_path, repo, github_token):
+    # 🚀 Pushing code to GitHub. Hold on tight!
     repo_url = f"https://{github_token}@github.com/{repo.full_name}.git"
     local_repo = Repo(local_path)
     
@@ -58,83 +63,60 @@ def push_to_github(local_path, repo, github_token):
     origin.push(refspec=f'{current_branch}:{current_branch}')
 
 def get_base64_download_link(file_path, file_name):
+    # 🧙‍♂️ Generating a magical download link!
     with open(file_path, "rb") as file:
         contents = file.read()
     base64_encoded = base64.b64encode(contents).decode()
-    return f'<a href="data:application/zip;base64,{base64_encoded}" download="{file_name}">Download {file_name}</a>'
+    return f'<a href="data:application/zip;base64,{base64_encoded}" download="{file_name}">⬇️ Download {file_name}</a>'
 
 
-# New functions for dynamic sidebar
+# 🧭 New functions for dynamic sidebar navigation
 def get_databases(client):
+    # 📚 Fetching list of databases. So many options!
     return [db['id'] for db in client.list_databases()]
 
 def get_containers(database):
+    # 📂 Getting containers. Containers within containers!
     return [container['id'] for container in database.list_containers()]
 
 def get_documents(container, limit=1000):
+    # 📝 Retrieving documents. Shhh, don't tell anyone!
     query = "SELECT * FROM c"
     items = list(container.query_items(query=query, enable_cross_partition_query=True, max_item_count=limit))
     return items
 
 
-# Cosmos DB functions
-def insert_record(record):
+# 🌟 Cosmos DB functions
+def insert_record(container, record):
     try:
-        response = container.create_item(body=record)
-        return True, response
+        container.create_item(body=record)
+        return True, "Record inserted successfully! 🎉"
     except exceptions.CosmosHttpResponseError as e:
-        return False, f"HTTP error occurred: {str(e)}. Status code: {e.status_code}"
+        return False, f"HTTP error occurred: {str(e)}. Status code: {e.status_code} 🚨"
     except Exception as e:
-        return False, f"An unexpected error occurred: {str(e)}"
+        return False, f"An unexpected error occurred: {str(e)} 😱"
 
-def call_stored_procedure(record):
-    try:
-        response = container.scripts.execute_stored_procedure(
-            sproc="processPrompt",
-            params=[record],
-            partition_key=record['id']
-        )
-        return True, response
-    except exceptions.CosmosHttpResponseError as e:
-        error_message = f"HTTP error occurred: {str(e)}. Status code: {e.status_code}"
-        return False, error_message
-    except Exception as e:
-        error_message = f"An unexpected error occurred: {str(e)}"
-        return False, error_message
-
-def fetch_all_records():
-    try:
-        query = "SELECT * FROM c"
-        items = list(container.query_items(query=query, enable_cross_partition_query=True))
-        return pd.DataFrame(items)
-    except exceptions.CosmosHttpResponseError as e:
-        st.error(f"HTTP error occurred while fetching records: {str(e)}. Status code: {e.status_code}")
-        return pd.DataFrame()
-    except Exception as e:
-        st.error(f"An unexpected error occurred while fetching records: {str(e)}")
-        return pd.DataFrame()
-
-def update_record(updated_record):
+def update_record(container, updated_record):
     try:
         container.upsert_item(body=updated_record)
-        return True, f"Record with id {updated_record['id']} successfully updated."
+        return True, f"Record with id {updated_record['id']} successfully updated. 🛠️"
     except exceptions.CosmosHttpResponseError as e:
-        return False, f"HTTP error occurred: {str(e)}. Status code: {e.status_code}"
+        return False, f"HTTP error occurred: {str(e)}. Status code: {e.status_code} 🚨"
     except Exception as e:
-        return False, f"An unexpected error occurred: {traceback.format_exc()}"
+        return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
-def delete_record(name, id):
+def delete_record(container, name, id):
     try:
         container.delete_item(item=id, partition_key=id)
-        return True, f"Successfully deleted record with name: {name} and id: {id}"
+        return True, f"Successfully deleted record with name: {name} and id: {id} 🗑️"
     except exceptions.CosmosResourceNotFoundError:
-        return False, f"Record with id {id} not found. It may have been already deleted."
+        return False, f"Record with id {id} not found. It may have been already deleted. 🕵️‍♂️"
     except exceptions.CosmosHttpResponseError as e:
-        return False, f"HTTP error occurred: {str(e)}. Status code: {e.status_code}"
+        return False, f"HTTP error occurred: {str(e)}. Status code: {e.status_code} 🚨"
     except Exception as e:
-        return False, f"An unexpected error occurred: {traceback.format_exc()}"
+        return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
-# New function to archive current container
+# 📦 Function to archive current container
 def archive_current_container(database_name, container_name, client):
     try:
         base_dir = "./cosmos_archive_current_container"
@@ -159,14 +141,14 @@ def archive_current_container(database_name, container_name, client):
         
         return get_base64_download_link(f"{archive_name}.zip", f"{archive_name}.zip")
     except Exception as e:
-        return f"An error occurred while archiving data: {str(e)}"
+        return f"An error occurred while archiving data: {str(e)} 😢"
 
 
-# Modify the main app
+# 🎈 Let's modify the main app to be more fun!
 def main():
     st.title("🌟 Cosmos DB and GitHub Integration")
 
-    # Initialize session state
+    # 🚦 Initialize session state
     if 'logged_in' not in st.session_state:
         st.session_state.logged_in = False
     if 'selected_records' not in st.session_state:
@@ -178,27 +160,27 @@ def main():
     if 'selected_container' not in st.session_state:
         st.session_state.selected_container = None
 
-    # Login section
+    # 🔐 Login section
     if not st.session_state.logged_in:
         st.subheader("🔐 Login")
-        #input_key = st.text_input("Enter your Cosmos DB key", type="password")
-        input_key=Key
+        # Let's be sneaky and use the Key from environment variables! 🕵️‍♂️
+        input_key = Key
 
-        # Cosmos DB configuration
+        # 🚀 Time to blast off!
         if st.button("🚀 Login"):
             if input_key:
                 st.session_state.primary_key = input_key
                 st.session_state.logged_in = True
-                st.rerun()
+                st.experimental_rerun()
             else:
-                st.error("Invalid key. Please check your input.")
+                st.error("Invalid key. Please check your input. 🔑❌")
     else:
-        # Initialize Cosmos DB client
+        # 🌌 Initialize Cosmos DB client
         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
+            # 🗄️ Sidebar for database, container, and document selection
             st.sidebar.title("🗄️ Cosmos DB Navigator")
             
             databases = get_databases(st.session_state.client)
@@ -207,7 +189,7 @@ def main():
             if selected_db != st.session_state.selected_database:
                 st.session_state.selected_database = selected_db
                 st.session_state.selected_container = None
-                st.rerun()
+                st.experimental_rerun()
             
             if st.session_state.selected_database:
                 database = st.session_state.client.get_database_client(st.session_state.selected_database)
@@ -216,12 +198,12 @@ def main():
                 
                 if selected_container != st.session_state.selected_container:
                     st.session_state.selected_container = selected_container
-                    st.rerun()
+                    st.experimental_rerun()
                 
                 if st.session_state.selected_container:
                     container = database.get_container_client(st.session_state.selected_container)
                     
-                    # Add Export button
+                    # 📦 Add Export button
                     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)
                         if download_link.startswith('<a'):
@@ -240,78 +222,78 @@ def main():
                             st.subheader(f"📄 Document Details: {selected_document}")
                             selected_doc = next((doc for doc in documents if doc.get('id') == selected_document), None)
                             if selected_doc:
-                                # Add Viewer/Editor selection
+                                # 🎨 Add Viewer/Editor selection
                                 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)
-
+    
                                 if selected_view == 'Show as Markdown':
-                                    # Show as Markdown
+                                    # 🖌️ Show as Markdown
                                     items = get_documents(container)
                                     for item in items:
-                                        st.markdown(f"### ID: {item.get('id', 'Unknown')}")
+                                        st.markdown(f"### 🆔 ID: {item.get('id', 'Unknown')}")
                                         content = item.get('content', '')
                                         if isinstance(content, dict) or isinstance(content, list):
                                             content = json.dumps(content, indent=2)
                                         st.markdown(content)
                                 elif selected_view == 'Show as Code Editor':
-                                    # Show as Code Editor
+                                    # 💻 Show as Code Editor
                                     items = get_documents(container)
                                     for item in items:
                                         st.code(json.dumps(item, indent=2), language='python')
                                 elif selected_view == 'Show as Edit and Save':
-                                    # Show as Edit and Save
+                                    # ✏️ Show as Edit and Save
                                     doc_str = st.text_area("Edit Document", value=json.dumps(selected_doc, indent=2), height=300)
                                     if st.button("💾 Save"):
                                         try:
                                             updated_doc = json.loads(doc_str)
-                                            success, message = update_record(updated_doc)
+                                            success, message = update_record(container, updated_doc)
                                             if success:
                                                 st.success(message)
                                             else:
                                                 st.error(message)
                                         except json.JSONDecodeError as e:
-                                            st.error(f"Invalid JSON: {str(e)}")
+                                            st.error(f"Invalid JSON: {str(e)} 🚫")
                                 elif selected_view == 'Clone Document':
-                                    # Clone Document
+                                    # 🧬 Clone Document
                                     if st.button("📄 Clone Document"):
                                         cloned_doc = selected_doc.copy()
                                         timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
                                         cloned_doc['id'] = f"{cloned_doc['id']}_clone_{timestamp}"
-                                        success, message = insert_record(cloned_doc)
+                                        success, message = insert_record(container, cloned_doc)
                                         if success:
-                                            st.success(f"Document cloned with new id: {cloned_doc['id']}")
+                                            st.success(f"Document cloned with new id: {cloned_doc['id']} 🎉")
                                         else:
                                             st.error(message)
                                 elif selected_view == 'New Record':
-                                    # New Record
+                                    # 🆕 New Record
                                     new_doc_str = st.text_area("New Document", value='{}', height=300)
                                     if st.button("➕ Create New Document"):
                                         try:
                                             new_doc = json.loads(new_doc_str)
                                             if 'id' not in new_doc:
                                                 new_doc['id'] = f"new_doc_{datetime.now().strftime('%Y%m%d%H%M%S')}"
-                                            success, message = insert_record(new_doc)
+                                            success, message = insert_record(container, new_doc)
                                             if success:
-                                                st.success(f"New document created with id: {new_doc['id']}")
+                                                st.success(f"New document created with id: {new_doc['id']} 🎉")
                                             else:
                                                 st.error(message)
                                         except json.JSONDecodeError as e:
-                                            st.error(f"Invalid JSON: {str(e)}")
+                                            st.error(f"Invalid JSON: {str(e)} 🚫")
                     else:
-                        st.sidebar.info("No documents found in this container.")
+                        st.sidebar.info("No documents found in this container. 📭")
             
-            # Main content area
+            # 🎉 Main content area
             st.subheader(f"📊 Container: {st.session_state.selected_container}")
             if st.session_state.selected_container:
                 df = pd.DataFrame(documents)
                 st.dataframe(df)
                     
-            # GitHub section
+            # 🐙 GitHub section
             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')}")
-
+    
             col1, col2 = st.columns(2)
             with col1:
                 if st.button("📥 Clone Repository"):
@@ -322,16 +304,16 @@ def main():
                             zip_filename = f"{new_repo_name}.zip"
                             create_zip_file(local_path, zip_filename[:-4])
                             st.markdown(get_base64_download_link(zip_filename, zip_filename), unsafe_allow_html=True)
-                            st.success("Repository cloned successfully!")
+                            st.success("Repository cloned successfully! 🎉")
                         except Exception as e:
-                            st.error(f"An error occurred: {str(e)}")
+                            st.error(f"An error occurred: {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("Please ensure GitHub token is set in environment variables and source repository URL is provided.")
+                        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"):
@@ -342,28 +324,28 @@ def main():
                             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"Repository pushed successfully to {new_repo.html_url}")
+                            st.success(f"Repository pushed successfully to {new_repo.html_url} 🚀")
                         except Exception as e:
-                            st.error(f"An error occurred: {str(e)}")
+                            st.error(f"An error occurred: {str(e)} 😢")
                         finally:
                             if os.path.exists(local_path):
                                 shutil.rmtree(local_path)
                     else:
-                        st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided.")
-
+                        st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
+    
         except exceptions.CosmosHttpResponseError as e:
-            st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)}. Status code: {e.status_code}")
+            st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)}. Status code: {e.status_code} 🚨")
         except Exception as e:
-            st.error(f"An unexpected error occurred: {str(e)}")
+            st.error(f"An unexpected error occurred: {str(e)} 😱")
 
-    # Logout button
+    # 🚪 Logout button
     if st.session_state.logged_in and st.sidebar.button("🚪 Logout"):
         st.session_state.logged_in = False
         st.session_state.selected_records.clear()
         st.session_state.client = None
         st.session_state.selected_database = None
         st.session_state.selected_container = None
-        st.rerun()
+        st.experimental_rerun()
 
 if __name__ == "__main__":
     main()