Update app.py

app.py

@@ -842,11 +842,18 @@ def main():
                                         except Exception as e:
                                             st.error(f"Error creating document: {str(e)}")
                     
+                        #elif selected_view == 'New Record':
                         elif selected_view == 'New Record':
-                            # 🆕 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.")
+                                auto_doc = {
+                                    "id": generate_unique_id(),
+                                    "name": f"Auto-generated Record {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
+                                    "content": "This is an auto-generated record.",
+                                    "timestamp": datetime.now().isoformat()
+                                }
+                                success, message = save_or_clone_to_cosmos_db(container, document=auto_doc)
                                 if success:
                                     st.success(message)
                                     st.rerun()
@@ -854,16 +861,24 @@ def main():
                                     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)
+                                default_doc = {
+                                    "id": new_id,
+                                    "name": "New Document",
+                                    "content": "",
+                                    "timestamp": datetime.now().isoformat()
+                                }
+                                new_doc_str = st.text_area("Document Content (in JSON format)", 
+                                                          value=json.dumps(default_doc, indent=2), 
+                                                          height=300)
+                                
                                 if st.button("➕ Create New Document"):
                                     try:
                                         new_doc = json.loads(new_doc_str)
-                                        new_doc['id'] = new_id  # Use the provided ID
+                                        new_doc['id'] = new_id  # Ensure ID matches input field
                                         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.rerun()
                                         else:
                                             st.error(message)