Update app.py

app.py

@@ -7,11 +7,6 @@ import json
 import os
 import pandas as pd
 import pytz
-
-
-
-
-
 import random
 import re
 import shutil
@@ -1152,7 +1147,7 @@ def main():
 
                     if documents_to_display:
                         # 🎨 View options - "Different strokes for different folks"
-                        view_options = ['Show as Markdown', 'Show as Code Editor', 'Show as Run AI', 'Clone Document', 'New Record']
+                        view_options = ['Show as Markdown', 'Show as Code Editor', 'Show as Run AI', 'Show as Run Media AI', 'Clone Document', 'New Record']
                         selected_view = st.sidebar.selectbox("Select Viewer/Editor", view_options, index=2)
 
                         
@@ -1295,6 +1290,110 @@ def main():
                             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}')
+
+                                    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)}")
+
+                        
+                            # File Editor (When you need to tweak things ✏️)
+                            if hasattr(st.session_state, 'current_file'):
+                                st.subheader(f"Editing: {st.session_state.current_file} 🛠")
+                                new_content = st.text_area("File Content ✏️:", st.session_state.file_content, height=300)
+                                if st.button("Save Changes 💾"):
+                                    with open(st.session_state.current_file, 'w', encoding='utf-8') as file:
+                                        file.write(new_content)
+                                    st.success("File updated successfully! 🎉")
+                        
+                            # Image Gallery (For your viewing pleasure 📸)
+                            st.subheader("Image Gallery 🖼")
+                            image_files = glob.glob("*.png") + glob.glob("*.jpg") + glob.glob("*.jpeg")
+                            image_cols = st.slider("Gallery Columns 🖼", min_value=1, max_value=15, value=5)
+                            cols = st.columns(image_cols)
+                            for idx, image_file in enumerate(image_files):
+                                with cols[idx % image_cols]:
+                                    img = Image.open(image_file)
+                                    #st.image(img, caption=image_file, use_column_width=True)
+                                    st.image(img, use_column_width=True)
+                                    display_glossary_entity(os.path.splitext(image_file)[0])
+                        
+                            # Video Gallery (Let’s roll the tapes 🎬)
+                            st.subheader("Video Gallery 🎥")
+                            video_files = glob.glob("*.mp4")
+                            video_cols = st.slider("Gallery Columns 🎬", min_value=1, max_value=5, value=3)
+                            cols = st.columns(video_cols)
+                            for idx, video_file in enumerate(video_files):
+                                with cols[idx % video_cols]:
+                                    st.markdown(get_video_html(video_file, width="100%"), unsafe_allow_html=True)
+                                    display_glossary_entity(os.path.splitext(video_file)[0])
+                        
+                            # Audio Gallery (Tunes for the mood 🎶)
+                            st.subheader("Audio Gallery 🎧")
+                            audio_files = glob.glob("*.mp3") + glob.glob("*.wav")
+                            audio_cols = st.slider("Gallery Columns 🎶", min_value=1, max_value=15, value=5)
+                            cols = st.columns(audio_cols)
+                            for idx, audio_file in enumerate(audio_files):
+                                with cols[idx % audio_cols]:
+                                    st.markdown(get_audio_html(audio_file, width="100%"), unsafe_allow_html=True)
+                                    display_glossary_entity(os.path.splitext(audio_file)[0])
+                                
+
+                        
+                        elif selected_view == 'Show as Run Media AI':
+                            Label = '#### ✏️ Run AI with wisdom, save with precision'
+                            st.markdown(Label)                            
+                            num_cols = len(documents_to_display)
+                            cols = st.columns(num_cols)
+
+                            # Video Generator call  -  the video generation UI for container:
+                            add_video_generation_ui(container)
+
+                            
+                        
                             for idx, (col, doc) in enumerate(zip(cols, documents_to_display)):
                                 with col:
                                     # ID and Name fields
@@ -1314,8 +1413,6 @@ def main():
                                     # Save and AI operations columns
 
                                     
-                                    # Video Generator call  -  the video generation UI for container:
-                                    add_video_generation_ui(container)