Update app.py

app.py

@@ -588,7 +588,8 @@ def main():
                             # Create columns for each document
                             num_cols = len(documents_to_display)
                             cols = st.columns(num_cols)
-                            
+
+                        
                             for idx, (col, doc) in enumerate(zip(cols, documents_to_display)):
                                 with col:
                                     st.markdown(f"##### Document ID: {doc.get('id', '')}")
@@ -597,19 +598,38 @@ def main():
                                     editable_doc = doc.copy()
                                     editable_doc.pop('id', 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}')
-                                    if st.button("💾 Save Changes", key=f'save_button_{idx}'):
-                                        try:
-                                            updated_doc = json.loads(doc_str)
-                                            updated_doc['id'] = editable_id  # Include the possibly edited ID
-                                            success, message = update_record(container, updated_doc)
-                                            if success:
-                                                st.success(f"Document {updated_doc['id']} saved successfully.")
-                                                st.session_state.selected_document_id = updated_doc['id']
-                                                st.rerun()
-                                            else:
-                                                st.error(message)
-                                        except json.JSONDecodeError as e:
-                                            st.error(f"Invalid JSON: {str(e)} 🚫")
+                                    
+                                    # Add the "Run With AI" button next to "Save Changes"
+                                    col_save, col_ai = st.columns(2)
+                                    with col_save:
+                                        if st.button("💾 Save Changes", key=f'save_button_{idx}'):
+                                            try:
+                                                updated_doc = json.loads(doc_str)
+                                                updated_doc['id'] = editable_id  # Include the possibly edited ID
+                                                success, message = update_record(container, updated_doc)
+                                                if success:
+                                                    st.success(f"Document {updated_doc['id']} saved successfully.")
+                                                    st.session_state.selected_document_id = updated_doc['id']
+                                                    st.rerun()
+                                                else:
+                                                    st.error(message)
+                                            except json.JSONDecodeError as e:
+                                                st.error(f"Invalid JSON: {str(e)} 🚫")
+                                    with col_ai:
+                                        if st.button("🤖 Run With AI", key=f'run_with_ai_button_{idx}'):
+                                            # Use the entire document as input
+                                            try:
+                                                doc_content = json.loads(doc_str)
+                                                # Convert the JSON document to a string representation
+                                                doc_input = json.dumps(doc_content, indent=2)
+                                                # Call the process_text function with the document content
+                                                process_text(doc_input)
+                                            except json.JSONDecodeError as e:
+                                                st.error(f"Invalid JSON: {str(e)} 🚫")
+                        
+                                                
+
+                            
                                             
                         elif selected_view == 'Clone Document':
                             # 🧬 Clone Document per record