| import gradio as gr |
| import json |
| from datetime import datetime |
| import base64 |
| import tempfile |
| import os |
| from fpdf import FPDF |
| from PIL import Image |
|
|
| |
| TEMPLATE_PRESETS = { |
| "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 = { |
| "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"} |
| } |
|
|
| class SimplePDF(FPDF): |
| def __init__(self, title="My Planner", colors=None): |
| super().__init__() |
| self.title_text = title |
| self.colors = colors or COLOR_SCHEMES["Professional Blue"] |
| |
| def header(self): |
| self.set_font('Arial', 'B', 16) |
| |
| 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")}', 0, 0, 'C') |
|
|
| def create_simple_template(prompt, title, color_scheme): |
| """Create a simple template based on user input""" |
| try: |
| |
| colors = COLOR_SCHEMES.get(color_scheme, COLOR_SCHEMES["Professional Blue"]) |
| |
| |
| 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 class="grid"> |
| """ |
| |
| |
| days = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday'] |
| for day in days: |
| html_content += f'<div class="grid-header">{day}</div>' |
| |
| |
| 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 = SimplePDF(title=title, colors=colors) |
| pdf.add_page() |
| |
| |
| pdf.set_font('Arial', '', 12) |
| pdf.set_text_color(0, 0, 0) |
| |
| |
| y_start = 50 |
| cell_width = 25 |
| cell_height = 20 |
| |
| |
| 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) |
| |
| if row == 0: |
| 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) |
| |
| |
| 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 & Important Reminders', 0, 1, 'L') |
| |
| |
| 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) |
| |
| 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> {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 html_content, pdf_path, success_msg |
| |
| except Exception as e: |
| error_msg = f"<div style='color: red; padding: 20px;'>â Error: {str(e)}</div>" |
| return error_msg, None, f"Error details: {str(e)}" |
|
|
| |
| with gr.Blocks(title="ð¨ AI Template Generator", theme=gr.themes.Default()) as demo: |
| gr.Markdown(""" |
| # ð¨ AI Template Generator |
| ### Create Beautiful Planners & Templates with AI |
| """) |
| |
| 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" |
| ) |
| |
| 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) |
| |
| with gr.Column(scale=1): |
| gr.Markdown("### ð¨ Customization") |
| |
| 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") |
| |
| |
| generate_btn.click( |
| create_simple_template, |
| inputs=[prompt_input, title_input, color_scheme], |
| outputs=[preview_output, download_output, status_output] |
| ) |
| |
| gr.Markdown(""" |
| --- |
| ### ð¡ 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() |
