Update app.py

app.py

@@ -736,9 +736,70 @@ def main():
                                             st.rerun()
                                     except Exception as e:
                                         st.error(f"Error deleting document: {str(e)}")                                        
+                        
+                        
+                        elif selected_view == 'Show as Run AI':
+                            Label = '#### ✏️ Run AI with wisdom, save with precision'
+                            st.markdown(Label)                            
+                            num_cols = len(documents_to_display)
+                            cols = st.columns(num_cols)
+                        
+                            for idx, (col, doc) in enumerate(zip(cols, documents_to_display)):
+                                with col:
+                                    # ID and Name fields
+                                    editable_id = st.text_input("ID", value=doc.get('id', ''), key=f'edit_id_{idx}')
+                                    editable_name = st.text_input("Name", value=doc.get('name', ''), key=f'edit_name_{idx}')
+                                    
+                                    # Create editable document copy without id and name
+                                    editable_doc = doc.copy()
+                                    editable_doc.pop('id', None)
+                                    editable_doc.pop('name', None)
+                        
+                                    doc_str = st.text_area("Document Content (in JSON format)", 
+                                                         value=json.dumps(editable_doc, indent=2), 
+                                                         height=300, 
+                                                         key=f'doc_str_{idx}')
+                        
+                                    # Save and AI operations columns
+                                    col_save, col_ai, col_delete = st.columns(3)
+                                    
+                                    with col_save:
+                                        if st.button("💾 Save Changes", key=f'save_runai_{idx}'):
+                                            try:
+                                                updated_doc = json.loads(doc_str)
+                                                # Reinsert ID and name from editable fields
+                                                updated_doc['id'] = editable_id
+                                                updated_doc['name'] = editable_name
+                                                response = container.upsert_item(body=updated_doc)
+                                                if response:
+                                                    st.success(f"Document {updated_doc['id']} saved successfully.")
+                                                    st.session_state.selected_document_id = updated_doc['id']
+                                                    st.rerun()
+                                            except Exception as e:
+                                                st.error(f"Error saving document: {str(e)}")
+                                    
+                                    with col_ai:
+                                        if st.button("🤖 Run AI", key=f'run_with_ai_button_{idx}'):
+                                            # Your existing AI processing code here
+                                            values_with_space = []
+                                            def extract_values2(obj):
+                                                if isinstance(obj, dict):
+                                                    for k, v in obj.items():
+                                                        extract_values2(v)
+                                                elif isinstance(obj, list):
+                                                    for item in obj:
+                                                        extract_values2(item)
+                                                elif isinstance(obj, str):
+                                                    if ' ' in obj:
+                                                        values_with_space.append(obj)
+                                            
+                                            extract_values2(doc)
+                                            for term in values_with_space:
+                                                display_glossary_entity(term)
+                                                search_glossary(term)
 
                                     
-                        elif selected_view == 'Show as Run AI':
+                        elif selected_view == 'Show as Run AI - Old':
                             Label = '#### ✏️ Run AI with wisdom, save with precision'
                             st.markdown(Label)                            
                             num_cols = len(documents_to_display)
@@ -799,9 +860,69 @@ def main():
 
                                             
                                             #search_glossary(json.dumps(editable_doc, indent=2))
+
+
+                    
+                    
+                        elif selected_view == 'Clone Document':
+                            st.markdown("#### Clone a document:")
+                            
+                            for idx, doc in enumerate(documents_to_display):
+                                st.markdown(f"##### Original Document ID: {doc.get('id', '')}")
+                                
+                                if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
+                                    # Generate unique filename using the provided function
+                                    unique_filename = gen_AI_IO_filename("Clone", doc.get('name', ''))
+                                    
+                                    # Create new document with unique ID and name based on timestamp
+                                    new_doc = {
+                                        'id': unique_filename,
+                                        'name': f"Clone_{unique_filename[:8]}",
+                                        **{k: v for k, v in doc.items() if k not in ['id', 'name', '_rid', '_self', '_etag', '_attachments', '_ts']}
+                                    }
+                                    
+                                    # Show editable preview
+                                    edited_doc = st.text_area(
+                                        "Edit cloned document:",
+                                        value=json.dumps(new_doc, indent=2),
+                                        height=300,
+                                        key=f'edit_clone_{idx}'
+                                    )
+                                    
+                                    col_save, col_update = st.columns(2)
+                                    
+                                    with col_save:
+                                        if st.button("💾 Save Clone", key=f'save_clone_{idx}'):
+                                            try:
+                                                final_doc = json.loads(edited_doc)
+                                                response = container.create_item(body=final_doc)
+                                                if response:
+                                                    st.success(f"New cloned document created with ID: {final_doc['id']}")
+                                                    st.rerun()
+                                                else:
+                                                    st.error("Failed to create new document")
+                                            except Exception as e:
+                                                st.error(f"Error creating document: {str(e)}")
+                                    
+                                    with col_update:
+                                        if st.button("🔄 Update Clone ID/Name", key=f'update_clone_{idx}'):
+                                            try:
+                                                # Generate new unique filename
+                                                new_unique_filename = gen_AI_IO_filename("Clone", doc.get('name', ''))
+                                                current_doc = json.loads(edited_doc)
+                                                current_doc['id'] = new_unique_filename
+                                                current_doc['name'] = f"Clone_{new_unique_filename[:8]}"
+                                                
+                                                # Update the text area with new values
+                                                st.session_state[f'edit_clone_{idx}'] = json.dumps(current_doc, indent=2)
+                                                st.rerun()
+                                            except Exception as e:
+                                                st.error(f"Error updating clone: {str(e)}")
+    
+
                                             
                                                 
-                        elif selected_view == 'Clone Document':
+                        elif selected_view == 'Clone Document - Old':
                             st.markdown("#### Clone a document:")
                             
                             for idx, doc in enumerate(documents_to_display):