Create app.py

app.py

@@ -0,0 +1,568 @@
+import gradio as gr
+from transformers import pipeline
+import json
+import base64
+from datetime import datetime, timedelta
+import os
+from io import BytesIO
+from PIL import Image, ImageDraw, ImageFont
+import tempfile
+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
+
+# 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"
+}
+
+# Color schemes
+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"}
+}
+
+class EnhancedTemplatePDF(FPDF):
+    def __init__(self, title="", logo_path=None, colors=None, font_style="Modern"):
+        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)
+            
+    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')
+
+def process_logo_upload(logo_file):
+    """Process uploaded logo and return path"""
+    if logo_file is None:
+        return None
+        
+    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
+        
+    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>
+                <h1 class="title">{title}</h1>
+                <p class="subtitle">Generated on {datetime.now().strftime('%B %d, %Y')}</p>
+            </div>
+        </div>
+        
+        <div class="planner-grid">
+    """
+    
+    # Generate grid headers
+    if structure.get('layout_type') == 'weekly':
+        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>
+        
+        <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')
+        
+        pdf = EnhancedTemplatePDF(title=title, logo_path=logo_path, colors=colors)
+        pdf.add_page()
+        
+        # Add main content (simplified for demo)
+        pdf.set_font('Arial', '', 12)
+        pdf.set_text_color(0, 0, 0)
+        
+        # Add planner grid simulation
+        y_start = 50
+        cell_width = 25
+        cell_height = 20
+        
+        # Draw grid
+        for row in range(6):
+            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')
+        
+        # Add notes section
+        pdf.set_xy(10, y_start + 140)
+        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')
+        
+        # Save to temp file
+        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>
+        </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)}"
+        
+    except Exception as e:
+        error_msg = f"<div style='color: red; padding: 20px;'>❌ Error generating template: {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:
+    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!
+    """)
+    
+    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.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"
+                )
+            
+            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"
+                )
+        
+        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):
+                download_output = gr.File(label="📥 Download PDF")
+                ai_output = gr.Textbox(label="🤖 AI Analysis", lines=10)
+    
+    # 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]
+    )
+    
+    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"
+    """)
+
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=True
+    )