Update app.py

app.py

@@ -12,7 +12,7 @@ import openpyxl
 import csv
 import base64
 
-# Supported file types and their handlers
+# 🎨 File type emojis - Making file types fun and visual!
 FILE_TYPES = {
     "md": "📝 Markdown",
     "txt": "📄 Text",
@@ -23,7 +23,7 @@ FILE_TYPES = {
     "xml": "🔗 XML"
 }
 
-# Enhanced state initialization
+# 🧠 Brain initialization - Setting up our app's memory!
 if 'file_data' not in st.session_state:
     st.session_state.file_data = {}
 if 'file_types' not in st.session_state:
@@ -41,6 +41,7 @@ if 'md_files_history' not in st.session_state:
 if 'combined_markdown' not in st.session_state:
     st.session_state.combined_markdown = ""
 
+# 🧹 Clean Sweep! - Decluttering our markdown files
 def delete_all_md_files():
     """Delete all markdown files except README.md"""
     files_to_remove = [
@@ -67,19 +68,23 @@ def delete_all_md_files():
     st.session_state.combined_markdown = ""
     return len(files_to_remove)
 
+# 🎁 Download Gift Wrapper - Making files downloadable with style!
 def get_binary_file_downloader_html(bin_file, file_label='File'):
     """Generate a link allowing the data in a given file to be downloaded"""
     b64 = base64.b64encode(bin_file.encode()).decode()
     return f'<a href="data:text/plain;base64,{b64}" download="{file_label}">📥 Download {file_label}</a>'
 
+# 🔐 Secret Keeper - Encoding our content safely
 def encode_content(content):
     """Encode content to base64"""
     return base64.b64encode(content.encode()).decode()
 
+# 🔓 Mystery Solver - Decoding our secret content
 def decode_content(encoded_content):
     """Decode content from base64"""
     return base64.b64decode(encoded_content.encode()).decode()
 
+# 📚 Book Maker - Combining markdown files into a beautiful book
 def combine_markdown_files():
     """Combine all markdown files into a single document"""
     combined = []
@@ -88,6 +93,7 @@ def combine_markdown_files():
         combined.append(f"# {md_file['filename']}\n\n{content}\n\n---\n\n")
     return "".join(combined)
 
+# 📝 History Scribe - Recording every file action with precision
 def add_to_history(filename, content, action="uploaded"):
     """Add a file action to the history with timestamp"""
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -100,7 +106,6 @@ def add_to_history(filename, content, action="uploaded"):
     }
     st.session_state.file_history.insert(0, history_entry)
     
-    # Store version if it's a markdown file
     if filename.endswith('.md'):
         if filename not in st.session_state.md_versions:
             st.session_state.md_versions[filename] = []
@@ -110,7 +115,6 @@ def add_to_history(filename, content, action="uploaded"):
             "action": action
         })
         
-        # Add to MD files history if not already present
         if filename not in [f["filename"] for f in st.session_state.md_files_history]:
             st.session_state.md_files_history.append({
                 "filename": filename,
@@ -118,15 +122,14 @@ def add_to_history(filename, content, action="uploaded"):
                 "content": encoded_content
             })
         
-        # Update combined markdown
         st.session_state.combined_markdown = combine_markdown_files()
 
+# 📚 Shelf Display - Showing our markdown collection in the sidebar
 def show_sidebar_history():
     """Display markdown file history in the sidebar"""
     st.sidebar.markdown("### 📚 Markdown Files History")
     
-    # Add Delete All button
-    if st.sidebar.button("🗑️ Delete All (except README.md)"):
+    if st.sidebar.button("🧹 Delete All (except README.md)"):
         deleted_count = delete_all_md_files()
         st.sidebar.success(f"Deleted {deleted_count} markdown files")
         st.experimental_rerun()
@@ -152,22 +155,180 @@ def show_sidebar_history():
             )
             st.markdown(download_link, unsafe_allow_html=True)
 
