Update app.py

app.py

@@ -1,41 +1,19 @@
 import gradio as gr
-from transformers import pipeline
 import json
+from datetime import datetime
 import base64
-from datetime import datetime, timedelta
-import os
-from io import BytesIO
-from PIL import Image, ImageDraw, ImageFont
 import tempfile
+import os
 from fpdf import FPDF
-import random
-
-# Initialize AI model
-try:
-    text_generator = pipeline("text-generation", model="microsoft/DialoGPT-medium", max_length=512)
-except:
-    print("Using mock AI for development")
-    text_generator = None
+from PIL import Image
 
-# Template presets
+# Simple template presets
 TEMPLATE_PRESETS = {
-    "Weekly Planner": "A4 weekly planner, portrait orientation, Monday start, 7 columns with time slots from 6 AM to 10 PM, notes section on right side, minimalist design",
-    "Monthly Calendar": "A4 monthly calendar, landscape orientation, Sunday start, large date boxes, space for notes, clean modern style",
-    "Daily Schedule": "A4 daily time-block planner, portrait, hourly slots from 5 AM to 11 PM, priority tasks section, habit tracker, productivity focused",
-    "Habit Tracker": "A4 habit tracker, 30-day grid, colorful design, progress bars, motivational quotes, wellness focused",
-    "Meal Planner": "A4 weekly meal planner, 7 days layout, breakfast/lunch/dinner/snacks, shopping list section, healthy lifestyle theme",
-    "Study Planner": "A4 study schedule, subject blocks, assignment tracker, exam dates, academic calendar, student-friendly design",
-    "Workout Planner": "A4 fitness tracker, weekly workout schedule, exercise log, progress tracking, motivational fitness theme",
-    "Budget Planner": "A4 monthly budget tracker, income/expense categories, savings goals, bill reminders, financial planning layout"
-}
-
-# Font options
-FONT_STYLES = {
-    "Modern": "Arial, sans-serif",
-    "Classic": "Times New Roman, serif",
-    "Handwritten": "Comic Sans MS, cursive",
-    "Professional": "Helvetica, sans-serif",
-    "Creative": "Impact, sans-serif"
+    "Weekly Planner": "A4 weekly planner, portrait orientation, Monday start, 7 columns, notes section",
+    "Monthly Calendar": "A4 monthly calendar, landscape orientation, large date boxes, clean design", 
+    "Daily Schedule": "A4 daily planner, hourly time slots, priority tasks section",
+    "Habit Tracker": "A4 habit tracker, 30-day grid, progress tracking",
+    "Study Planner": "A4 study schedule, subject blocks, assignment tracker"
 }
 
 # Color schemes
@@ -43,526 +21,247 @@ COLOR_SCHEMES = {
     "Professional Blue": {"primary": "#2563EB", "secondary": "#1E40AF", "accent": "#3B82F6"},
     "Forest Green": {"primary": "#059669", "secondary": "#047857", "accent": "#10B981"},
     "Sunset Orange": {"primary": "#EA580C", "secondary": "#C2410C", "accent": "#FB923C"},
-    "Royal Purple": {"primary": "#7C3AED", "secondary": "#5B21B6", "accent": "#8B5CF6"},
-    "Rose Pink": {"primary": "#E11D48", "secondary": "#BE185D", "accent": "#F43F5E"},
-    "Charcoal Black": {"primary": "#374151", "secondary": "#1F2937", "accent": "#6B7280"}
+    "Royal Purple": {"primary": "#7C3AED", "secondary": "#5B21B6", "accent": "#8B5CF6"}
 }
 
-class EnhancedTemplatePDF(FPDF):
-    def __init__(self, title="", logo_path=None, colors=None, font_style="Modern"):
+class SimplePDF(FPDF):
+    def __init__(self, title="My Planner", colors=None):
         super().__init__()
         self.title_text = title
-        self.logo_path = logo_path
         self.colors = colors or COLOR_SCHEMES["Professional Blue"]
-        self.font_style = font_style
         
     def header(self):
