Update app.py

app.py

@@ -34,19 +34,79 @@ def scan_and_load_files():
                     
                     # Special handling for markdown files
                     if file_type == 'md':
+                        st.session_state.md_outline[filepath] = parse_markdown_outline(content)
+                        st.session_state.rendered_content[filepath] = content
+                        
+                        # Add to markdown files history if not already present
                         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")
+                            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                            encoded_content = encode_content(content)
+                            st.session_state.md_files_history.append({
+                                "filename": filepath,
+                                "timestamp": timestamp,
+                                "content": encoded_content
+                            })
+                            
+                            # Create initial version history
+                            if filepath not in st.session_state.md_versions:
+                                st.session_state.md_versions[filepath] = []
+                            st.session_state.md_versions[filepath].append({
+                                "timestamp": timestamp,
+                                "content": encoded_content,
+                                "action": "auto-loaded"
+                            })
                     
                     loaded_files.append(filepath)
             except Exception as e:
                 st.error(f"🚨 Error loading {filepath}: {str(e)}")
     
+    # Update combined markdown after loading all files
+    if loaded_files:
+        st.session_state.combined_markdown = combine_markdown_files()
+    
     return loaded_files
 
+
+
+
 # Modify the main function to include initial file scanning
 def main():
+
+    # Add a reset button in sidebar
+    if st.sidebar.button("🔄 Reset & Rescan Files"):
+        # Clear all session state
+        for key in list(st.session_state.keys()):
+            del st.session_state[key]
+        st.experimental_rerun()
+
+    st.title("📚✨ Super Smart File Handler with Markdown Magic! ✨📚")
+    
+    # Initialize session state
+    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 = {}
+    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 = []
+    if 'md_versions' not in st.session_state:
+        st.session_state.md_versions = {}
+    if 'md_files_history' not in st.session_state:
+        st.session_state.md_files_history = []
+    if 'combined_markdown' not in st.session_state:
+        st.session_state.combined_markdown = ""
+    
+    # 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
+
+        
     st.title("📚✨ Super Smart File Handler with Markdown Magic! ✨📚")
     
     # Scan for existing files on startup