from io import BytesIO
from reportlab.lib.pagesizes import A4
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.units import inch
from reportlab.lib import colors
from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle

def export_blast_pdf(result: dict, sequence: str = "") -> bytes:
    buffer = BytesIO()
    doc = SimpleDocTemplate(buffer, pagesize=A4, topMargin=0.75 * inch, bottomMargin=0.75 * inch)
    styles = getSampleStyleSheet()
    elements = []

    title_style = ParagraphStyle("Title2", parent=styles["Title"], fontSize=16, spaceAfter=12)
    heading_style = ParagraphStyle("Heading", parent=styles["Heading2"], fontSize=12, spaceAfter=6, textColor=colors.HexColor("#16a34a"))
    body_style = ParagraphStyle("Body", parent=styles["BodyText"], fontSize=10, spaceAfter=6)

    elements.append(Paragraph("Bio Nexus \u2014 BLAST Results", title_style))
    elements.append(Spacer(1, 12))

    if sequence:
        preview = sequence[:100] + ("..." if len(sequence) > 100 else "")
        elements.append(Paragraph(f"<b>Query:</b> {preview}", body_style))

    elements.append(Paragraph(f"<b>Hits found:</b> {result.get('count', 0)}", body_style))
    elements.append(Paragraph(f"<b>Provider:</b> {result.get('source') or 'not recorded'}", body_style))
    elements.append(Paragraph(f"<b>Requested database:</b> {result.get('requested_database') or result.get('database') or 'not recorded'}", body_style))
    elements.append(Paragraph(f"<b>Executed database:</b> {result.get('executed_database') or result.get('database') or 'not recorded'}", body_style))
    elements.append(Spacer(1, 12))

    hits = result.get("hits", [])
    if not hits:
        elements.append(Paragraph("No hits found.", body_style))
        doc.build(elements)
        return buffer.getvalue()

    table_data = [["#", "Accession", "Description", "E-value", "Identity", "Coverage"]]
    for i, hit in enumerate(hits[:20], 1):
        desc = (hit.get("description", "")[:60] + "...") if len(hit.get("description", "")) > 60 else hit.get("description", "")
        table_data.append([
            str(i),
            hit.get("accession", ""),
            desc,
            f"{hit.get('evalue', 0):.1e}",
            f"{hit.get('identity_pct', 0)}%",
            f"{hit['query_coverage_pct']}%" if hit.get('query_coverage_pct') is not None else "not available",
        ])

    col_widths = [0.4 * inch, 1.0 * inch, 2.5 * inch, 1.0 * inch, 0.8 * inch, 0.8 * inch]
    t = Table(table_data, colWidths=col_widths, repeatRows=1)
    t.setStyle(TableStyle([
        ("BACKGROUND", (0, 0), (-1, 0), colors.HexColor("#16a34a")),
        ("TEXTCOLOR", (0, 0), (-1, 0), colors.white),
        ("FONTSIZE", (0, 0), (-1, 0), 9),
        ("FONTSIZE", (0, 1), (-1, -1), 8),
        ("FONTNAME", (0, 0), (-1, 0), "Helvetica-Bold"),
        ("ALIGN", (0, 0), (-1, -1), "LEFT"),
        ("VALIGN", (0, 0), (-1, -1), "TOP"),
        ("GRID", (0, 0), (-1, -1), 0.5, colors.HexColor("#e5e7eb")),
        ("ROWBACKGROUNDS", (0, 1), (-1, -1), [colors.white, colors.HexColor("#f9fafb")]),
        ("TOPPADDING", (0, 0), (-1, -1), 4),
        ("BOTTOMPADDING", (0, 0), (-1, -1), 4),
        ("LEFTPADDING", (0, 0), (-1, -1), 4),
        ("RIGHTPADDING", (0, 0), (-1, -1), 4),
    ]))
    elements.append(t)
    elements.append(Spacer(1, 16))

    elements.append(Paragraph("Interpretation", heading_style))
    elements.append(Paragraph("E-values depend on the searched database, program and search space. Evaluate each match with its bitscore, identity, query coverage and the cutoff selected for this analysis.", body_style))
    elements.append(Spacer(1, 8))
    elements.append(Paragraph(f"<i>Generated by Bio Nexus Platform \u2014 {result.get('count', 0)} hits</i>", body_style))

    doc.build(elements)
    return buffer.getvalue()


def export_uniprot_pdf(data: dict) -> bytes:
    buffer = BytesIO()
    doc = SimpleDocTemplate(buffer, pagesize=A4, topMargin=0.75 * inch, bottomMargin=0.75 * inch)
    styles = getSampleStyleSheet()
    elements = []

    title_style = ParagraphStyle("Title2", parent=styles["Title"], fontSize=16, spaceAfter=12)
    heading_style = ParagraphStyle("Heading", parent=styles["Heading2"], fontSize=12, spaceAfter=6, textColor=colors.HexColor("#16a34a"))
    body_style = ParagraphStyle("Body", parent=styles["BodyText"], fontSize=10, spaceAfter=6)

    elements.append(Paragraph("Bio Nexus \u2014 UniProt Annotation", title_style))
    elements.append(Spacer(1, 12))

    elements.append(Paragraph(f"<b>Protein:</b> {data.get('full_name', 'Unknown')}", body_style))
    elements.append(Paragraph(f"<b>Accession:</b> {data.get('accession', '')}", body_style))
    elements.append(Paragraph(f"<b>Organism:</b> {data.get('organism', 'Unknown')}", body_style))

    genes = data.get("gene_names", [])
    if genes:
        elements.append(Paragraph(f"<b>Gene:</b> {', '.join(genes)}", body_style))

    ec = data.get("ec_number", "")
    if ec:
        elements.append(Paragraph(f"<b>EC Number:</b> {ec}", body_style))

    elements.append(Paragraph(f"<b>Sequence Length:</b> {data.get('sequence_length', 0)} amino acids", body_style))
    elements.append(Spacer(1, 12))

    functions = data.get("functions", [])
    if functions:
        elements.append(Paragraph("Function", heading_style))
        for f in functions:
            elements.append(Paragraph(f, body_style))

    locations = data.get("subcellular_locations", [])
    if locations:
        elements.append(Paragraph("Subcellular Location", heading_style))
        elements.append(Paragraph(", ".join(locations), body_style))

    pdb_ids = data.get("pdb_ids", [])
    if pdb_ids:
        elements.append(Paragraph("PDB Structures", heading_style))
        elements.append(Paragraph(", ".join(pdb_ids), body_style))

    keywords = data.get("keywords", [])
    if keywords:
        elements.append(Paragraph("Keywords", heading_style))
        elements.append(Paragraph(", ".join(keywords[:10]), body_style))

    elements.append(Spacer(1, 12))
    elements.append(Paragraph("<i>Generated by Bio Nexus Platform</i>", body_style))

    doc.build(elements)
    return buffer.getvalue()