-        # Add logo if provided
-        if self.logo_path and os.path.exists(self.logo_path):
-            try:
-                self.image(self.logo_path, 10, 8, 20)
-                logo_offset = 35
-            except:
-                logo_offset = 10
-        else:
-            logo_offset = 10
-            
-        # Title
-        if self.title_text:
-            self.set_font('Arial', 'B', 16)
-            # Convert hex to RGB
-            primary_rgb = tuple(int(self.colors["primary"][1:][i:i+2], 16) for i in (0, 2, 4))
-            self.set_text_color(*primary_rgb)
-            self.set_x(logo_offset)
-            self.cell(0, 10, self.title_text, 0, 1, 'L')
-            self.ln(5)
-            
+        self.set_font('Arial', 'B', 16)
+        # Convert hex to RGB
+        primary_rgb = tuple(int(self.colors["primary"][1:][i:i+2], 16) for i in (0, 2, 4))
+        self.set_text_color(*primary_rgb)
+        self.cell(0, 10, self.title_text, 0, 1, 'C')
+        self.ln(10)
+        
     def footer(self):
         self.set_y(-15)
         self.set_font('Arial', 'I', 8)
         self.set_text_color(128, 128, 128)
-        self.cell(0, 10, f'Generated on {datetime.now().strftime("%Y-%m-%d")} | Page {self.page_no()}', 0, 0, 'C')
+        self.cell(0, 10, f'Generated on {datetime.now().strftime("%Y-%m-%d")}', 0, 0, 'C')
 
-def process_logo_upload(logo_file):
-    """Process uploaded logo and return path"""
-    if logo_file is None:
-        return None
-        
+def create_simple_template(prompt, title, color_scheme):
+    """Create a simple template based on user input"""
     try:
-        # Create temp directory if it doesn't exist
-        temp_dir = "temp_uploads"
-        os.makedirs(temp_dir, exist_ok=True)
-        
-        # Save uploaded file
-        logo_path = os.path.join(temp_dir, f"logo_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png")
-        
-        # Handle different file types
-        if hasattr(logo_file, 'name'):
-            # Gradio file object
-            with open(logo_file.name, 'rb') as src:
-                image = Image.open(src)
-                # Resize logo to reasonable size
-                image = image.resize((100, 100), Image.Resampling.LANCZOS)
-                image.save(logo_path, 'PNG')
-        else:
-            # Direct file path
-            image = Image.open(logo_file)
-            image = image.resize((100, 100), Image.Resampling.LANCZOS)
-            image.save(logo_path, 'PNG')
-            
-        return logo_path
-    except Exception as e:
-        print(f"Logo processing error: {e}")
-        return None
-
-def generate_ai_structure(prompt, template_type="General"):
-    """Generate template structure using AI"""
-    enhanced_prompt = f"""
-    Create a detailed structure for a {template_type} template based on this description: {prompt}
-    
-    Return a JSON structure with these elements:
-    - layout_type: (weekly/monthly/daily/custom)
-    - columns: number of columns
-    - rows: number of rows
-    - sections: list of content sections
-    - time_format: (12h/24h)
-    - date_range: start and end dates if applicable
-    - special_features: any unique elements
-    
-    Keep the response concise and structured.
-    """
-    
-    try:
-        if text_generator:
-            response = text_generator(enhanced_prompt, max_length=200, num_return_sequences=1)
-            ai_text = response[0]['generated_text']
-        else:
-            # Mock AI response for development
-            ai_text = f"Generated structure for {template_type}: 7-column weekly layout with time slots, notes section, and date headers."
-            
-        # Create structured response
-        structure = {
-            "layout_type": "weekly" if "week" in prompt.lower() else "monthly" if "month" in prompt.lower() else "daily",
-            "columns": 7 if "week" in prompt.lower() else 5 if "daily" in prompt.lower() else 6,
-            "rows": 24 if "hour" in prompt.lower() else 6,
-            "sections": ["header", "main_content", "notes", "footer"],
-            "time_format": "12h",
-            "features": ["date_headers", "time_slots", "notes_area"]
-        }
-        
-        return structure, ai_text
+        # Get colors
+        colors = COLOR_SCHEMES.get(color_scheme, COLOR_SCHEMES["Professional Blue"])
         
