Update app.py

app.py

@@ -10,6 +10,7 @@ import yaml
 from io import StringIO
 import openpyxl
 import csv
+import base64
 
 # Enhanced state initialization
 if 'file_data' not in st.session_state:
@@ -21,29 +22,27 @@ if 'md_outline' not in st.session_state:
 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
+    st.session_state.file_history = []
 if 'md_versions' not in st.session_state:
-    st.session_state.md_versions = {}  # Dictionary to store versions of each file
+    st.session_state.md_versions = {}
 
-# Supported file types and their handlers
-FILE_TYPES = {
-    "md": "📝 Markdown",
-    "txt": "📄 Text",
-    "json": "🔧 JSON",
-    "csv": "📊 CSV",
-    "xlsx": "📗 Excel",
-    "yaml": "⚙️ YAML",
-    "xml": "🔗 XML"
-}
+def encode_content(content):
+    """Encode content to base64"""
+    return base64.b64encode(content.encode()).decode()
+
+def decode_content(encoded_content):
+    """Decode content from base64"""
+    return base64.b64decode(encoded_content.encode()).decode()
 
 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")
+    encoded_content = encode_content(content)
     history_entry = {
         "timestamp": timestamp,
         "filename": filename,
         "action": action,
-        "content": content
+        "content": encoded_content
     }
     st.session_state.file_history.insert(0, history_entry)
     
@@ -53,7 +52,7 @@ def add_to_history(filename, content, action="uploaded"):
             st.session_state.md_versions[filename] = []
         st.session_state.md_versions[filename].append({
             "timestamp": timestamp,
-            "content": content,
+            "content": encoded_content,
             "action": action
         })
 
@@ -75,20 +74,27 @@ def show_markdown_versions(filename):
         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()
+            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()
 
 def parse_markdown_outline(content):
     """Generate an outline from markdown content"""