app.py

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

# 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 = {}

# Supported file types and their handlers
FILE_TYPES = {
    "md": "📝 Markdown",
    "txt": "📄 Text",
    "json": "🔧 JSON",
    "csv": "📊 CSV",
    "xlsx": "📗 Excel",
    "yaml": "⚙️ YAML",
    "xml": "🔗 XML"
}

def parse_markdown_outline(content):
    """Generate an outline from markdown content"""
    lines = content.split('\n')
    outline = []
    for line in lines:
        if line.strip().startswith('#'):
            level = len(line.split()[0])
            title = line.strip('#').strip()
            outline.append({
                'level': level,
                'title': title,
                'indent': '    ' * (level - 1)
            })
    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 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':
            df = pd.read_csv(uploaded_file)
            return df.to_string(), "csv"
            
        elif file_type == 'xlsx':
            df = pd.read_excel(uploaded_file)
            return df.to_string(), "xlsx"
            
        elif file_type == 'json':
            content = json.load(uploaded_file)
            return json.dumps(content, indent=2), "json"
            
        elif file_type == 'yaml':
            content = yaml.safe_load(uploaded_file)
            return yaml.dump(content), "yaml"
            
        else:  # Default text handling
            return uploaded_file.getvalue().decode(), "txt"
            
    except Exception as e:
        st.error(f"🚨 Oops! Error reading {uploaded_file.name}: {str(e)}")
        return None, None

def save_file_content(content, filename, file_type):
    """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"]:
            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)
            else:
                df.to_excel(filename, index=False)
                
        elif file_type == "json":
            with open(filename, 'w') as f:
                json.dump(json.loads(content), f, indent=2)
                
        elif file_type == "yaml":
            with open(filename, 'w') as f:
                yaml.dump(yaml.safe_load(content), f)
                
        else:  # Default text handling
            with open(filename, 'w') as f:
                f.write(content)
                
        return True
    except Exception as e:
        st.error(f"🚨 Error saving {filename}: {str(e)}")
        return False

def main():
    st.title("📚✨ Super Smart File Handler with Markdown Magic! ✨📚")
    
    uploaded_file = st.file_uploader(
        "📤 Upload your file!", 
        type=list(FILE_TYPES.keys()),
        help="Supports: " + ", ".join([f"{v} (.{k})" for k, v in FILE_TYPES.items()])
    )
    
    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}")
    
    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!")

if __name__ == "__main__":
    main()