-    except Exception as e:
-        return {"error": f"AI processing failed: {str(e)}"}, "Error generating structure"
-
-def create_html_template(structure, customization):
-    """Create HTML template with customization"""
-    colors = customization.get('colors', COLOR_SCHEMES["Professional Blue"])
-    font_family = FONT_STYLES.get(customization.get('font_style', 'Modern'), 'Arial, sans-serif')
-    title = customization.get('title', 'My Planner')
-    
-    html = f"""
-    <!DOCTYPE html>
-    <html>
-    <head>
-        <meta charset="UTF-8">
-        <title>{title}</title>
-        <style>
-            @page {{
-                size: A4;
-                margin: 0.5in;
-            }}
-            body {{
-                font-family: {font_family};
-                margin: 0;
-                padding: 20px;
-                color: #333;
-                background: white;
-            }}
-            .header {{
-                display: flex;
-                align-items: center;
-                margin-bottom: 30px;
-                padding-bottom: 15px;
-                border-bottom: 3px solid {colors['primary']};
-            }}
-            .logo {{
-                width: 60px;
-                height: 60px;
-                margin-right: 20px;
-                border-radius: 50%;
-            }}
-            .title {{
-                font-size: 28px;
-                font-weight: bold;
-                color: {colors['primary']};
-                margin: 0;
-            }}
-            .subtitle {{
-                font-size: 14px;
-                color: {colors['secondary']};
-                margin: 5px 0 0 0;
-            }}
-            .planner-grid {{
-                display: grid;
-                grid-template-columns: repeat({structure.get('columns', 7)}, 1fr);
-                gap: 2px;
-                margin: 20px 0;
-                border: 2px solid {colors['primary']};
-            }}
-            .grid-header {{
-                background: {colors['primary']};
-                color: white;
-                padding: 10px;
-                text-align: center;
-                font-weight: bold;
-                font-size: 14px;
-            }}
-            .grid-cell {{
-                border: 1px solid {colors['accent']};
-                min-height: 100px;
-                padding: 8px;
-                background: #fafafa;
-            }}
-            .date-number {{
-                font-weight: bold;
-                color: {colors['secondary']};
-                font-size: 16px;
-                margin-bottom: 5px;
-            }}
-            .notes-section {{
-                margin-top: 30px;
-                padding: 20px;
-                background: #f8f9fa;
-                border-left: 5px solid {colors['accent']};
-            }}
-            .notes-title {{
-                font-weight: bold;
-                color: {colors['primary']};
-                margin-bottom: 15px;
-                font-size: 18px;
-            }}
-            .notes-lines {{
-                line-height: 24px;
-                min-height: 120px;
-                background-image: repeating-linear-gradient(
-                    transparent,
-                    transparent 23px,
-                    {colors['accent']} 23px,
-                    {colors['accent']} 24px
-                );
-            }}
-            .footer {{
-                margin-top: 30px;
-                text-align: center;
-                font-size: 12px;
-                color: #666;
-                border-top: 1px solid #ddd;
-                padding-top: 15px;
-            }}
-        </style>
-    </head>
-    <body>
-        <div class="header">
-            <div>
+        # Create HTML preview
+        html_content = f"""
+        <!DOCTYPE html>
+        <html>
+        <head>
+            <meta charset="UTF-8">
+            <title>{title}</title>
+            <style>
+                body {{
+                    font-family: Arial, sans-serif;
+                    margin: 20px;
+                    color: #333;
+                }}
+                .header {{
+                    text-align: center;
+                    color: {colors['primary']};
+                    border-bottom: 3px solid {colors['primary']};
+                    padding-bottom: 15px;
+                    margin-bottom: 30px;
+                }}
+                .title {{
+                    font-size: 24px;
+                    font-weight: bold;
+                    margin: 0;
+                }}
+                .subtitle {{
+                    font-size: 14px;
+                    color: {colors['secondary']};
+                    margin: 5px 0;
+                }}
+                .grid {{
+                    display: grid;
+                    grid-template-columns: repeat(7, 1fr);
+                    gap: 2px;
+                    border: 2px solid {colors['primary']};
+                    margin: 20px 0;
+                }}
+                .grid-header {{
+                    background: {colors['primary']};
+                    color: white;
+                    padding: 10px;
+                    text-align: center;
+                    font-weight: bold;
+                }}
+                .grid-cell {{
+                    border: 1px solid {colors['accent']};
+                    min-height: 80px;
+                    padding: 8px;
+                    background: #fafafa;
+                }}
+                .notes {{
+                    margin-top: 30px;
+                    padding: 20px;
+                    background: #f8f9fa;
+                    border-left: 5px solid {colors['accent']};
+                }}
+            </style>
+        </head>
+        <body>
+            <div class="header">
                 <h1 class="title">{title}</h1>
                 <p class="subtitle">Generated on {datetime.now().strftime('%B %d, %Y')}</p>
             </div>
