Paper2Poster/PosterAgent/gen_beamer_code.py

import re
import json
import os
from typing import List, Dict, Any

def sanitize_for_latex(name):
    """Convert any character that is not alphanumeric into underscore for LaTeX compatibility."""
    return re.sub(r'[^0-9a-zA-Z_]+', '_', name)

def initialize_beamer_document(width_cm=120, height_cm=90, theme="default"):
    """
    Initialize a Beamer document with specified dimensions and theme.
    
    Args:
        width_cm: Width in centimeters (default 120cm for poster)
        height_cm: Height in centimeters (default 90cm for poster) 
        theme: Beamer theme name (default, Madrid, Warsaw, etc.)
    """
    code = f'''\\documentclass[aspectratio=169]{{beamer}}
\\usepackage[utf8]{{inputenc}}
\\usepackage[T1]{{fontenc}}
\\usepackage{{graphicx}}
\\usepackage{{tikz}}
\\usepackage{{xcolor}}
\\usepackage{{geometry}}
\\usepackage{{multicol}}
\\usepackage{{array}}
\\usepackage{{booktabs}}
\\usepackage{{adjustbox}}

% Set page dimensions for poster
\\geometry{{paperwidth={width_cm}cm, paperheight={height_cm}cm, margin=1cm}}

% Beamer theme
\\usetheme{{{theme}}}
\\usecolortheme{{default}}

% Custom colors
\\definecolor{{titlecolor}}{{RGB}}{{47, 85, 151}}
\\definecolor{{textcolor}}{{RGB}}{{0, 0, 0}}
\\definecolor{{bgcolor}}{{RGB}}{{255, 255, 255}}

% Remove navigation symbols
\\setbeamertemplate{{navigation symbols}}{{}}

% Custom title page
\\setbeamertemplate{{title page}}{{
    \\begin{{center}}
        \\vspace{{1cm}}
        {{\\color{{titlecolor}}\\Huge\\textbf{{\\inserttitle}}}}
        \\vspace{{0.5cm}}
        \\Large{{\\insertauthor}}
        \\vspace{{0.3cm}}
        \\normalsize{{\\insertinstitute}}
    \\end{{center}}
}}

% Custom frame title
\\setbeamertemplate{{frametitle}}{{
    \\vspace{{0.5cm}}
    \\begin{{flushleft}}
        {{\\color{{titlecolor}}\\Large\\textbf{{\\insertframetitle}}}}
    \\end{{flushleft}}
    \\vspace{{0.3cm}}
}}

\\begin{{document}}

% Title frame
\\title{{POSTER_TITLE_PLACEHOLDER}}
\\author{{POSTER_AUTHOR_PLACEHOLDER}}
\\institute{{POSTER_INSTITUTE_PLACEHOLDER}}
\\date{{\\today}}

\\begin{{frame}}[plain]
    \\titlepage
\\end{{frame}}

'''
    return code

def generate_beamer_section_code(section_data: Dict[str, Any], section_index: int):
    """
    兼容 Paper2Poster bullet JSON:
    - section_data 包含 title_blocks / textbox1_blocks / textbox2_blocks
    - 每个 *_blocks 是 list[ {bullet: bool, runs: [{text: str, ...}], ...} ]
    """
    def blocks_to_lines(blocks):
        """把 blocks 转成 list[str]，并标注是否 bullet"""
        lines = []
        for blk in blocks or []:
            text = " ".join([r.get("text","") for r in blk.get("runs", [])]).strip()
            if not text:
                continue
            lines.append({
                "text": text,
                "bullet": bool(blk.get("bullet", False))
            })
        return lines

    # Frame title 优先用 title_blocks 的文本，否则用 title_str，否则 Untitled
    if isinstance(section_data.get("title_blocks"), list) and section_data["title_blocks"]:
        frame_title = " ".join([r.get("text","") for r in section_data["title_blocks"][0].get("runs", [])]).strip()
    else:
        frame_title = section_data.get("title_str") or "Untitled"

    frame_title = frame_title.replace("{","\\{").replace("}","\\}")  # 简单转义以防标题含花括号

    code = f"\n% ===== Section {section_index} =====\n"
    code += f"\\begin{{frame}}[t]{{{frame_title}}}\n"
    code += "  \\vspace{-0.5cm}\n"

    for key in ["textbox1_blocks", "textbox2_blocks"]:
        lines = blocks_to_lines(section_data.get(key, []))
        if not lines:
            continue

        # 如果全是 bullet，就合并成一个 itemize；否则分别处理
        if all(l["bullet"] for l in lines):
            code += "  \\begin{itemize}\n"
            for l in lines:
                code += f"    \\item {l['text']}\n"
            code += "  \\end{itemize}\n"
        else:
            for l in lines:
                if l["bullet"]:
                    code += f"  \\begin{{itemize}}\\item {l['text']}\\end{{itemize}}\n"
                else:
                    code += f"  {l['text']}\\\\\n"

    code += "\\end{frame}\n\n"
    return code