+# 📖 Book Display - Showing our combined markdown masterpiece
 def show_book_view():
     """Display all markdown files in a book-like format"""
     if st.session_state.combined_markdown:
         st.markdown("## 📖 Book View")
         st.markdown(st.session_state.combined_markdown)
 
-# [Previous functions remain the same: show_file_history, show_markdown_versions, 
-# parse_markdown_outline, create_markdown_tabs, read_file_content, save_file_content]
+# 📋 Time Traveler - Showing our file's journey through time
+def show_file_history():
+    """Display the file history in a collapsible section"""
+    if st.session_state.file_history:
+        with st.expander("📋 File History", expanded=False):
+            st.markdown("### Recent File Activities")
+            for entry in st.session_state.file_history:
+                col1, col2 = st.columns([2, 3])
+                with col1:
+                    st.markdown(f"**{entry['timestamp']}**")
+                with col2:
+                    st.markdown(f"{entry['action'].title()}: {entry['filename']}")
 
+# ⏰ Time Machine - Exploring previous versions of our files
+def show_markdown_versions(filename):
+    """Display previous versions of a markdown file"""
+    if filename in st.session_state.md_versions:
+        versions = st.session_state.md_versions[filename]
+        if len(versions) > 1:
+            st.markdown("### 📚 Previous Versions")
+            version_idx = st.selectbox(
+                "Select version to view",
+                range(len(versions)-1),
+                format_func=lambda x: f"Version {len(versions)-1-x} - {versions[x]['timestamp']}"
+            )
+            
+            if version_idx is not None:
+                version = versions[version_idx]
+                decoded_content = decode_content(version['content'])
+                st.text_area(
+                    "Content",
+                    decoded_content,
+                    height=200,
+                    key=f"version_{filename}_{version_idx}",
+                    disabled=True
+                )
+                if st.button(f"Restore to this version", key=f"restore_{filename}_{version_idx}"):
+                    st.session_state.file_data[filename] = decoded_content
+                    st.session_state.md_outline[filename] = parse_markdown_outline(decoded_content)
+                    add_to_history(filename, decoded_content, "restored")
+                    st.experimental_rerun()
+
+# 🗺️ Map Maker - Creating a beautiful outline of our markdown
+def parse_markdown_outline(content):
+    """Generate an outline from markdown content"""
+    lines = content.split('\n')
+    outline = []
+    for line in lines:
+        if line.strip().startswith('#'):
+            level = len(line.split()[0])
+            title = line.strip('#').strip()
+            outline.append({
+                'level': level,
+                'title': title,
+                'indent': '    ' * (level - 1)
+            })
+    return outline
+
+# 📑 Tab Master - Creating beautiful tabs for our markdown content
+def create_markdown_tabs(content, filename):
+    """Create tabs for markdown content viewing and editing"""
+    tab1, tab2, tab3 = st.tabs(["📝 Editor", "👀 Preview", "🕒 History"])
+    
+    with tab1:
+        edited_content = st.text_area(
+            "Edit your markdown",
+            content,
+            height=300,
+            key=f"edit_{filename}"
+        )
+        
+        if edited_content != content:
+            st.session_state.file_data[filename] = edited_content
+            st.session_state.md_outline[filename] = parse_markdown_outline(edited_content)
+            add_to_history(filename, edited_content, "edited")
+            content = edited_content
+    
+    with tab2:
+        st.markdown("### Preview")
+        st.markdown(content)
+        
+        if filename in st.session_state.md_outline:
+            st.markdown("---")
+            st.markdown("### 📑 Document Outline")
+            for item in st.session_state.md_outline[filename]:
+                st.markdown(f"{item['indent']}• {item['title']}")
+    
+    with tab3:
+        show_markdown_versions(filename)
+    
+    return content
+
+# 📤 File Reader - Smart file reading with a smile!
+def read_file_content(uploaded_file):
+    """Smart file reader with enhanced markdown handling"""
+    file_type = uploaded_file.name.split('.')[-1].lower()
+    try:
+        if file_type == 'md':
+            content = uploaded_file.getvalue().decode()
+            st.session_state.md_outline[uploaded_file.name] = parse_markdown_outline(content)
+            st.session_state.rendered_content[uploaded_file.name] = content
+            add_to_history(uploaded_file.name, content)
+            return content, "md"
+            
+        elif file_type == 'csv':
+            df = pd.read_csv(uploaded_file)
+            return df.to_string(), "csv"
+            
+        elif file_type == 'xlsx':
+            df = pd.read_excel(uploaded_file)
+            return df.to_string(), "xlsx"
+            
+        elif file_type == 'json':
+            content = json.load(uploaded_file)
+            return json.dumps(content, indent=2), "json"
+            
+        elif file_type == 'yaml':
+            content = yaml.safe_load(uploaded_file)
+            return yaml.dump(content), "yaml"
+            
+        else:  # Default text handling
+            return uploaded_file.getvalue().decode(), "txt"
+            
+    except Exception as e:
+        st.error(f"🚨 Oops! Error reading {uploaded_file.name}: {str(e)}")
+        return None, None
+
+# 💾 File Saver - Keeping our files safe and sound
+def save_file_content(content, filename, file_type):
+    """Smart file saver with enhanced markdown handling"""
+    try:
+        if file_type == "md":
+            with open(filename, 'w') as f:
+                f.write(content)
+                st.session_state.rendered_content[filename] = content
+                add_to_history(filename, content, "saved")
+                
+        elif file_type in ["csv", "xlsx"]:
+            df = pd.read_csv(StringIO(content)) if file_type == "csv" else pd.read_excel(StringIO(content))
+            if file_type == "csv":
+                df.to_csv(filename, index=False)
+            else:
+                df.to_excel(filename, index=False)
+                
+        elif file_type == "json":
+            with open(filename, 'w') as f:
+                json.dump(json.loads(content), f, indent=2)
+                
+        elif file_type == "yaml":
+            with open(filename, 'w') as f:
+                yaml.dump(yaml.safe_load(content), f)
+                
+        else:  # Default text handling
+            with open(filename, 'w') as f:
+                f.write(content)
+                
+        return True
+    except Exception as e:
+        st.error(f"🚨 Error saving {filename}: {str(e)}")
+        return False
+
+# 🎭 Main Show - Where the magic happens!
 def main():
     st.title("📚✨ Super Smart File Handler with Markdown Magic! ✨📚")
     