-        </div>
+            
+            <div class="grid">
+        """
         
-        <div class="planner-grid">
-    """
-    
-    # Generate grid headers
-    if structure.get('layout_type') == 'weekly':
+        # Add grid headers and cells
         days = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday']
         for day in days:
-            html += f'<div class="grid-header">{day}</div>'
-    else:
-        for i in range(structure.get('columns', 7)):
-            html += f'<div class="grid-header">Column {i+1}</div>'
-    
-    # Generate grid cells
-    total_cells = structure.get('columns', 7) * structure.get('rows', 6)
-    for i in range(total_cells):
-        date_num = (i % 31) + 1 if structure.get('layout_type') == 'monthly' else ''
-        html += f'''
-        <div class="grid-cell">
-            <div class="date-number">{date_num}</div>
-            <div style="height: 70px;"></div>
-        </div>
-        '''
-    
-    html += f"""
-        </div>
-        
-        <div class="notes-section">
-            <div class="notes-title">Notes & Important Reminders</div>
-            <div class="notes-lines"></div>
-        </div>
+            html_content += f'<div class="grid-header">{day}</div>'
         
-        <div class="footer">
-            Created with AI Template Generator | {datetime.now().strftime('%Y')}
-        </div>
-    </body>
-    </html>
-    """
-    
-    return html
-
-def generate_pdf_from_html(html_content, customization):
-    """Generate PDF from HTML content"""
-    try:
-        # Create PDF using FPDF
-        colors = customization.get('colors', COLOR_SCHEMES["Professional Blue"])
-        title = customization.get('title', 'My Planner')
-        logo_path = customization.get('logo_path')
+        # Add grid cells (4 weeks)
+        for i in range(28):
+            html_content += f'<div class="grid-cell"><strong>{i+1}</strong></div>'
+            
+        html_content += f"""
+            </div>
+            
+            <div class="notes">
+                <h3 style="color: {colors['primary']}; margin-top: 0;">Notes & Reminders</h3>
+                <div style="height: 100px; background: white; border: 1px solid #ddd; padding: 10px;"></div>
+            </div>
+        </body>
+        </html>
+        """
         
-        pdf = EnhancedTemplatePDF(title=title, logo_path=logo_path, colors=colors)
+        # Create PDF
+        pdf = SimplePDF(title=title, colors=colors)
         pdf.add_page()
         
-        # Add main content (simplified for demo)
+        # Add content
         pdf.set_font('Arial', '', 12)
         pdf.set_text_color(0, 0, 0)
         
-        # Add planner grid simulation
+        # Simple grid for PDF
         y_start = 50
         cell_width = 25
         cell_height = 20
         
-        # Draw grid
-        for row in range(6):
+        # Draw weekly grid
+        for row in range(5):
             for col in range(7):
                 x = 10 + (col * cell_width)
                 y = y_start + (row * cell_height)
                 pdf.rect(x, y, cell_width, cell_height)
                 
