Update app.py

app.py

@@ -830,32 +830,33 @@ def main():
                             # 🆕 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)
+                                #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:
-                                    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)} 🚫")
+                                    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
+                                        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. 📭")