Update app.py

app.py

@@ -1,23 +1,21 @@
 import streamlit as st
-from azure.cosmos import CosmosClient, PartitionKey, exceptions
+from azure.cosmos import CosmosClient, exceptions
 import os
 import pandas as pd
 import traceback
-import requests
 import shutil
-import zipfile
 from github import Github
 from git import Repo
 from datetime import datetime
 import base64
 import json
+import uuid  # 🎲 For generating unique IDs
 
 # 🎉 Welcome to our fun-filled Cosmos DB and GitHub Integration app!
 st.set_page_config(layout="wide")
 
 # 🌌 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")  # 🔑 Don't forget your key!
@@ -116,10 +114,10 @@ def delete_record(container, name, id):
     except Exception as e:
         return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
-# 🕒 Function to generate a unique ID based on timestamp
-def generate_timestamp_id():
-    # ⏰ Time waits for no one, but IDs can!
-    return datetime.now().strftime('%Y%m%d%H%M%S%f')
+# 🎲 Function to generate a unique UUID
+def generate_unique_id():
+    # 🧙‍♂️ Generating a unique UUID!
+    return str(uuid.uuid4())
 
 # 📦 Function to archive current container
 def archive_current_container(database_name, container_name, client):
@@ -164,22 +162,18 @@ def main():
         st.session_state.selected_database = None
     if 'selected_container' not in st.session_state:
         st.session_state.selected_container = None
+    if 'selected_document_id' not in st.session_state:
+        st.session_state.selected_document_id = None
 
-    # 🔐 Login section
-    if not st.session_state.logged_in:
-        st.subheader("🔐 Login")
-        # Let's be sneaky and use the Key from environment variables! 🕵️‍♂️
-        input_key = Key
-
-        # 🚀 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()
-            else:
-                st.error("Invalid key. Please check your input. 🔑❌")
+    # 🔐 Automatic Login
+    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
+
+    if st.session_state.logged_in:
         # 🌌 Initialize Cosmos DB client
         try:
             if st.session_state.client is None:
@@ -194,7 +188,8 @@ 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.session_state.selected_document_id = None
+                st.experimental_rerun()
             
             if st.session_state.selected_database:
                 database = st.session_state.client.get_database_client(st.session_state.selected_database)
@@ -203,7 +198,8 @@ def main():
                 
                 if selected_container != st.session_state.selected_container:
                     st.session_state.selected_container = selected_container
-                    st.rerun()
+                    st.session_state.selected_document_id = None
+                    st.experimental_rerun()
                 
                 if st.session_state.selected_container:
                     container = database.get_container_client(st.session_state.selected_container)
@@ -221,11 +217,15 @@ def main():
                     
                     if documents:
                         document_ids = [doc.get('id', 'Unknown') for doc in documents]
-                        selected_document = st.sidebar.selectbox("📄 Select Document", document_ids)
+                        selected_document = st.sidebar.selectbox("📄 Select Document", document_ids, index=document_ids.index(st.session_state.selected_document_id) if st.session_state.selected_document_id in document_ids else 0)
                         
-                        if selected_document:
-                            st.subheader(f"📄 Document Details: {selected_document}")
-                            selected_doc = next((doc for doc in documents if doc.get('id') == selected_document), None)
+                        if selected_document != st.session_state.selected_document_id:
+                            st.session_state.selected_document_id = selected_document
+                            st.experimental_rerun()
+                        
+                        if st.session_state.selected_document_id:
+                            st.subheader(f"📄 Document Details: {st.session_state.selected_document_id}")
+                            selected_doc = next((doc for doc in documents if doc.get('id') == st.session_state.selected_document_id), None)
                             if selected_doc:
                                 # 🎨 Add Viewer/Editor selection
                                 view_options = ['Show as Markdown', 'Show as Code Editor', 'Show as Edit and Save', 'Clone Document', 'New Record']
@@ -261,6 +261,8 @@ def main():
                                             success, message = update_record(container, updated_doc)
                                             if success:
                                                 st.success(message)
+                                                st.session_state.selected_document_id = updated_doc['id']
+                                                st.experimental_rerun()
                                             else:
                                                 st.error(message)
                                         except json.JSONDecodeError as e:
@@ -269,17 +271,20 @@ def main():
                                     # 🧬 Clone Document
                                     if st.button("📄 Clone Document"):
                                         cloned_doc = selected_doc.copy()
-                                        # Generate a unique ID based on timestamp
-                                        cloned_doc['id'] = generate_timestamp_id()
+                                        # Generate a unique ID
+                                        cloned_doc['id'] = generate_unique_id()
                                         success, message = insert_record(container, cloned_doc)
                                         if success:
                                             st.success(f"Document cloned with new id: {cloned_doc['id']} 🎉")
+                                            st.session_state.selected_document_id = cloned_doc['id']
+                                            # Switch to 'Show as Edit and Save' mode
+                                            st.experimental_rerun()
                                         else:
                                             st.error(message)
                                 elif selected_view == 'New Record':
                                     # 🆕 New Record
                                     st.markdown("#### Create a new document:")
-                                    new_id = st.text_input("ID", value=generate_timestamp_id(), key='new_id')
+                                    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:
@@ -288,6 +293,9 @@ def main():
                                             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.experimental_rerun()
                                             else:
                                                 st.error(message)
                                         except json.JSONDecodeError as e:
@@ -315,7 +323,7 @@ def main():
                 if st.button("📥 Clone Repository"):
                     if github_token and source_repo:
                         try:
-                            local_path = f"./temp_repo_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
+                            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])
@@ -361,7 +369,8 @@ def main():
         st.session_state.client = None
         st.session_state.selected_database = None
         st.session_state.selected_container = None
-        st.rerun()
+        st.session_state.selected_document_id = None
+        st.experimental_rerun()
 
 if __name__ == "__main__":
     main()