Update app.py

app.py

@@ -511,6 +511,147 @@ def preprocess_text(text):
     return text
 
 
+
+
+
+
+def load_file_content(file_path):
+    """Load and return file content with error handling"""
+    try:
+        with open(file_path, 'r', encoding='utf-8') as file:
+            return file.read()
+    except Exception as e:
+        st.error(f"Error loading file: {str(e)}")
+        return None
+
+def save_file_content(file_path, content):
+    """Save file content with error handling"""
+    try:
+        with open(file_path, 'w', encoding='utf-8') as file:
+            file.write(content)
+        return True
+    except Exception as e:
+        st.error(f"Error saving file: {str(e)}")
+        return False
+
+def display_file_viewer(file_path):
+    """Display file content in markdown viewer"""
+    content = load_file_content(file_path)
+    if content:
+        st.markdown("### 📄 File Viewer")
+        st.markdown(f"**Viewing:** {file_path}")
+        
+        # Add file metadata
+        file_stats = os.stat(file_path)
+        st.markdown(f"**Last modified:** {datetime.fromtimestamp(file_stats.st_mtime).strftime('%Y-%m-%d %H:%M:%S')}")
+        st.markdown(f"**Size:** {file_stats.st_size} bytes")
+        
+        # Display content in markdown
+        st.markdown("---")
+        st.markdown(content)
+        
+        # Add download button
+        st.download_button(
+            label="⬇️ Download File",
+            data=content,
+            file_name=os.path.basename(file_path),
+            mime="text/markdown"
+        )
+
+def display_file_editor(file_path):
+    """Display file content in editor with save functionality"""
+    content = load_file_content(file_path)
+    if content:
+        st.markdown("### ✏️ File Editor")
+        st.markdown(f"**Editing:** {file_path}")
+        
+        # Create a unique key for the text area
+        editor_key = f"editor_{hash(file_path)}"
+        
+        # Editor with syntax highlighting for markdown
+        new_content = st.text_area(
+            "Edit content below:",
+            value=content,
+            height=400,
+            key=editor_key
+        )
+        
+        col1, col2 = st.columns([1, 5])
+        with col1:
+            if st.button("💾 Save Changes"):
+                if save_file_content(file_path, new_content):
+                    st.success("File saved successfully! 🎉")
+                    # Update the file content in session state
+                    st.session_state.file_content = new_content
+                    time.sleep(1)  # Give user time to see success message
+                    st.rerun()
+        
+        with col2:
+            st.download_button(
+                label="⬇️ Download File",
+                data=new_content,
+                file_name=os.path.basename(file_path),
+                mime="text/markdown"
+            )
+
+# Update the file management section in main()
+def update_file_management_section():
+    all_files = glob.glob("*.md")
+    all_files.sort(reverse=True)
+
+    # File management buttons in sidebar
+    st.sidebar.title("📁 File Management")
+    
+    if st.sidebar.button("🗑 Delete All Files"):
+        for file in all_files:
+            os.remove(file)
+        st.rerun()
+
+    if st.sidebar.button("⬇️ Download All Files"):
+        zip_file = create_zip_of_files(all_files)
+        st.sidebar.markdown(get_download_link(zip_file), unsafe_allow_html=True)
+
+    # Initialize session state for view/edit mode if not exists
+    if 'file_view_mode' not in st.session_state:
+        st.session_state.file_view_mode = None
+    if 'current_file' not in st.session_state:
+        st.session_state.current_file = None
+
+    # Display files in sidebar with action buttons
+    for file in all_files:
+        col1, col2, col3, col4 = st.sidebar.columns([1,3,1,1])
+        
+        with col1:
+            if st.button("🌐", key=f"view_{file}"):
+                st.session_state.current_file = file
+                st.session_state.file_view_mode = 'view'
+                st.rerun()
+                
+        with col2:
+            st.markdown(get_download_link(file), unsafe_allow_html=True)
+            
+        with col3:
+            if st.button("📂", key=f"edit_{file}"):
+                st.session_state.current_file = file
+                st.session_state.file_view_mode = 'edit'
+                st.rerun()
+                
+        with col4:
+            if st.button("🗑", key=f"delete_{file}"):
+                os.remove(file)
+                st.rerun()
+
+    # Display viewer or editor in main area based on mode
+    if st.session_state.current_file:
+        if st.session_state.file_view_mode == 'view':
+            display_file_viewer(st.session_state.current_file)
+        elif st.session_state.file_view_mode == 'edit':
+            display_file_editor(st.session_state.current_file)
+
+
+
+
+
                 
 # 🎭 Main function - "All the world's a stage, and all the code merely players" -Shakespeare, probably
 def main():
@@ -1132,36 +1273,7 @@ def main():
                     st.success("File updated successfully! 🎉")
 
             # 📂 File management - "Manage many, maintain order"
-            st.sidebar.title("📁 File Management")
-            
-            all_files = glob.glob("*.md")
-            all_files.sort(reverse=True)
-
-            if st.sidebar.button("🗑 Delete All Files"):
-                for file in all_files:
-                    os.remove(file)
-                st.rerun()
-
-            if st.sidebar.button("⬇️ Download All Files"):
-                zip_file = create_zip_of_files(all_files)
-                st.sidebar.markdown(get_download_link(zip_file), unsafe_allow_html=True)
-
-            for file in all_files:
-                col1, col2, col3, col4 = st.sidebar.columns([1,3,1,1])
-                with col1:
-                    if st.button("🌐", key="view_"+file):
-                        st.session_state.current_file = file
-                        st.session_state.file_content = load_file(file)
-                with col2:
-                    st.markdown(get_download_link(file), unsafe_allow_html=True)
-                with col3:
-                    if st.button("📂", key="edit_"+file):
-                        st.session_state.current_file = file
-                        st.session_state.file_content = load_file(file)
-                with col4:
-                    if st.button("🗑", key="delete_"+file):
-                        os.remove(file)
-                        st.rerun()
+            update_file_management_section()
 
         except exceptions.CosmosHttpResponseError as e:
             st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)} 🚨")