def generate_beamer_figure_code(figure_data: Dict[str, Any], figure_index: int):
    """
    Generate Beamer code for including figures.
    
    Args:
        figure_data: Dictionary containing figure information
        figure_index: Index of the figure
    """
    figure_name = sanitize_for_latex(figure_data.get('figure_name', f'figure_{figure_index}'))
    figure_path = figure_data.get('figure_path', '')
    
    # Convert inches to centimeters (1 inch = 2.54 cm)
    width_cm = figure_data.get('width', 10) * 2.54
    height_cm = figure_data.get('height', 8) * 2.54
    
    code = f'''
% Figure: {figure_name}
\\begin{{frame}}[t]{{{figure_data.get('title', 'Figure')}}}
    \\vspace{{-0.5cm}}
    \\begin{{center}}
        \\includegraphics[width={width_cm:.2f}cm, height={height_cm:.2f}cm]{{{figure_path}}}
    \\end{{center}}
    \\vspace{{0.3cm}}
    \\begin{{center}}
        \\small{{\\textbf{{{figure_data.get('caption', 'Figure Caption')}}}}}
    \\end{{center}}
\\end{{frame}}

'''
    return code

def generate_beamer_poster_code(
    sections: List[Dict[str, Any]],
    figures: List[Dict[str, Any]],
    poster_info: Dict[str, str],
    width_cm: float = 120,
    height_cm: float = 90,
    theme: str = "default",
    output_path: str = "poster.tex"
):
    """
    Generate complete Beamer poster code.
    
    Args:
        sections: List of section dictionaries
        figures: List of figure dictionaries  
        poster_info: Dictionary with title, author, institute
        width_cm: Poster width in centimeters
        height_cm: Poster height in centimeters
        theme: Beamer theme name
        output_path: Output .tex file path
    """
    code = initialize_beamer_document(width_cm, height_cm, theme)
    
    # Replace placeholders with actual content
    code = code.replace('POSTER_TITLE_PLACEHOLDER', poster_info.get('title', 'Poster Title'))
    code = code.replace('POSTER_AUTHOR_PLACEHOLDER', poster_info.get('author', 'Author Name'))
    code = code.replace('POSTER_INSTITUTE_PLACEHOLDER', poster_info.get('institute', 'Institute Name'))
    
    # Add sections
    for i, section in enumerate(sections):
        code += generate_beamer_section_code(section, i)
    
    # Add figures
    for i, figure in enumerate(figures):
        code += generate_beamer_figure_code(figure, i)
    
    # Close document
    code += '''
\\end{document}
'''
    
    return code

def save_beamer_code(code: str, output_path: str):
    """Save Beamer code to file."""
    with open(output_path, 'w', encoding='utf-8') as f:
        f.write(code)

def compile_beamer_to_pdf(tex_path: str, output_dir: str = "."):
    """
    Compile Beamer .tex file to PDF using pdflatex.
    
    Args:
        tex_path: Path to .tex file
        output_dir: Output directory for PDF
    """
    import subprocess
    
    try:
        # Run pdflatex twice for proper cross-references
        result1 = subprocess.run(
            ['pdflatex', '-output-directory', output_dir, tex_path],
            capture_output=True,
            text=True,
            timeout=60
        )
        
        result2 = subprocess.run(
            ['pdflatex', '-output-directory', output_dir, tex_path],
            capture_output=True,
            text=True,
            timeout=60
        )
        
        if result1.returncode == 0 and result2.returncode == 0:
            print(f"Successfully compiled {tex_path} to PDF")
            return True
        else:
            print(f"Error compiling {tex_path}:")
            print(result1.stderr)
            print(result2.stderr)
            return False
            
    except subprocess.TimeoutExpired:
        print(f"Timeout while compiling {tex_path}")
        return False
    except Exception as e:
        print(f"Error compiling {tex_path}: {e}")
        return False

# Example usage functions
def convert_pptx_layout_to_beamer(pptx_layout_data: Dict[str, Any]) -> Dict[str, Any]:
    """
    Convert PowerPoint layout data to Beamer-compatible format.
    
    Args:
        pptx_layout_data: Layout data from PowerPoint generation
    """
    beamer_data = {
        'sections': [],
        'figures': [],
        'poster_info': {
            'title': 'Default Title',
            'author': 'Default Author', 
            'institute': 'Default Institute'
        }
    }
    
    # Convert text arrangements to sections
    if 'text_arrangement' in pptx_layout_data:
        for i, text_item in enumerate(pptx_layout_data['text_arrangement']):
            section = {
                'section_name': text_item.get('textbox_name', f'section_{i}'),
                'title': text_item.get('title', f'Section {i+1}'),
                'content': text_item.get('content', 'Content placeholder')
            }
            beamer_data['sections'].append(section)
    
    # Convert figure arrangements to figures
    if 'figure_arrangement' in pptx_layout_data:
        for i, figure_item in enumerate(pptx_layout_data['figure_arrangement']):
            figure = {
                'figure_name': figure_item.get('figure_name', f'figure_{i}'),
                'figure_path': figure_item.get('figure_path', ''),
                'width': figure_item.get('width', 10),
                'height': figure_item.get('height', 8),
                'title': figure_item.get('title', f'Figure {i+1}'),
                'caption': figure_item.get('caption', 'Figure caption')
            }
            beamer_data['figures'].append(figure)
    
    return beamer_data