Update app.py

app.py

@@ -11,6 +11,127 @@ from io import StringIO
 import openpyxl
 import csv
 import base64
+import glob
+import os
+
+# Add this function after the imports and before other functions
+def scan_and_load_files():
+    """🔍 File Detective - Scans directory for supported files and loads them automatically"""
+    loaded_files = []
+    
+    # Get all files with supported extensions
+    for ext in FILE_TYPES.keys():
+        files = glob.glob(f"*.{ext}")
+        for filepath in files:
+            try:
+                with open(filepath, 'r', encoding='utf-8') as f:
+                    content = f.read()
+                    file_type = filepath.split('.')[-1].lower()
+                    
+                    # Store file content and type
+                    st.session_state.file_data[filepath] = content
+                    st.session_state.file_types[filepath] = file_type
+                    
+                    # Special handling for markdown files
+                    if file_type == 'md':
+                        if filepath not in [f["filename"] for f in st.session_state.md_files_history]:
+                            st.session_state.md_outline[filepath] = parse_markdown_outline(content)
+                            st.session_state.rendered_content[filepath] = content
+                            add_to_history(filepath, content, "auto-loaded")
+                    
+                    loaded_files.append(filepath)
+            except Exception as e:
+                st.error(f"🚨 Error loading {filepath}: {str(e)}")
+    
+    return loaded_files
+
+# Modify the main function to include initial file scanning
+def main():
+    st.title("📚✨ Super Smart File Handler with Markdown Magic! ✨📚")
+    
+    # Scan for existing files on startup
+    if 'files_scanned' not in st.session_state:
+        loaded_files = scan_and_load_files()
+        if loaded_files:
+            st.success(f"🎉 Auto-loaded {len(loaded_files)} existing files: {', '.join(loaded_files)}")
+        st.session_state.files_scanned = True
+    
+    # Show markdown history in sidebar
+    show_sidebar_history()
+    
+    # Add tabs for different upload methods
+    upload_tab, book_tab = st.tabs(["📤 File Upload", "📖 Book View"])
+    
+    with upload_tab:
+        # Add a rescan button
+        if st.button("🔄 Rescan Directory"):
+            st.session_state.files_scanned = False
+            st.experimental_rerun()
+        
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            single_uploaded_file = st.file_uploader(
+                "📤 Upload single file", 
+                type=list(FILE_TYPES.keys()),
+                help="Supports: " + ", ".join([f"{v} (.{k})" for k, v in FILE_TYPES.items()]),
+                key="single_uploader"
+            )
+        
+        with col2:
+            multiple_uploaded_files = st.file_uploader(
+                "📚 Upload multiple files",
+                type=list(FILE_TYPES.keys()),
+                accept_multiple_files=True,
+                help="Upload multiple files to view as a book",
+                key="multiple_uploader"
+            )
+        
+        # Process single file upload
+        if single_uploaded_file:
+            content, file_type = read_file_content(single_uploaded_file)
+            if content is not None:
+                st.session_state.file_data[single_uploaded_file.name] = content
+                st.session_state.file_types[single_uploaded_file.name] = file_type
+                st.success(f"🎉 Loaded {FILE_TYPES.get(file_type, '📄')} file: {single_uploaded_file.name}")
+        
+        # Process multiple file upload
+        if multiple_uploaded_files:
+            for uploaded_file in multiple_uploaded_files:
+                content, file_type = read_file_content(uploaded_file)
+                if content is not None:
+                    st.session_state.file_data[uploaded_file.name] = content
+                    st.session_state.file_types[uploaded_file.name] = file_type
+            st.success(f"🎉 Loaded {len(multiple_uploaded_files)} files")
+        
+        # Show file history
+        show_file_history()
+        
+        # Show individual files
+        if st.session_state.file_data:
+            st.subheader("📂 Your Files")
+            for filename, content in st.session_state.file_data.items():
+                file_type = st.session_state.file_types[filename]
+                with st.expander(f"{FILE_TYPES.get(file_type, '📄')} {filename}"):
+                    if file_type == "md":
+                        content = create_markdown_tabs(content, filename)
+                    else:
+                        edited_content = st.text_area(
+                            "Content",
+                            content,
+                            height=300,
+                            key=f"edit_{filename}"
+                        )
+                        if edited_content != content:
+                            st.session_state.file_data[filename] = edited_content
+                            content = edited_content
+                    
+                    if st.button(f"💾 Save {filename}"):
+                        if save_file_content(content, filename, file_type):
+                            st.success(f"✨ Saved {filename} successfully!")
+    
+    with book_tab:
+        show_book_view()
 
 # 🎨 File type emojis - Making file types fun and visual!
 FILE_TYPES = {