import streamlit as st from streamlit.components.v1 import html from pathlib import Path import json import time from datetime import datetime import re import pandas as pd import yaml from io import StringIO import openpyxl import csv import base64 # Supported file types and their handlers FILE_TYPES = { "md": "📝 Markdown", "txt": "📄 Text", "json": "🔧 JSON", "csv": "📊 CSV", "xlsx": "📗 Excel", "yaml": "⚙️ YAML", "xml": "🔗 XML" } # Enhanced 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 = {} 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 = "" def delete_all_md_files(): """Delete all markdown files except README.md""" files_to_remove = [ f for f in st.session_state.md_files_history if f["filename"] != "README.md" ] for file in files_to_remove: filename = file["filename"] if filename in st.session_state.file_data: del st.session_state.file_data[filename] if filename in st.session_state.file_types: del st.session_state.file_types[filename] if filename in st.session_state.md_outline: del st.session_state.md_outline[filename] if filename in st.session_state.rendered_content: del st.session_state.rendered_content[filename] if filename in st.session_state.md_versions: del st.session_state.md_versions[filename] st.session_state.md_files_history = [ f for f in st.session_state.md_files_history if f["filename"] == "README.md" ] st.session_state.combined_markdown = "" return len(files_to_remove) 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'📥 Download {file_label}' 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 combine_markdown_files(): """Combine all markdown files into a single document""" combined = [] for md_file in sorted(st.session_state.md_files_history, key=lambda x: x["filename"]): content = decode_content(md_file["content"]) combined.append(f"# {md_file['filename']}\n\n{content}\n\n---\n\n") return "".join(combined) 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": encoded_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": encoded_content, "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, "timestamp": timestamp, "content": encoded_content }) # Update combined markdown st.session_state.combined_markdown = combine_markdown_files() 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)"): deleted_count = delete_all_md_files() st.sidebar.success(f"Deleted {deleted_count} markdown files") st.experimental_rerun() for md_file in st.session_state.md_files_history: col1, col2, col3 = st.sidebar.columns([2, 1, 1]) with col1: st.markdown(f"**{md_file['filename']}**") with col2: if st.button("📂 Open", key=f"open_{md_file['filename']}"): content = decode_content(md_file['content']) st.session_state.file_data[md_file['filename']] = content st.session_state.file_types[md_file['filename']] = "md" st.session_state.md_outline[md_file['filename']] = parse_markdown_outline(content) st.experimental_rerun() with col3: download_link = get_binary_file_downloader_html( decode_content(md_file['content']), md_file['filename'] ) st.markdown(download_link, unsafe_allow_html=True) 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] 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: col1, col2 = st.columns(2) with col1: uploaded_file = st.file_uploader( "📤 Upload single file", type=list(FILE_TYPES.keys()), help="Supports: " + ", ".join([f"{v} (.{k})" for k, v in FILE_TYPES.items()]) ) with col2: uploaded_files = st.file_uploader( "📚 Upload multiple files", type=list(FILE_TYPES.keys()), accept_multiple_files=True, 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 multiple file upload if uploaded_files: for uploaded_file in uploaded_files: 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 {len(uploaded_files)} files") # Show file history show_file_history() # Show individual files 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}"): 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(content, filename, file_type): st.success(f"✨ Saved {filename} successfully!") with book_tab: show_book_view() if __name__ == "__main__": main()