Update app.py

app.py

@@ -23,7 +23,6 @@ import requests
 from PIL import Image as PilImage
 import io
 import tempfile
-import time
 from datetime import datetime
 
 # Set up logging
@@ -33,9 +32,6 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-# Get Gemini API key from Hugging Face Space secrets
-GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
-
 class MarkdownToPDFConverter:
     """
     Class to convert Markdown content to PDF using ReportLab.
@@ -49,8 +45,7 @@ class MarkdownToPDFConverter:
         base_font_size: int = 10,
         heading_scale: Dict[int, float] = None,
         include_toc: bool = True,
-        code_style: str = "github",
-        color_theme: str = "Default"
+        code_style: str = "github"
     ):
         """
         Initialize the converter with configuration options.
@@ -64,7 +59,6 @@ class MarkdownToPDFConverter:
             heading_scale: Dictionary of heading levels to font size multipliers
             include_toc: Whether to include a table of contents
             code_style: Style to use for code blocks
-            color_theme: Color theme for the PDF
         """
         self.output_path = output_path
         self.page_size = A4 if page_size.upper() == "A4" else letter
@@ -81,7 +75,6 @@ class MarkdownToPDFConverter:
         }
         self.include_toc = include_toc
         self.code_style = code_style
-        self.color_theme = color_theme
         
         # Initialize styles
         self.styles = getSampleStyleSheet()
@@ -93,18 +86,6 @@ class MarkdownToPDFConverter:
     
     def _setup_styles(self) -> None:
         """Set up custom paragraph styles for the document."""
-        # Color theme configuration
-        theme_colors = {
-            "Default": {"headings": HexColor('#000000'), "code_bg": HexColor('#EEEEEE')},
-            "Modern": {"headings": HexColor('#2962FF'), "code_bg": HexColor('#F5F5F5')},
-            "Professional": {"headings": HexColor('#1565C0'), "code_bg": HexColor('#F8F9FA')},
-            "Creative": {"headings": HexColor('#6200EA'), "code_bg": HexColor('#F3E5F5')},
-            "Elegant": {"headings": HexColor('#00695C'), "code_bg": HexColor('#E0F2F1')}
-        }
-        
-        # Use selected theme or default
-        theme = theme_colors.get(self.color_theme, theme_colors["Default"])
-        
         # Modify existing Normal style
         self.styles['Normal'].fontName = self.font_name
         self.styles['Normal'].fontSize = self.base_font_size
@@ -125,7 +106,6 @@ class MarkdownToPDFConverter:
                 self.styles[heading_name].leading = int(self.base_font_size * size_multiplier * 1.2)
                 self.styles[heading_name].spaceAfter = self.base_font_size
                 self.styles[heading_name].spaceBefore = self.base_font_size * (1 + (0.2 * (7 - level)))
-                self.styles[heading_name].textColor = theme["headings"]
             else:
                 # Create new style
                 self.styles.add(
@@ -137,7 +117,6 @@ class MarkdownToPDFConverter:
                         leading=int(self.base_font_size * size_multiplier * 1.2),
                         spaceAfter=self.base_font_size,
                         spaceBefore=self.base_font_size * (1 + (0.2 * (7 - level))),
-                        textColor=theme["headings"]
                     )
                 )
         
@@ -151,26 +130,12 @@ class MarkdownToPDFConverter:
                 spaceAfter=self.base_font_size,
                 spaceBefore=self.base_font_size,
                 leftIndent=self.base_font_size,
-                backgroundColor=theme["code_bg"],
+                backgroundColor=HexColor('#EEEEEE'),
                 borderWidth=0,
                 borderPadding=self.base_font_size * 0.5,
             )
         )
         
-        # Blockquote style
-        self.styles.add(
-            ParagraphStyle(
-                name='Blockquote',
-                parent=self.styles['Normal'],
-                leftIndent=self.base_font_size * 2,
-                rightIndent=self.base_font_size,
-                fontStyle='italic',
-                spaceAfter=self.base_font_size,
-                spaceBefore=self.base_font_size,
-                textColor=HexColor('#555555'),
-            )
-        )
-        
         # List item style
         self.styles.add(
             ParagraphStyle(
@@ -188,7 +153,6 @@ class MarkdownToPDFConverter:
                 parent=self.styles['Heading1'],
                 fontSize=int(self.base_font_size * 1.5),
                 spaceAfter=self.base_font_size * 1.5,
-                textColor=theme["headings"],
             )
         )
         
@@ -278,8 +242,7 @@ class MarkdownToPDFConverter:
                     base_font_size=self.base_font_size,
                     heading_scale=self.heading_scale,
                     include_toc=separate_toc,
-                    code_style=self.code_style,
-                    color_theme=self.color_theme
+                    code_style=self.code_style
                 )
                 converter.convert_file(file_path)
     
@@ -355,19 +318,6 @@ class MarkdownToPDFConverter:
                 Preformatted(code, self.styles['CodeBlock'])
             )
             
-        elif element.name == 'blockquote':
-            # Process blockquote content
-            blockquote_text = ""
-            for child in element.children:
-                if hasattr(child, 'name') and child.name:
-                    blockquote_text += self._process_inline_elements(child)
-                else:
-                    blockquote_text += str(child)
-            
-            self.elements.append(
-                Paragraph(blockquote_text, self.styles['Blockquote'])
-            )
-            
         elif element.name == 'img':
             src = element.get('src', '')
             alt = element.get('alt', 'Image')
@@ -428,7 +378,7 @@ class MarkdownToPDFConverter:
             
         elif element.name == 'ul' or element.name == 'ol':
             list_items = []
-            is_ordered = element.name == 'ol'
+            bullet_type = 'bullet' if element.name == 'ul' else 'numbered'
             
             for item in element.find_all('li', recursive=False):
                 text = self._process_inline_elements(item)
@@ -439,21 +389,12 @@ class MarkdownToPDFConverter:
                     )
                 )
             
-            if is_ordered:
-                self.elements.append(
-                    ListFlowable(
-                        list_items,
-                        bulletType='1',
-                        start=1,
-                        bulletFormat='%s.'
-                    )
-                )
-        else:
             self.elements.append(
                 ListFlowable(
                     list_items,
-                    bulletType='bullet',
-                    bulletFormat='%s'
+                    bulletType=bullet_type,
+                    start=1 if bullet_type == 'numbered' else None,
+                    bulletFormat='%s.' if bullet_type == 'numbered' else '%s'
                 )
             )
             
@@ -651,7 +592,7 @@ class MarkdownToPDFAgent:
         try:
             from langchain_google_genai import ChatGoogleGenerativeAI
             
-            api_key = api_key or os.getenv("GEMINI_API_KEY") or GEMINI_API_KEY
+            api_key = api_key or os.getenv("GOOGLE_API_KEY")
             if not api_key:
                 logger.warning("No Google API key provided. Agent will run without LLM enhancement.")
                 return False
@@ -753,8 +694,7 @@ class MarkdownToPDFAgent:
         return '\n'.join(result_lines)
     
     def process_file(self, input_path: str, output_path: str = None, enhance: bool = False, 
-                     enhancement_instructions: str = None, page_size: str = "A4",
-                     color_theme: str = "Default") -> str:
+                     enhancement_instructions: str = None, page_size: str = "A4") -> str:
         """
         Process a single markdown file and convert it to PDF.
         
