Update app.py

app.py

@@ -11,6 +11,8 @@ 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")
@@ -92,7 +94,7 @@ def insert_record(container, record):
         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)} 🚨"
     except Exception as e:
         return False, f"An unexpected error occurred: {str(e)} 😱"
 
@@ -101,7 +103,7 @@ def update_record(container, updated_record):
         container.upsert_item(body=updated_record)
         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)} 🚨"
     except Exception as e:
         return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
@@ -112,10 +114,15 @@ def delete_record(container, name, id):
     except exceptions.CosmosResourceNotFoundError:
         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)} 🚨"
     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 archive current container
 def archive_current_container(database_name, container_name, client):
     try:
@@ -171,7 +178,7 @@ def main():
             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. 🔑❌")
     else:
@@ -189,7 +196,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)
@@ -198,7 +205,7 @@ 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)
@@ -257,8 +264,8 @@ def main():
                                     # 🧬 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}"
+                                        # Generate a unique, short ID for the cloned document
+                                        cloned_doc['id'] = generate_short_id()
                                         success, message = insert_record(container, cloned_doc)
                                         if success:
                                             st.success(f"Document cloned with new id: {cloned_doc['id']} 🎉")
@@ -271,7 +278,7 @@ def main():
                                         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')}"
+                                                new_doc['id'] = generate_short_id()
                                             success, message = insert_record(container, new_doc)
                                             if success:
                                                 st.success(f"New document created with id: {new_doc['id']} 🎉")
@@ -334,7 +341,7 @@ def main():
                         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)} 🚨")
         except Exception as e:
             st.error(f"An unexpected error occurred: {str(e)} 😱")
 
@@ -345,7 +352,10 @@ def main():
         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()
+
+
+