Create start.py

start.py

@@ -0,0 +1,732 @@
+import streamlit as st
+import os
+import tempfile
+import base64
+from pathlib import Path
+import time
+from datetime import datetime
+
+# Import the MarkdownToPDFAgent class from your module
+try:
+    from app import MarkdownToPDFAgent, MarkdownToPDFConverter
+except ImportError as e:
+    st.error(f"Error importing MarkdownToPDFAgent: {e}")
+    st.stop()
+
+def download_link(object_to_download, download_filename, download_link_text):
+    """
+    Generates a link to download the given object_to_download.
+    
+    Parameters:
+    object_to_download (str, Path): The file path to download
+    download_filename (str): filename that will be downloaded
+    download_link_text (str): Text to display for download link
+    
+    Returns:
+    (str): the download link as HTML
+    """
+    if isinstance(object_to_download, str) or isinstance(object_to_download, Path):
+        with open(object_to_download, 'rb') as f:
+            object_to_download = f.read()
+    
+    b64 = base64.b64encode(object_to_download).decode()
+    return f'<a href="data:application/pdf;base64,{b64}" download="{download_filename}" class="download-btn">{download_link_text}</a>'
+
+def main():
+    st.set_page_config(
+        page_title="MarkdownMuse", 
+        page_icon="📝", 
+        layout="wide",
+        initial_sidebar_state="expanded"
+    )
+    
+    # Apply custom CSS
+    st.markdown('''
+    <style>
+        .main .block-container {
+            padding-top: 2rem;
+            padding-bottom: 2rem;
+        }
+        h1, h2, h3, h4, h5, h6 {
+            font-family: 'Helvetica Neue', sans-serif;
+        }
+        .stButton>button {
+            background-color: #4CAF50;
+            color: white;
+            font-weight: 500;
+            border-radius: 6px;
+            border: none;
+            padding: 0.5rem 1rem;
+            transition: all 0.3s;
+        }
+        .stButton>button:hover {
+            background-color: #45a049;
+            box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+        }
+        .css-1v3fvcr {
+            background-color: #f8f9fa;
+        }
+        .stTextInput>div>div>input, .stTextArea>div>div>textarea {
+            border-radius: 6px;
+            border: 1px solid #ced4da;
+        }
+        .stSidebar .sidebar-content {
+            background-color: #f8f9fa;
+        }
+        iframe {
+            border: 1px solid #e9ecef;
+            border-radius: 8px;
+            box-shadow: 0 4px 6px rgba(0,0,0,0.05);
+        }
+        /* PDF Container Styling */
+        .pdf-container {
+            background-color: white;
+            border-radius: 10px;
+            padding: 20px;
+            box-shadow: 0 4px 12px rgba(0,0,0,0.08);
+            margin-bottom: 20px;
+        }
+        /* Empty state styling */
+        .empty-state {
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            justify-content: center;
+            padding: 3rem;
+            background-color: #f8f9fa;
+            border-radius: 10px;
+            border: 2px dashed #dee2e6;
+            margin: 1rem 0;
+        }
+        .empty-state img {
+            width: 120px;
+            margin-bottom: 1.5rem;
+            opacity: 0.7;
+        }
+        /* Success message styling */
+        .success-msg {
+            background-color: #d4edda;
+            color: #155724;
+            padding: 10px 15px;
+            border-radius: 6px;
+            margin-bottom: 15px;
+            animation: fadeIn 0.5s;
+        }
+        @keyframes fadeIn {
+            from { opacity: 0; }
+            to { opacity: 1; }
+        }
+        .download-btn {
+            display: inline-block;
+            background-color: #1976D2;
+            color: white;
+            text-decoration: none;
+            padding: 10px 18px;
+            border-radius: 6px;
+            margin: 0.5rem 0;
+            text-align: center;
+            transition: all 0.3s;
+            font-weight: 500;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        }
+        .download-btn:hover {
+            background-color: #1565C0;
+            box-shadow: 0 4px 8px rgba(0,0,0,0.15);
+            transform: translateY(-1px);
+        }
+        /* Tab styling */
+        .stTabs [data-baseweb="tab-list"] {
+            gap: 8px;
+        }
+        .stTabs [data-baseweb="tab"] {
+            height: 40px;
+            white-space: pre-wrap;
+            background-color: #f8f9fa;
+            border-radius: 4px 4px 0 0;
+            gap: 1px;
+            padding-top: 10px;
+            padding-bottom: 10px;
+        }
+        .stTabs [aria-selected="true"] {
+            background-color: #e9ecef;
+            border-bottom: 2px solid #1976D2;
+        }
+        .success-msg {
+            padding: 1rem;
+            background-color: #d4edda;
+            color: #155724;
+            border-radius: 6px;
+            margin-bottom: 1rem;
+        }
+        .info-msg {
+            padding: 1rem;
+            background-color: #d1ecf1;
+            color: #0c5460;
+            border-radius: 6px;
+            margin-bottom: 1rem;
+        }
+        .pdf-container {
+            background-color: white;
+            border-radius: 8px;
+            padding: 1rem;
+            box-shadow: 0 4px 8px rgba(0,0,0,0.05);
+        }
+        .app-header {
+            text-align: center;
+            margin-bottom: 2rem;
+        }
+        .app-header h1 {
+            color: #2E7D32;
+            margin-bottom: 0.5rem;
+        }
+        .app-header p {
+            color: #6c757d;
+            font-size: 1.1rem;
+        }
+        .footer {
+            text-align: center;
+            margin-top: 2rem;
+            padding: 1rem;
+            border-top: 1px solid #e9ecef;
+            color: #6c757d;
+        }
+    </style>
+    ''', unsafe_allow_html=True)
+    
+    # App header with logo and title
+    st.markdown(
+        '''
+        <div class="app-header">
+            <h1>📝 MarkdownMuse</h1>
+            <p>Transform your Markdown files into beautifully formatted PDFs with ease</p>
+        </div>
+        ''', 
+        unsafe_allow_html=True
+    )
+    
+    # Initialize agent with error handling
+    try:
+        agent = MarkdownToPDFAgent()
+    except Exception as e:
+        st.error(f"Error initializing MarkdownToPDFAgent: {e}")
+        st.info("Please make sure all required dependencies are installed. Run 'pip install -r requirements.txt' in your terminal.")
+        st.stop()
+    
+    # Sidebar for configuration with improved styling
+    with st.sidebar:
+        st.markdown("<h2 style='text-align: center; color: #2E7D32;'>⚙️ Configuration</h2>", unsafe_allow_html=True)
+        st.markdown("<hr style='margin: 0.5rem 0 1.5rem 0; opacity: 0.3;'>", unsafe_allow_html=True)
+        
+        # LLM Options with better UI
+        st.markdown("<h3 style='color: #1976D2;'>🧠 AI Enhancement</h3>", unsafe_allow_html=True)
+        use_ai = st.checkbox("Enable AI enhancement", value=False, help="Use AI to improve formatting and structure of your markdown content")
+    
+    if use_ai:
+        with st.sidebar:
+            llm_provider = st.selectbox(
+                "Select AI Provider",
+                ["OpenAI", "Google Gemini", "None"],
+                index=1,
+                format_func=lambda x: {"OpenAI": "🤖 OpenAI", "Google Gemini": "🌀 Google Gemini", "None": "❌ None"}[x]
+            )
+            
+            if llm_provider != "None":
+                api_key_container = st.container()
+                with api_key_container:
+                    st.markdown(f"<div style='background-color: #f8f9fa; padding: 0.75rem; border-radius: 6px; border-left: 4px solid #1976D2;'><strong>{llm_provider} API Key:</strong></div>", unsafe_allow_html=True)
+                    api_key = st.text_input(
+                        "",
+                        type="password",
+                        help=f"Your {llm_provider} API key. Leave empty to use environment variable.",
+                        placeholder="Enter API key or leave empty to use environment variable"
+                    )
+                
+                # Custom enhancement instructions with better UI
+                st.markdown("<div style='margin-top: 1rem;'></div>", unsafe_allow_html=True)
+                st.markdown("<div style='background-color: #f8f9fa; padding: 0.75rem; border-radius: 6px; border-left: 4px solid #1976D2;'><strong>Enhancement Instructions:</strong></div>", unsafe_allow_html=True)
+                custom_instructions = st.text_area(
+                    "",
+                    height=120,
+                    help="Provide specific instructions for how the AI should enhance your markdown.",
+                    placeholder="Example: Improve formatting, add section numbers, enhance tables, etc."
+                )
+    
+    # PDF Options with improved UI
+    with st.sidebar:
+        st.markdown("<h3 style='color: #1976D2; margin-top: 2rem;'>📄 PDF Settings</h3>", unsafe_allow_html=True)
+        
+        # Create columns for layout
+        col1, col2 = st.columns(2)
+        with col1:
+            page_size = st.selectbox(
+                "Page Size",
+                ["A4", "Letter"],
+                index=0,
+                help="Select the paper size for your PDF"
+            )
+        with col2:
+            include_toc = st.checkbox(
+            "Table of Contents",
+            value=True,
+            help="Include an automatic table of contents at the beginning of the PDF"
+            )
+        
+        # Font size with visual indicator
+        font_size = st.slider(
+            "Base Font Size",
+            min_value=8,
+            max_value=14,
+            value=10,
+            step=1,
+            format="%d pt",
+            help="Set the base font size for normal text. Headings will scale proportionally."
+        )
+        
+        # Advanced settings with better styling
+        with st.expander("🔧 Advanced PDF Settings"):
+            st.markdown("<div style='padding: 0.5rem 0;'></div>", unsafe_allow_html=True)
+            
+            font_name = st.selectbox(
+                "Font Family",
+                ["Helvetica", "Times-Roman", "Courier"],
+                index=0,
+                help="Select the primary font family for your document"
+            )
+            
+            margins = st.slider(
+                "Margins",
+                min_value=0.5,
+                max_value=2.0,
+                value=0.75,
+                step=0.25,
+                format="%.2f inches",
+                help="Set the page margins (applies to all sides)"
+            )
+            
+            # Add theme selection
+            color_theme = st.selectbox(
+                "Color Theme",
+                ["Default", "Professional", "Modern", "Classic"],
+                index=0,
+                help="Select a color theme for your PDF (affects headings and accents)"
+            )
+    
+    # Add a separator between sidebar and main content
+    st.markdown("<hr style='margin: 1.5rem 0; opacity: 0.2;'>", unsafe_allow_html=True)
+    
+    # Left and Right columns for input and preview with improved styling
+    left_col, right_col = st.columns([1, 1], gap="large")
+    
+    with left_col:
+        st.markdown("<h2 style='color: #1976D2;'>📝 Input</h2>", unsafe_allow_html=True)
+        
+        # Input method selector with icons
+        input_method = st.radio(
+            "Choose input method:",
+            options=["Enter Markdown", "Upload File(s)"],
+            format_func=lambda x: f"✏️ {x}" if x == "Enter Markdown" else f"📂 {x}",
+            horizontal=True,
+            help="Choose how you want to provide your markdown content"
+        )
+        
+        if input_method == "Enter Markdown":
+            # Add a sample button to help users
+            sample_col1, sample_col2 = st.columns([3, 1])
+            with sample_col1:
+                st.markdown("<p style='margin-bottom: 0.5rem;'>Enter or paste your markdown content below:</p>", unsafe_allow_html=True)
+            with sample_col2:
+                if st.button("📋 Load Sample", help="Load a sample markdown document to see how it works"):
+                    markdown_content = """# Sample Markdown Document
+
+## Introduction
+This is a sample markdown document to demonstrate the capabilities of **MarkdownMuse**. You can use this as a starting point for your own documents.
+
+## Features
+- Convert markdown to PDF
+- Support for tables and code blocks
+- AI enhancement options
+
+### Code Example
+```python
+def hello_world():
+    print("Hello from MarkdownMuse!")
+    return True
+```
+
+## Table Example
+| Feature | Description | Status |
+|---------|-------------|---------|
+| Markdown Conversion | Convert MD to PDF | ✅ |
+| AI Enhancement | Improve content with AI | ✅ |
+| Custom Styling | Apply custom styles | ✅ |
+
+> **Note:** This is just a sample document. Feel free to modify it or create your own!
+"""
+                else:
+                    markdown_content = st.session_state.get('markdown_content', '')
+            
+            # Improved text area with syntax highlighting hint
+            st.markdown("<div style='background-color: #f8f9fa; padding: 0.5rem; border-radius: 6px; margin-bottom: 0.5rem; font-size: 0.8rem;'>💡 <strong>Tip:</strong> Use markdown syntax for formatting - headings (#), bold (**), lists (-), code blocks (```), and more!</div>", unsafe_allow_html=True)
+            
+            markdown_content = st.text_area(
+                "",
+                value=markdown_content,
+                height=400,
+                placeholder="# Your Markdown Content Here\n\nStart typing or paste your markdown...",
+                key="markdown_content"
+            )
+            
+            # Styled convert button
+            convert_button = st.button("🔄 Convert to PDF", use_container_width=True)
+            
+            if convert_button and markdown_content:
+                with st.spinner("Converting to PDF..."):
+                    # Configure AI if selected
+                    if use_ai and llm_provider != "None":
+                        if llm_provider == "OpenAI":
+                            agent.setup_from_openai(api_key)
+                        elif llm_provider == "Google Gemini":
+                            agent.setup_from_gemini(api_key)
+                    
+                    # Create a temporary file for the markdown content
+                    with tempfile.NamedTemporaryFile(suffix='.md', delete=False) as temp_md_file:
+                        temp_md_path = temp_md_file.name
+                        temp_md_file.write(markdown_content.encode('utf-8'))
+                    
+                    # Create temporary output file
+                    temp_pdf_path = f"{temp_md_path}.pdf"
+                    
+                    # Process the file
+                    output_file = agent.process_file(
+                        temp_md_path,
+                        temp_pdf_path,
+                        enhance=use_ai and llm_provider != "None",
+                        enhancement_instructions=custom_instructions if use_ai and custom_instructions else None,
+                        page_size=page_size.lower()
+                    )
+                    
+                    # Apply font size and other settings to the converter
+                    if output_file:
+                        agent.converter.base_font_size = font_size
+                        if 'font_name' in locals():
+                            agent.converter.font_name = font_name
+                        if 'margins' in locals():
+                            agent.converter.margins = (margins, margins, margins, margins)
+                        agent.converter.include_toc = include_toc
+                    
+                    # Remove the temporary md file
+                    os.unlink(temp_md_path)
+                    
+                    # Display output
+                    if output_file:
+                        st.markdown("<div class='success-msg'><strong>✅ Success!</strong> PDF created successfully!</div>", unsafe_allow_html=True)
+                        with open(output_file, "rb") as pdf_file:
+                            pdf_bytes = pdf_file.read()
+                        
+                        with right_col:
+                            st.markdown("<h2 style='color: #1976D2;'>📋 PDF Output</h2>", unsafe_allow_html=True)
+                            
+                            # Create a container for the PDF viewer with better styling
+                            st.markdown("<div class='pdf-container'>", unsafe_allow_html=True)
+                            
+                            # Add download button with stats
+                            file_size = len(pdf_bytes) / 1024  # Size in KB
+                            download_container = st.container()
+                            with download_container:
+                                download_col1, download_col2 = st.columns([3, 1])
+                                with download_col1:
+                                    st.markdown(
+                                        download_link(pdf_bytes, "output.pdf", "📥 Download PDF"), 
+                                        unsafe_allow_html=True
+                                    )
+                                with download_col2:
+                                    st.markdown(f"<div style='text-align: right; color: #6c757d; padding-top: 0.5rem;'>{file_size:.1f} KB</div>", unsafe_allow_html=True)
+                            
+                            # Add a timestamp
+                            timestamp = datetime.now().strftime("%B %d, %Y at %I:%M %p")
+                            st.markdown(f"<div style='color: #6c757d; font-size: 0.8rem; margin-bottom: 1rem;'>Generated on {timestamp}</div>", unsafe_allow_html=True)
+                            
+                            # Display PDF using HTML iframe with base64 encoding and improved styling
+                            b64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+                            pdf_display = f'<iframe src="data:application/pdf;base64,{b64_pdf}" width="100%" height="500" type="application/pdf"></iframe>'
+                            st.markdown(pdf_display, unsafe_allow_html=True)
+                            
+                            # Close the container div
+                            st.markdown("</div>", unsafe_allow_html=True)
+                            
+                            # Add a share section
+                            with st.expander("🔗 Share Options"):
+                                st.markdown("<p>Share your PDF with others:</p>", unsafe_allow_html=True)
+                                share_col1, share_col2, share_col3 = st.columns(3)
+                                with share_col1:
+                                    st.button("📧 Email", use_container_width=True, key="email_single")
+                                with share_col2:
+                                    st.button("💬 Message", use_container_width=True, key="message_single")
+                                with share_col3:
+                                    st.button("🔗 Copy Link", use_container_width=True, key="copy_single")
+                        
+                        # Clean up temporary files
+                        os.unlink(output_file)
+                    else:
+                        st.markdown("<div style='padding: 1rem; background-color: #f8d7da; color: #721c24; border-radius: 6px; margin-bottom: 1rem;'><strong>❌ Error:</strong> Failed to generate PDF. Check logs for details.</div>", unsafe_allow_html=True)
+        
+        else:  # Upload File(s)
+            # Enhanced file uploader with visual cues
+            st.markdown("<div style='background-color: #f8f9fa; padding: 0.5rem; border-radius: 6px; margin-bottom: 0.5rem; font-size: 0.8rem;'>💡 <strong>Tip:</strong> You can upload multiple files and choose to merge them into a single PDF</div>", unsafe_allow_html=True)
+            
+            uploaded_files = st.file_uploader(
+                "Upload Markdown File(s)",
+                type=['md', 'markdown', 'txt'],
+                accept_multiple_files=True,
+                help="Select one or more markdown files to convert"
+            )
+            
+            # Show options only if files are uploaded
+            if uploaded_files:
+                st.markdown(f"<div style='background-color: #e8f4f8; padding: 0.75rem; border-radius: 6px; margin: 0.5rem 0;'><strong>📂 {len(uploaded_files)} file(s) selected</strong></div>", unsafe_allow_html=True)
+                merge_files = st.checkbox(
+                    "Merge multiple files into a single PDF",
+                    value=False,
+                    help="Combine all uploaded files into one PDF document"
+                )
+            else:
+                merge_files = False
+                st.markdown("<div style='background-color: #f8f9fa; padding: 0.75rem; border-radius: 6px; margin: 0.5rem 0; color: #6c757d;'><em>No files selected. Upload markdown files to continue.</em></div>", unsafe_allow_html=True)
+            
+            # Styled convert button
+            convert_disabled = len(uploaded_files) == 0
+            convert_button = st.button(
+                "🔄 Convert to PDF",
+                disabled=convert_disabled,
+                use_container_width=True
+            )
+            
+            if convert_button and uploaded_files:
+                with st.spinner("Converting to PDF..."):
+                    # Configure AI if selected
+                    if use_ai and llm_provider != "None":
+                        if llm_provider == "OpenAI":
+                            agent.setup_from_openai(api_key)
+                        elif llm_provider == "Google Gemini":
+                            agent.setup_from_gemini(api_key)
+                    
+                    # Create a temporary directory
+                    temp_dir = tempfile.mkdtemp()
+                    output_dir = tempfile.mkdtemp()
+                    
+                    # Save uploaded files to temporary directory
+                    temp_files = []
+                    for uploaded_file in uploaded_files:
+                        # Create temporary file
+                        temp_file_path = os.path.join(temp_dir, uploaded_file.name)
+                        with open(temp_file_path, "wb") as f:
+                            f.write(uploaded_file.getbuffer())
+                        temp_files.append(temp_file_path)
+                    
+                    if merge_files:
+                        # Merge all files
+                        output_file = os.path.join(output_dir, "merged_document.pdf")
+                        result = agent.process_directory(
+                            temp_dir,
+                            output_dir,
+                            pattern="*.*",  # Get all files in temp directory
+                            enhance=use_ai and llm_provider != "None",
+                            merge=True,
+                            output_filename="merged_document.pdf",
+                            page_size=page_size.lower()
+                        )
+                        
+                        # Apply font size and other settings to the converter
+                        if result and len(result) > 0:
+                            agent.converter.base_font_size = font_size
+                            if 'font_name' in locals():
+                                agent.converter.font_name = font_name
+                            if 'margins' in locals():
+                                agent.converter.margins = (margins, margins, margins, margins)
+                            agent.converter.include_toc = include_toc
+                        
+                        if result and len(result) > 0:
+                            with open(result[0], "rb") as pdf_file:
+                                pdf_bytes = pdf_file.read()
+                            
+                            with right_col:
+                                st.markdown("<h2 style='color: #1976D2;'>📋 PDF Output</h2>", unsafe_allow_html=True)
+                                
+                                # Create a container for the PDF viewer with better styling
+                                st.markdown("<div class='pdf-container'>", unsafe_allow_html=True)
+                                
+                                # Add download button with stats
+                                file_size = len(pdf_bytes) / 1024  # Size in KB
+                                download_container = st.container()
+                                with download_container:
+                                    download_col1, download_col2 = st.columns([3, 1])
+                                    with download_col1:
+                                        st.markdown(
+                                            download_link(pdf_bytes, "merged_document.pdf", "📥 Download Merged PDF"), 
+                                            unsafe_allow_html=True
+                                        )
+                                    with download_col2:
+                                        st.markdown(f"<div style='text-align: right; color: #6c757d; padding-top: 0.5rem;'>{file_size:.1f} KB</div>", unsafe_allow_html=True)
+                                
+                                # Add a timestamp and file count
+                                timestamp = datetime.now().strftime("%B %d, %Y at %I:%M %p")
+                                st.markdown(f"<div style='color: #6c757d; font-size: 0.8rem; margin-bottom: 1rem;'>Generated on {timestamp} | Merged {len(uploaded_files)} files</div>", unsafe_allow_html=True)
+                                
+                                # Display PDF using HTML iframe with base64 encoding and improved styling
+                                b64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+                                pdf_display = f'<iframe src="data:application/pdf;base64,{b64_pdf}" width="100%" height="500" type="application/pdf"></iframe>'
+                                st.markdown(pdf_display, unsafe_allow_html=True)
+                                
+                                # Close the container div
+                                st.markdown("</div>", unsafe_allow_html=True)
+                                
+                                # Add a share section
+                                with st.expander("🔗 Share Options"):
+                                    st.markdown("<p>Share your PDF with others:</p>", unsafe_allow_html=True)
+                                    share_col1, share_col2, share_col3 = st.columns(3)
+                                    with share_col1:
+                                        st.button("📧 Email", use_container_width=True, key="email_merged")
+                                    with share_col2:
+                                        st.button("💬 Message", use_container_width=True, key="message_merged")
+                                    with share_col3:
+                                        st.button("🔗 Copy Link", use_container_width=True, key="copy_merged")
+                        else:
+                            st.error("Failed to generate merged PDF.")
+                    
+                    else:
+                        # Process each file individually
+                        all_results = []
+                        for temp_file in temp_files:
+                            output_file = os.path.join(output_dir, os.path.basename(temp_file).replace('.md', '.pdf'))
+                            result = agent.process_file(
+                                temp_file,
+                                output_file,
+                                enhance=use_ai and llm_provider != "None",
+                                enhancement_instructions=custom_instructions if use_ai and custom_instructions else None,
+                                page_size=page_size.lower()
+                            )
+                            
+                            # Apply font size and other settings to the converter
+                            if result:
+                                agent.converter.base_font_size = font_size
+                                if 'font_name' in locals():
+                                    agent.converter.font_name = font_name
+                                if 'margins' in locals():
+                                    agent.converter.margins = (margins, margins, margins, margins)
+                                agent.converter.include_toc = include_toc
+                            if result:
+                                all_results.append(result)
+                        
+                        if all_results:
+                            with right_col:
+                                # Add a header for multiple PDFs
+                                st.markdown(f"<h2 style='color: #1976D2;'>📋 PDF Output ({len(all_results)} files)</h2>", unsafe_allow_html=True)
+                                
+                                # Create tabs for each PDF
+                                if len(all_results) > 1:
+                                    tabs = st.tabs([f"File {idx+1}: {os.path.basename(pdf_path)}" for idx, pdf_path in enumerate(all_results)])
+                                    
+                                    for idx, (tab, pdf_path) in enumerate(zip(tabs, all_results)):
+                                        with tab:
+                                            with open(pdf_path, "rb") as pdf_file:
+                                                pdf_bytes = pdf_file.read()
+                                            
+                                            file_name = os.path.basename(pdf_path)
+                                            
+                                            # Create a container for the PDF viewer with better styling
+                                            st.markdown("<div class='pdf-container'>", unsafe_allow_html=True)
+                                            
+                                            # Add download button with stats
+                                            file_size = len(pdf_bytes) / 1024  # Size in KB
+                                            download_container = st.container()
+                                            with download_container:
+                                                download_col1, download_col2 = st.columns([3, 1])
+                                                with download_col1:
+                                                    st.markdown(
+                                                        download_link(pdf_bytes, file_name, f"📥 Download {file_name}"), 
+                                                        unsafe_allow_html=True
+                                                    )
+                                                with download_col2:
+                                                    st.markdown(f"<div style='text-align: right; color: #6c757d; padding-top: 0.5rem;'>{file_size:.1f} KB</div>", unsafe_allow_html=True)
+                                            
+                                            # Add a timestamp
+                                            timestamp = datetime.now().strftime("%B %d, %Y at %I:%M %p")
+                                            st.markdown(f"<div style='color: #6c757d; font-size: 0.8rem; margin-bottom: 1rem;'>Generated on {timestamp}</div>", unsafe_allow_html=True)
+                                            
+                                            # Display PDF using HTML iframe with base64 encoding and improved styling
+                                            b64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+                                            pdf_display = f'<iframe src="data:application/pdf;base64,{b64_pdf}" width="100%" height="500" type="application/pdf"></iframe>'
+                                            st.markdown(pdf_display, unsafe_allow_html=True)
+                                            
+                                            # Close the container div
+                                            st.markdown("</div>", unsafe_allow_html=True)
+                                else:
+                                    # Just one PDF, no need for tabs
+                                    for idx, pdf_path in enumerate(all_results):
+                                        with open(pdf_path, "rb") as pdf_file:
+                                            pdf_bytes = pdf_file.read()
+                                        
+                                        file_name = os.path.basename(pdf_path)
+                                        
+                                        # Create a container for the PDF viewer with better styling
+                                        st.markdown("<div class='pdf-container'>", unsafe_allow_html=True)
+                                        
+                                        # Add download button with stats
+                                        file_size = len(pdf_bytes) / 1024  # Size in KB
+                                        download_container = st.container()
+                                        with download_container:
+                                            download_col1, download_col2 = st.columns([3, 1])
+                                            with download_col1:
+                                                st.markdown(
+                                                    download_link(pdf_bytes, file_name, f"📥 Download {file_name}"), 
+                                                    unsafe_allow_html=True
+                                                )
+                                            with download_col2:
+                                                st.markdown(f"<div style='text-align: right; color: #6c757d; padding-top: 0.5rem;'>{file_size:.1f} KB</div>", unsafe_allow_html=True)
+                                        
+                                        # Add a timestamp
+                                        timestamp = datetime.now().strftime("%B %d, %Y at %I:%M %p")
+                                        st.markdown(f"<div style='color: #6c757d; font-size: 0.8rem; margin-bottom: 1rem;'>Generated on {timestamp}</div>", unsafe_allow_html=True)
+                                        
+                                        # Display PDF using HTML iframe with base64 encoding and improved styling
+                                        b64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+                                        pdf_display = f'<iframe src="data:application/pdf;base64,{b64_pdf}" width="100%" height="500" type="application/pdf"></iframe>'
+                                        st.markdown(pdf_display, unsafe_allow_html=True)
+                                        
+                                        # Close the container div
+                                        st.markdown("</div>", unsafe_allow_html=True)
+                        else:
+                            st.error("Failed to generate PDFs.")
+                    
+                    
+    # Enhanced initial empty state for the right column
+    with right_col:
+        if "pdf_bytes" not in locals():
+            st.markdown("<h2 style='color: #1976D2;'>📋 PDF Output</h2>", unsafe_allow_html=True)
+            
+            # Create a more visually appealing empty state
+            st.markdown(
+                """
+                <div style="text-align: center; padding: 2rem; background-color: #f8f9fa; border-radius: 8px; margin-top: 1rem; border: 1px dashed #ced4da;">
+                    <img src="https://cdn-icons-png.flaticon.com/512/5610/5610944.png" width="100" style="opacity: 0.5; margin-bottom: 1rem;">
+                    <h3 style="color: #6c757d; font-weight: normal; margin-bottom: 0.5rem;">No PDF Generated Yet</h3>
+                    <p style="color: #6c757d;">Your PDF will appear here after conversion.<br>Use the options on the left to get started.</p>
+                </div>
+                """,
+                unsafe_allow_html=True
+            )
+    
+    # Add footer with current year
+    current_year = datetime.now().year
+    st.markdown(
+        f'''
+        <div class="footer">
+            <p>Made with ❤️ using Streamlit and the MarkdownToPDF Agent</p>
+            <p>© {current_year} MarkdownMuse | Last updated: {datetime.now().strftime('%B %Y')}</p>
+        </div>
+        ''',
+        unsafe_allow_html=True
+    )
+
+if __name__ == "__main__":
+    main()