Update app.py

app.py

@@ -826,76 +826,8 @@ def main():
 
                         
                         
-                        elif selected_view == 'Show as Run AI':
-                            Label = '#### ✏️ Run AI with wisdom, save with precision'
-                            st.markdown(Label)
-                            
-                            for idx, doc in enumerate(documents_to_display):
-                                st.divider()  # Visual separator between documents
-                                
-                                # ID and Name fields in a smaller container
-                                col1, col2 = st.columns(2)
-                                with col1:
-                                    editable_id = st.text_input("ID", value=doc.get('id', ''), key=f'edit_id_{idx}')
-                                with col2:
-                                    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)
-                        
-                                # Document content using full width
-                                doc_str = st.text_area("Document Content (in JSON format)", 
-                                                     value=json.dumps(editable_doc, indent=2), 
-                                                     height=300, 
-                                                     key=f'doc_str_{idx}')
-                        
-                                # Action buttons in a row
-                                col_save, col_ai, col_delete = st.columns([1, 1, 1])
-                                
-                                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}'):
-                                        # AI output will use full width of the page
-                                        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)
-                                        # AI results will use full page width
-                                        with st.container():
-                                            for term in values_with_space:
-                                                display_glossary_entity(term)
-                                                search_glossary(term)
-
-
-
                         
-                        
-                        elif selected_view == 'Show as Run AI - Old':
+                        elif selected_view == 'Show as Run AI':
                             Label = '#### ✏️ Run AI with wisdom, save with precision'
                             st.markdown(Label)                            
                             num_cols = len(documents_to_display)
@@ -918,44 +850,44 @@ def main():
                                                          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)
 
-                        
+                                    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)
+
+                                    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)}")
+
+
+
+
+                                                
                         elif selected_view == 'Clone Document':
                             st.markdown("#### 📄 Clone Document (Save As)")