Update app.py

app.py

@@ -6,21 +6,22 @@ import time
 from datetime import datetime
 import re
 import pandas as pd
-#import markdown
 import yaml
 from io import StringIO
 import openpyxl
 import csv
 
-# 🦄✨ Enhanced state initialization with file type tracking! 
+# State initialization
 if 'file_data' not in st.session_state:
     st.session_state.file_data = {}
 if 'file_types' not in st.session_state:
-    st.session_state.file_types = {}  # 📁 Track file types for special handling
+    st.session_state.file_types = {}
 if 'md_outline' not in st.session_state:
-    st.session_state.md_outline = {}  # 📑 Store markdown outlines
+    st.session_state.md_outline = {}
+if 'rendered_content' not in st.session_state:
+    st.session_state.rendered_content = {}
 
-# 🎯 Supported file types and their handlers
+# Supported file types and their handlers
 FILE_TYPES = {
     "md": "📝 Markdown",
     "txt": "📄 Text",
@@ -32,12 +33,12 @@ FILE_TYPES = {
 }
 
 def parse_markdown_outline(content):
-    """📑 Generate a cute outline from markdown!"""
+    """Generate an outline from markdown content"""
     lines = content.split('\n')
     outline = []
     for line in lines:
         if line.strip().startswith('#'):
-            level = len(line.split()[0])  # Count #'s
+            level = len(line.split()[0])
             title = line.strip('#').strip()
             outline.append({
                 'level': level,
@@ -46,13 +47,43 @@ def parse_markdown_outline(content):
             })
     return outline
 
+def create_markdown_tabs(content, filename):
+    """Create tabs for markdown content viewing and editing"""
+    tab1, tab2 = st.tabs(["📝 Editor", "👀 Preview"])
+    
+    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)
+            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']}")
+    
+    return content
+
 def read_file_content(uploaded_file):
-    """📚 Smart file reader with type detection!"""
+    """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
             return content, "md"
             
         elif file_type == 'csv':
@@ -79,14 +110,14 @@ def read_file_content(uploaded_file):
         return None, None
 
 def save_file_content(content, filename, file_type):
-    """💾 Smart file saver with type handling!"""
+    """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
                 
         elif file_type in ["csv", "xlsx"]:
-            # Convert string back to DataFrame
             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)
@@ -110,22 +141,9 @@ def save_file_content(content, filename, file_type):
         st.error(f"🚨 Error saving {filename}: {str(e)}")
         return False
 
-def show_markdown_preview(content):
-    """👀 Show pretty markdown preview!"""
-    st.markdown("### 👀 Markdown Preview")
-    st.markdown(content)
-
-def show_markdown_outline(filename):
-    """📑 Display pretty markdown outline!"""
-    if filename in st.session_state.md_outline:
-        st.markdown("### 📑 Document Outline")
-        for item in st.session_state.md_outline[filename]:
-            st.markdown(f"{item['indent']}• {item['title']}")
-
 def main():
     st.title("📚✨ Super Smart File Handler with Markdown Magic! ✨📚")
     
-    # 📁 Enhanced file upload with type support
     uploaded_file = st.file_uploader(
         "📤 Upload your file!", 
         type=list(FILE_TYPES.keys()),
@@ -138,34 +156,27 @@ def main():
             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}")
-            
-            # 📝 Special handling for markdown files
-            if file_type == "md":
-                show_markdown_preview(content)
-                show_markdown_outline(uploaded_file.name)
     
-    # 📂 File Display and Editing
     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}"):
-                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
-                    if file_type == "md":
-                        st.session_state.md_outline[filename] = parse_markdown_outline(edited_content)
-                        show_markdown_preview(edited_content)
-                        show_markdown_outline(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(edited_content, filename, file_type):
+                    if save_file_content(content, filename, file_type):
                         st.success(f"✨ Saved {filename} successfully!")
 
 if __name__ == "__main__":