-                # Add day labels for first row
-                if row == 0:
-                    days = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
-                    pdf.set_xy(x + 2, y + 2)
-                    pdf.cell(cell_width-4, cell_height-4, days[col], 0, 0, 'C')
+                if row == 0:  # Headers
+                    pdf.set_xy(x + 1, y + 1)
+                    pdf.set_font('Arial', 'B', 8)
+                    pdf.cell(cell_width-2, cell_height-2, days[col][:3], 0, 0, 'C')
+                    pdf.set_font('Arial', '', 8)
         
         # Add notes section
-        pdf.set_xy(10, y_start + 140)
+        pdf.set_xy(10, y_start + 120)
         pdf.set_font('Arial', 'B', 14)
         primary_rgb = tuple(int(colors["primary"][1:][i:i+2], 16) for i in (0, 2, 4))
         pdf.set_text_color(*primary_rgb)
-        pdf.cell(0, 10, 'Notes & Reminders', 0, 1, 'L')
-        
-        # Add lines for notes
-        pdf.set_text_color(0, 0, 0)
-        for i in range(8):
-            pdf.ln(6)
-            pdf.cell(0, 0, '_' * 80, 0, 1, 'L')
+        pdf.cell(0, 10, 'Notes & Important Reminders', 0, 1, 'L')
         
-        # Save to temp file
+        # Save PDF
         temp_dir = "temp_outputs"
         os.makedirs(temp_dir, exist_ok=True)
         pdf_path = os.path.join(temp_dir, f"template_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf")
         pdf.output(pdf_path)
         
-        return pdf_path
-        
-    except Exception as e:
-        print(f"PDF generation error: {e}")
-        return None
-
-def generate_template(prompt, template_preset, title, logo_file, primary_color, secondary_color, accent_color, font_style, color_scheme, include_logo, include_notes, page_size):
-    """Main function to generate customized template"""
-    
-    try:
-        # Use preset if selected
-        if template_preset and template_preset != "Custom":
-            prompt = TEMPLATE_PRESETS[template_preset]
-        
-        # Process customization
-        if color_scheme and color_scheme != "Custom":
-            colors = COLOR_SCHEMES[color_scheme]
-        else:
-            colors = {
-                "primary": primary_color,
-                "secondary": secondary_color,
-                "accent": accent_color
-            }
-        
-        # Process logo
-        logo_path = None
-        if include_logo and logo_file:
-            logo_path = process_logo_upload(logo_file)
-        
-        customization = {
-            'title': title or 'My Custom Planner',
-            'colors': colors,
-            'font_style': font_style,
-            'logo_path': logo_path,
-            'include_notes': include_notes,
-            'page_size': page_size
-        }
-        
-        # Generate AI structure
-        template_type = template_preset if template_preset != "Custom" else "General"
-        structure, ai_response = generate_ai_structure(prompt, template_type)
-        
-        if "error" in structure:
-            return f"<div style='color: red;'>Error: {structure['error']}</div>", None, f"AI Response: {ai_response}"
-        
-        # Create HTML preview
-        html_content = create_html_template(structure, customization)
-        
-        # Generate PDF
-        pdf_path = generate_pdf_from_html(html_content, customization)
-        
-        # Create success message
         success_msg = f"""
         <div style='padding: 20px; background: #f0f9ff; border-left: 4px solid {colors['primary']};'>
             <h3 style='color: {colors['primary']}; margin-top: 0;'>✅ Template Generated Successfully!</h3>
-            <p><strong>Title:</strong> {customization['title']}</p>
-            <p><strong>Type:</strong> {structure.get('layout_type', 'custom').title()}</p>
-            <p><strong>Dimensions:</strong> {structure.get('columns', 'N/A')} columns × {structure.get('rows', 'N/A')} rows</p>
-            <p><strong>Features:</strong> {', '.join(structure.get('features', []))}</p>
-            <p><strong>Font:</strong> {font_style}</p>
-            <p><strong>Colors:</strong> {color_scheme if color_scheme != 'Custom' else 'Custom Colors'}</p>
+            <p><strong>Title:</strong> {title}</p>
+            <p><strong>Description:</strong> {prompt}</p>
+            <p><strong>Color Scheme:</strong> {color_scheme}</p>
+            <p><strong>Generated at:</strong> {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}</p>
         </div>
         """
         