-    # 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:
@@ -188,15 +349,16 @@ def main():
                 help="Upload multiple files to view as a book"
             )
         
-        # Process single file upload
-        if uploaded_file:
-            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 {FILE_TYPES.get(file_type, '📄')} file: {uploaded_file.name}")
+# Process single file upload
+if uploaded_file:
+    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  # This is the correct line
+        st.success(f"🎉 Loaded {FILE_TYPES.get(file_type, '📄')} file: {uploaded_file.name}")
+
         
-        # Process multiple file upload
+        # 📚 Process multiple file upload - Bulk file loading magic!
         if uploaded_files:
             for uploaded_file in uploaded_files:
                 content, file_type = read_file_content(uploaded_file)
@@ -205,10 +367,10 @@ def main():
                     st.session_state.file_types[uploaded_file.name] = file_type
             st.success(f"🎉 Loaded {len(uploaded_files)} files")
         
-        # Show file history
+        # 📋 Show the journey - Display file history
         show_file_history()
         
-        # Show individual files
+        # 📂 File Explorer - Show and manage individual files
         if st.session_state.file_data:
             st.subheader("📂 Your Files")
             for filename, content in st.session_state.file_data.items():
@@ -231,8 +393,10 @@ def main():
                         if save_file_content(content, filename, file_type):
                             st.success(f"✨ Saved {filename} successfully!")
     
+    # 📖 Book Mode - Show all markdown files in book format
     with book_tab:
         show_book_view()
 
+# 🚀 Launch the app!
 if __name__ == "__main__":
     main()