@@ -764,7 +704,6 @@ class MarkdownToPDFAgent:
             enhance: Whether to enhance the content with LLM
             enhancement_instructions: Specific instructions for enhancement
             page_size: Page size for the PDF ("A4" or "letter")
-            color_theme: Color theme for the PDF
             
         Returns:
             Path to the generated PDF
@@ -790,8 +729,7 @@ class MarkdownToPDFAgent:
         # Configure converter
         self.converter = MarkdownToPDFConverter(
             output_path=output_path,
-            page_size=page_size,
-            color_theme=color_theme
+            page_size=page_size
         )
             
         # Convert to PDF
@@ -803,7 +741,7 @@ class MarkdownToPDFAgent:
     def process_directory(self, input_dir: str, output_dir: str = None, pattern: str = "*.md",
                         enhance: bool = False, merge: bool = False,
                         output_filename: str = "merged_document.pdf", 
-                        page_size: str = "A4", color_theme: str = "Default") -> List[str]:
+                        page_size: str = "A4") -> List[str]:
         """
         Process all markdown files in a directory.
         
@@ -815,7 +753,6 @@ class MarkdownToPDFAgent:
             merge: Whether to merge all files into a single PDF
             output_filename: Filename for merged PDF
             page_size: Page size for the PDF ("A4" or "letter")
-            color_theme: Color theme for the PDF
             
         Returns:
             List of paths to generated PDFs
@@ -867,8 +804,7 @@ class MarkdownToPDFAgent:
                 output_path = os.path.join(output_dir, output_filename)
                 self.converter = MarkdownToPDFConverter(
                     output_path=output_path,
-                    page_size=page_size,
-                    color_theme=color_theme
+                    page_size=page_size
                 )
                 self.converter.convert_content(merged_content)
                 
@@ -878,8 +814,7 @@ class MarkdownToPDFAgent:
                 output_path = os.path.join(output_dir, output_filename)
                 self.converter = MarkdownToPDFConverter(
                     output_path=output_path,
-                    page_size=page_size,
-                    color_theme=color_theme
+                    page_size=page_size
                 )
                 self.converter.convert_multiple_files(md_files, merge=True)
                 
@@ -896,8 +831,7 @@ class MarkdownToPDFAgent:
                     md_file, 
                     output_path, 
                     enhance=enhance,
-                    page_size=page_size,
-                    color_theme=color_theme
+                    page_size=page_size
                 )
                 
                 if processed_file:
@@ -905,159 +839,10 @@ class MarkdownToPDFAgent:
             
             return output_files
 
-# Sample markdown templates
-SAMPLE_TEMPLATES = {
-    "report": """# Professional Report
-    
-## Executive Summary
-
-This report presents an analysis of the key findings, methodologies, and recommendations for consideration.
-
-## Introduction
-
-In this section, we provide context and background information essential for understanding the purpose and scope of this report.
-
-## Methodology
-
-Our approach included the following steps:
-
-1. Data collection from multiple sources
-2. Rigorous analysis using industry-standard techniques
-3. Validation of findings through peer review
-4. Development of actionable recommendations
-
-## Key Findings
-
-| Area | Status | Impact |
-|------|--------|--------|
-| Operations | Satisfactory | Medium |
-| Finance | Needs Improvement | High |
-| Marketing | Excellent | Medium |
-
-## Recommendations
-
-- Implement process improvements in the finance department
-- Continue successful marketing strategies
-- Monitor operational metrics quarterly
-
-## Conclusion
-
-The analysis reveals several opportunities for improvement while highlighting existing strengths.
-
-> **Note:** This report should be reviewed by all stakeholders before implementation.
-""",
-    
-    "documentation": """# Product Documentation
-
-## Overview
-
-This document provides comprehensive information about the product, its features, and usage instructions.
-
-## Features
-
-- **Fast Processing**: Optimized algorithms ensure quick results
-- **User-Friendly Interface**: Intuitive design for ease of use
-- **Advanced Analytics**: In-depth insights and reporting
-- **Secure Storage**: End-to-end encryption for all data
-
-## Installation
-
-```bash
-# Install via package manager
-npm install product-name
-
-# Initialize with configuration
-product-init --config=standard
-```
-
-## Usage Examples
-
-Here's a simple example of how to use the main features:
-
-```javascript
-import { Product } from 'product-name';
-
-// Initialize with settings
-const myProduct = new Product({
-  mode: 'advanced',
-  logging: true
-});
-
-// Process data
-const results = myProduct.analyze(data);
-console.log(results);
-```
-
-## Troubleshooting
-
-| Issue | Solution |
-|-------|----------|
-| Connection Error | Check network settings and firewall |
-| Performance Issues | Increase memory allocation in config |
-| Data Import Failure | Verify file format compatibility |
-
-## Support
-
-For additional assistance, contact [email protected] or visit our help center.
-""",
-    
-    "article": """# The Future of Technology: Trends to Watch
-
-## Introduction
-
-Technology continues to evolve at an unprecedented pace, transforming how we live, work, and interact. This article explores emerging trends that will shape our future.
-
-## Artificial Intelligence Advancements
-
-AI systems are becoming increasingly sophisticated, with applications across:
-
-- **Healthcare**: Diagnostic tools and personalized treatment plans
-- **Finance**: Algorithmic trading and fraud detection
-- **Manufacturing**: Predictive maintenance and quality control
-- **Transportation**: Autonomous vehicles and smart traffic management
-
-Recent research has shown that AI can now outperform human experts in specific domains while complementing human skills in others.
-
-## Sustainable Technology
-
-As environmental concerns mount, sustainable technology is gaining prominence:
-
-1. Renewable energy systems with improved efficiency
-2. Eco-friendly materials for hardware manufacturing
-3. Energy-optimized algorithms and software
-4. Circular economy approaches to technology lifecycle
-
-## Decentralized Systems
-
-Blockchain and related technologies are enabling new decentralized approaches:
-
-```
-Traditional Model → Decentralized Alternative
-Central Database → Distributed Ledger
-Single Authority → Consensus Mechanism
-Opaque Processes → Transparent Verification
-```
-
-## Conclusion
-
-These technological trends promise significant benefits but also present challenges that require thoughtful approaches to governance, ethics, and implementation.
-
-> "The best way to predict the future is to invent it." - Alan Kay
-"""
-}
-
 # Gradio Interface for Hugging Face
-def load_sample(template_choice="Basic"):
-    """Load a sample markdown document based on template choice."""
-    if template_choice == "Report":
-        return SAMPLE_TEMPLATES["report"]
-    elif template_choice == "Documentation":
-        return SAMPLE_TEMPLATES["documentation"] 
-    elif template_choice == "Article":
-        return SAMPLE_TEMPLATES["article"]
-    else:
-        # Default basic sample
-        return """# Sample Markdown Document
+def load_sample():
+    """Load a sample markdown document."""
+    return """# 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.
@@ -1085,8 +870,7 @@ def hello_world():
 """
 
 def process_markdown(markdown_text, page_size, font_size, font_name, 
-                    margin_size, include_toc, color_theme, use_ai, 
-                    enhancement_instructions=None):
+                     margin_size, include_toc, use_ai, api_key, enhancement_instructions):
     """
     Process markdown text and generate a PDF.
     
@@ -1098,7 +882,7 @@ def process_markdown(markdown_text, page_size, font_size, font_name,
     output_path = temp_file.name
     temp_file.close()
     
-    # Initialize the agent
+    # Initialize the agent and process the markdown
     agent = MarkdownToPDFAgent()
     
     # Configure converter
@@ -1108,17 +892,12 @@ def process_markdown(markdown_text, page_size, font_size, font_name,
         base_font_size=font_size,
         font_name=font_name,
         margins=(margin_size, margin_size, margin_size, margin_size),
-        include_toc=include_toc,
-        color_theme=color_theme
+        include_toc=include_toc
     )
     
-    # Check for API key in environment if AI enhancement is enabled
+    # Setup AI enhancement if requested
     enhance = False
-    
-    if use_ai:
-        # Use the API key from Hugging Face secrets if available
-        api_key = GEMINI_API_KEY
-        
+    if use_ai and api_key:
         if api_key:
             success = agent.setup_from_gemini(api_key)
             enhance = success
@@ -1134,9 +913,8 @@ def process_markdown(markdown_text, page_size, font_size, font_name,
             temp_md_path,
             output_path,
             enhance=enhance,
-            enhancement_instructions=enhancement_instructions,
-            page_size=page_size.lower(),
-            color_theme=color_theme
+            enhancement_instructions=enhancement_instructions if enhancement_instructions else None,
+            page_size=page_size.lower()
         )
         
         # Remove the temporary md file
@@ -1150,116 +928,29 @@ def process_markdown(markdown_text, page_size, font_size, font_name,
         logger.error(f"Error processing markdown: {e}")
         return None
 
-def upload_handler(files):
-    """Handle file uploads and extract markdown content"""
-    if not files:
-        return ""
-    
-    # Get the first file
-    file = files[0]
-    
-    try:
-        # Read the file content
-        if hasattr(file, 'read'):
-            # File is a file-like object
-            content = file.read().decode('utf-8')
-        else:
-            # File is a path
-            with open(file, 'r', encoding='utf-8') as f:
-                content = f.read()
-        
-        return content
-    except Exception as e:
-        logger.error(f"Error reading uploaded file: {e}")
-        return f"Error reading file: {str(e)}"
-
-# Define custom CSS for improved UI
-custom_css = """
-.gradio-container {
-    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-}
-
-.app-header {
-    text-align: center;
-    margin-bottom: 2rem;
-}
-
-.app-header h1 {
-    font-size: 2.5rem;
-    font-weight: 700;
-    color: #2E7D32;
-    margin-bottom: 0.5rem;
-}
-
-.app-header p {
-    font-size: 1.1rem;
-    color: #6c757d;
-}
-
-.template-selector {
-    margin-bottom: 1rem;
-}
-
-footer {
-    text-align: center;
-    margin-top: 2rem;
-    padding-top: 1rem;
-    border-top: 1px solid #e9ecef;
-    color: #6c757d;
-}
-
-.output-container {
-    border: 1px solid #e9ecef;
-    border-radius: 0.5rem;
-    padding: 1rem;
-    margin-top: 1rem;
-}
-"""
-
-# Define the Gradio interface - Using a simpler structure without tabs to fix the error
-with gr.Blocks(title="MarkdownMuse", css=custom_css, theme=gr.themes.Soft()) as app:
-    # App header
-    gr.HTML(
+# Define the Gradio interface
+with gr.Blocks(title="MarkdownMuse", theme=gr.themes.Soft()) as app:
+    gr.Markdown(
         """