-        # Return preview, PDF file, and AI response
-        return html_content, pdf_path, f"AI Analysis: {ai_response}\n\nStructure: {json.dumps(structure, indent=2)}"
+        return html_content, pdf_path, success_msg
         
     except Exception as e:
-        error_msg = f"<div style='color: red; padding: 20px;'>❌ Error generating template: {str(e)}</div>"
+        error_msg = f"<div style='color: red; padding: 20px;'>❌ Error: {str(e)}</div>"
         return error_msg, None, f"Error details: {str(e)}"
 
 # Create Gradio interface
-with gr.Blocks(title="🎨 AI Template Generator Pro", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="🎨 AI Template Generator", theme=gr.themes.Default()) as demo:
     gr.Markdown("""
-    # 🎨 AI Template Generator Pro
-    ### Create Beautiful, Customizable Planners & Templates with AI
-    Simply describe what you need, customize the design, and download your professional template!
+    # 🎨 AI Template Generator
+    ### Create Beautiful Planners & Templates with AI
     """)
     
-    with gr.Tab("📝 Create Template"):
-        with gr.Row():
-            with gr.Column(scale=2):
-                gr.Markdown("### 📋 Template Description")
-                
-                template_preset = gr.Dropdown(
-                    choices=["Custom"] + list(TEMPLATE_PRESETS.keys()),
-                    value="Weekly Planner",
-                    label="🔖 Quick Presets",
-                    info="Choose a preset or select 'Custom' for your own description"
-                )
-                
-                prompt_input = gr.Textbox(
-                    label="📝 Describe Your Template",
-                    placeholder="Example: A4 weekly planner, portrait, Monday start, colorful design with habit tracker",
-                    lines=3,
-                    value=TEMPLATE_PRESETS["Weekly Planner"]
-                )
-                
-                def update_prompt(preset):
-                    if preset and preset != "Custom":
-                        return TEMPLATE_PRESETS[preset]
-                    return ""
-                
-                template_preset.change(update_prompt, template_preset, prompt_input)
+    with gr.Row():
+        with gr.Column(scale=2):
+            gr.Markdown("### 📝 Template Description")
             
-            with gr.Column(scale=1):
-                gr.Markdown("### 🎨 Customization")
-                
-                title_input = gr.Textbox(
-                    label="📌 Template Title",
-                    placeholder="My Personal Planner",
-                    value="My Weekly Planner"
-                )
-                
-                include_logo = gr.Checkbox(label="🖼️ Include Logo", value=False)
-                
-                logo_upload = gr.File(
-                    label="📁 Upload Logo",
-                    file_types=[".png", ".jpg", ".jpeg"],
-                    visible=False
-                )
-                
-                include_logo.change(lambda x: gr.update(visible=x), include_logo, logo_upload)
-        
-        with gr.Row():
-            with gr.Column():
-                gr.Markdown("### 🎨 Design Options")
-                
-                color_scheme = gr.Dropdown(
-                    choices=["Custom"] + list(COLOR_SCHEMES.keys()),
-                    value="Professional Blue",
-                    label="🌈 Color Scheme"
-                )
-                
-                with gr.Row():
-                    primary_color = gr.ColorPicker(label="Primary Color", value="#2563EB", visible=False)
-                    secondary_color = gr.ColorPicker(label="Secondary Color", value="#1E40AF", visible=False)
-                    accent_color = gr.ColorPicker(label="Accent Color", value="#3B82F6", visible=False)
-                
-                def toggle_custom_colors(scheme):
-                    is_custom = scheme == "Custom"
-                    return [gr.update(visible=is_custom)] * 3
-                
-                color_scheme.change(toggle_custom_colors, color_scheme, [primary_color, secondary_color, accent_color])
-                
-                font_style = gr.Dropdown(
-                    choices=list(FONT_STYLES.keys()),
-                    value="Modern",
-                    label="🔤 Font Style"
-                )
+            template_preset = gr.Dropdown(
+                choices=["Custom"] + list(TEMPLATE_PRESETS.keys()),
+                value="Weekly Planner",
+                label="🔧 Quick Presets"
+            )
             
