Update app.py

app.py

@@ -11,8 +11,6 @@ from git import Repo
 from datetime import datetime
 import base64
 import json
-import random
-import string  # 🎲 For generating random IDs
 
 # 🎉 Welcome to our fun-filled Cosmos DB and GitHub Integration app!
 st.set_page_config(layout="wide")
@@ -118,10 +116,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 short, unique alphanumeric ID
-def generate_short_id(length=8):
-    # 🧙‍♂️ Generating a short, unique ID!
-    return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
+# 🕒 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 archive current container
 def archive_current_container(database_name, container_name, client):
@@ -237,7 +235,7 @@ def main():
                                     # 🖌️ 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)
@@ -246,13 +244,20 @@ def main():
                                     # 💻 Show as Code Editor
                                     items = get_documents(container)
                                     for item in items:
+                                        st.markdown(f"### **🆔 ID: {item.get('id', 'Unknown')}**")
                                         st.code(json.dumps(item, indent=2), language='python')
                                 elif selected_view == '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"):
+                                    st.markdown("#### Edit the document fields below:")
+                                    editable_id = st.text_input("ID", value=selected_doc.get('id', ''), key='edit_id')
+                                    # Remove 'id' from the document for editing other fields
+                                    editable_doc = selected_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)
+                                    if st.button("💾 Save Changes"):
                                         try:
                                             updated_doc = json.loads(doc_str)
+                                            updated_doc['id'] = editable_id  # Include the possibly edited ID
                                             success, message = update_record(container, updated_doc)
                                             if success:
                                                 st.success(message)
@@ -264,8 +269,8 @@ def main():
                                     # 🧬 Clone Document
                                     if st.button("📄 Clone Document"):
                                         cloned_doc = selected_doc.copy()
-                                        # Generate a unique, short ID for the cloned document
-                                        cloned_doc['id'] = generate_short_id()
+                                        # Generate a unique ID based on timestamp
+                                        cloned_doc['id'] = generate_timestamp_id()
                                         success, message = insert_record(container, cloned_doc)
                                         if success:
                                             st.success(f"Document cloned with new id: {cloned_doc['id']} 🎉")
@@ -273,12 +278,13 @@ def main():
                                             st.error(message)
                                 elif selected_view == 'New Record':
                                     # 🆕 New Record
-                                    new_doc_str = st.text_area("New Document", value='{}', height=300)
+                                    st.markdown("#### Create a new document:")
+                                    new_id = st.text_input("ID", value=generate_timestamp_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)
-                                            if 'id' not in new_doc:
-                                                new_doc['id'] = generate_short_id()
+                                            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']} 🎉")
@@ -292,8 +298,11 @@ def main():
             # 🎉 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)
+                if documents:
+                    df = pd.DataFrame(documents)
+                    st.dataframe(df)
+                else:
+                    st.info("No documents to display. 🧐")
                     
             # 🐙 GitHub section
             st.subheader("🐙 GitHub Operations")
@@ -356,6 +365,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-
-
-