-        <div class="app-header">
-            <h1>📝 MarkdownMuse</h1>
-            <p>Transform your Markdown files into beautifully formatted PDFs with ease</p>
-        </div>
+        # 📝 MarkdownMuse
+        
+        Transform your Markdown files into beautifully formatted PDFs with ease.
         """
     )
     
-    # Two column layout - left for input, right for templates/settings
     with gr.Row():
-        # Left column - Main input
-        with gr.Column(scale=3):
+        with gr.Column(scale=1):
+            gr.Markdown("### 📝 Input")
+            
             markdown_input = gr.TextArea(
                 placeholder="Enter your markdown content here...", 
                 label="Markdown Content",
-                lines=20,
-                elem_id="markdown-input"
+                lines=15
             )
             
-            # Upload file option
-            gr.Markdown("### 📤 Or upload a file")
-            uploaded_file = gr.File(
-                label="Upload Markdown File", 
-                file_types=[".md", ".markdown", ".txt"],
-                file_count="single"
-            )
-            upload_btn = gr.Button("Load from File")
+            sample_btn = gr.Button("📋 Load Sample")
             
-        # Right column - Templates and settings
-        with gr.Column(scale=1):
-            gr.Markdown("### 📋 Templates")
-            template_selector = gr.Dropdown(
-                choices=["Basic", "Report", "Documentation", "Article"],
-                value="Basic",
-                label="Choose a template",
-                elem_classes="template-selector"
-            )
-            sample_btn = gr.Button("Load Template")
-            
-            with gr.Accordion("⚙️ PDF Settings", open=True):
+            with gr.Accordion("⚙️ PDF Settings", open=False):
                 page_size = gr.Radio(
                     ["A4", "Letter"], 
                     label="Page Size",
@@ -1293,21 +984,18 @@ with gr.Blocks(title="MarkdownMuse", css=custom_css, theme=gr.themes.Soft()) as
                     label="Include Table of Contents"
                 )
                 
-                color_theme = gr.Dropdown(
-                    ["Default", "Modern", "Professional", "Creative", "Elegant"],
-                    value="Modern",
-                    label="Color Theme"
-                )
-                
             with gr.Accordion("🧠 AI Enhancement", open=False):
                 use_ai = gr.Checkbox(
-                    value=True,
+                    value=False,
                     label="Enable AI Enhancement"
                 )
                 
-                # Show API key status
-                api_status = "API Key: Available from Hugging Face Secrets" if GEMINI_API_KEY else "API Key: Not configured"
-                gr.Markdown(f"**{api_status}**")
+                api_key = gr.Textbox(
+                    placeholder="Enter your Google Gemini API key...",
+                    label="Google Gemini API Key",
+                    type="password",
+                    visible=True
+                )
                 
                 enhancement_instructions = gr.TextArea(
                     placeholder="Optional: Provide specific instructions for how the AI should enhance your markdown...",
@@ -1315,25 +1003,24 @@ with gr.Blocks(title="MarkdownMuse", css=custom_css, theme=gr.themes.Soft()) as
                     lines=3,
                     visible=True
                 )
-    
-    # Convert button and output
-    convert_btn = gr.Button("🔄 Convert to PDF", variant="primary", size="lg")
-    output_pdf = gr.File(label="Generated PDF")
+            
+            convert_btn = gr.Button("🔄 Convert to PDF", variant="primary")
+            
+        with gr.Column(scale=1):
+            gr.Markdown("### 📋 Output")
+            
+            output_pdf = gr.File(label="Generated PDF")
     
     # Set up event handlers
-    sample_btn.click(
-        load_sample, 
-        inputs=[template_selector], 
-        outputs=markdown_input
-    )
+    sample_btn.click(load_sample, outputs=markdown_input)
     
-    upload_btn.click(
-        upload_handler,
-        inputs=[uploaded_file],
-        outputs=markdown_input
+    # Add visibility toggle for API key based on checkbox
+    use_ai.change(
+        lambda x: [gr.update(visible=x), gr.update(visible=x)],
+        inputs=[use_ai],
+        outputs=[api_key, enhancement_instructions]
     )
     
-    # Connect the conversion button to the processing function
     convert_btn.click(
         process_markdown,
         inputs=[
@@ -1343,20 +1030,20 @@ with gr.Blocks(title="MarkdownMuse", css=custom_css, theme=gr.themes.Soft()) as
             font_name,
             margin_size,
             include_toc,
-            color_theme,
             use_ai,
+            api_key,
             enhancement_instructions
         ],
         outputs=output_pdf
     )
     
-    # Footer
-    gr.HTML(
-        f"""
-        <footer>
-            <p>Made with ❤️ using Gradio and ReportLab</p>
-            <p>© {datetime.now().year} MarkdownMuse | Deployed on Hugging Face Spaces</p>
-        </footer>
+    gr.Markdown(
+        """
+        ### 📚 About MarkdownMuse
+        
+        This tool allows you to convert Markdown documents to beautifully formatted PDFs. Use the options to customize your output.
+        
+        Made with ❤️ using Gradio and ReportLab | © 2024
         """
     )