-            with gr.Column():
-                gr.Markdown("### ⚙️ Additional Options")
-                
-                include_notes = gr.Checkbox(label="📝 Include Notes Section", value=True)
-                
-                page_size = gr.Dropdown(
-                    choices=["A4", "Letter", "A3", "Legal"],
-                    value="A4",
-                    label="📄 Page Size"
-                )
+            prompt_input = gr.Textbox(
+                label="📋 Describe Your Template",
+                placeholder="Describe what kind of planner you need...",
+                lines=3,
+                value=TEMPLATE_PRESETS["Weekly Planner"]
+            )
+            
+            def update_prompt(preset):
+                if preset and preset != "Custom":
+                    return TEMPLATE_PRESETS[preset]
+                return ""
+            
+            template_preset.change(update_prompt, template_preset, prompt_input)
         
-        generate_btn = gr.Button("🚀 Generate Template", variant="primary", size="lg")
-    
-    with gr.Tab("👀 Preview & Download"):
-        with gr.Row():
-            with gr.Column(scale=2):
-                preview_output = gr.HTML(label="📋 Template Preview")
+        with gr.Column(scale=1):
+            gr.Markdown("### 🎨 Customization")
             
-            with gr.Column(scale=1):
-                download_output = gr.File(label="📥 Download PDF")
-                ai_output = gr.Textbox(label="🤖 AI Analysis", lines=10)
+            title_input = gr.Textbox(
+                label="📌 Template Title",
+                placeholder="My Planner",
+                value="My Weekly Planner"
+            )
+            
+            color_scheme = gr.Dropdown(
+                choices=list(COLOR_SCHEMES.keys()),
+                value="Professional Blue",
+                label="🌈 Color Scheme"
+            )
+    
+    generate_btn = gr.Button("🚀 Generate Template", variant="primary", size="lg")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            preview_output = gr.HTML(label="📋 Template Preview")
+        
+        with gr.Column(scale=1):
+            download_output = gr.File(label="📥 Download PDF")
+            status_output = gr.HTML(label="📊 Generation Status")
     
     # Connect the generate button
     generate_btn.click(
-        generate_template,
-        inputs=[
-            prompt_input, template_preset, title_input, logo_upload,
-            primary_color, secondary_color, accent_color, font_style,
-            color_scheme, include_logo, include_notes, page_size
-        ],
-        outputs=[preview_output, download_output, ai_output]
+        create_simple_template,
+        inputs=[prompt_input, title_input, color_scheme],
+        outputs=[preview_output, download_output, status_output]
     )
     
     gr.Markdown("""
     ---
-    ### 💡 Tips for Best Results:
-    - **Be Specific**: Mention orientation (portrait/landscape), size (A4/Letter), and layout preferences
-    - **Include Details**: Specify time ranges, number of columns, special sections needed
-    - **Try Presets**: Use quick presets as starting points and modify as needed
-    - **Experiment**: Try different color schemes and fonts to match your style
-    
-    ### 🎯 Example Descriptions:
-    - "A4 monthly calendar, landscape, Sunday start, large date boxes, pastel colors, space for appointments"
-    - "Portrait daily planner, hourly time blocks 6 AM to 10 PM, habit tracker, meal planning section"
-    - "Weekly meal prep planner, grocery list section, nutritional goals tracker, clean modern design"
+    ### 💡 Tips:
+    - **Be specific**: Mention size (A4), orientation (portrait/landscape)
+    - **Include details**: Time ranges, number of columns, special sections
+    - **Try presets**: Use quick presets as starting points
     """)
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+    demo.launch()