Update app.py

app.py

@@ -11,7 +11,7 @@ from io import StringIO
 import openpyxl
 import csv
 
-# State initialization
+# Enhanced state initialization
 if 'file_data' not in st.session_state:
     st.session_state.file_data = {}
 if 'file_types' not in st.session_state:
@@ -20,6 +20,10 @@ if 'md_outline' not in st.session_state:
     st.session_state.md_outline = {}
 if 'rendered_content' not in st.session_state:
     st.session_state.rendered_content = {}
+if 'file_history' not in st.session_state:
+    st.session_state.file_history = []  # List to store file history
+if 'md_versions' not in st.session_state:
+    st.session_state.md_versions = {}  # Dictionary to store versions of each file
 
 # Supported file types and their handlers
 FILE_TYPES = {
@@ -32,6 +36,60 @@ FILE_TYPES = {
     "xml": "🔗 XML"
 }
 
+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")
+    history_entry = {
+        "timestamp": timestamp,
+        "filename": filename,
+        "action": action,
+        "content": content
+    }
+    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] = []
+        st.session_state.md_versions[filename].append({
+            "timestamp": timestamp,
+            "content": content,
+            "action": action
+        })
+
+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']}")
+
+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")
+            for idx, version in enumerate(versions[:-1], 1):
+                with st.expander(f"Version {len(versions) - idx} - {version['timestamp']}"):
+                    st.text_area(
+                        "Content",
+                        version['content'],
+                        height=200,
+                        key=f"version_{filename}_{idx}",
+                        disabled=True
+                    )
+                    if st.button(f"Restore to this version", key=f"restore_{filename}_{idx}"):
+                        st.session_state.file_data[filename] = version['content']
+                        st.session_state.md_outline[filename] = parse_markdown_outline(version['content'])
+                        add_to_history(filename, version['content'], "restored")
+                        st.experimental_rerun()
+
 def parse_markdown_outline(content):
     """Generate an outline from markdown content"""
     lines = content.split('\n')
@@ -49,7 +107,7 @@ def parse_markdown_outline(content):
 
 def create_markdown_tabs(content, filename):
     """Create tabs for markdown content viewing and editing"""
-    tab1, tab2 = st.tabs(["📝 Editor", "👀 Preview"])
+    tab1, tab2, tab3 = st.tabs(["📝 Editor", "👀 Preview", "🕒 History"])
     
     with tab1:
         edited_content = st.text_area(
@@ -62,6 +120,7 @@ def create_markdown_tabs(content, 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:
@@ -74,6 +133,9 @@ def create_markdown_tabs(content, filename):
             for item in st.session_state.md_outline[filename]:
                 st.markdown(f"{item['indent']}• {item['title']}")
     
+    with tab3:
+        show_markdown_versions(filename)
+    
     return content
 
 def read_file_content(uploaded_file):
@@ -84,6 +146,7 @@ def read_file_content(uploaded_file):
             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':
@@ -116,6 +179,7 @@ def save_file_content(content, filename, file_type):
             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))
@@ -150,6 +214,9 @@ def main():
         help="Supports: " + ", ".join([f"{v} (.{k})" for k, v in FILE_TYPES.items()])
     )
     
+    # Show file history at the top
+    show_file_history()
+    
     if uploaded_file:
         content, file_type = read_file_content(uploaded_file)
         if content is not None: