import gradio as gr
import numpy as np
from fpdf import FPDF
import os
import tempfile

def disease_report(img, scan_for, generate_report):
    results = []
    for i, mode in enumerate(["Red", "Green", "Blue"]):
        color_filter = np.array([0, 0, 0])
        color_filter[i] = 1
        results.append([mode, img * color_filter])
    tmp_dir = tempfile.gettempdir()
    report = os.path.join(tmp_dir, "report.pdf")
    if generate_report:
        pdf = FPDF()
        pdf.add_page()
        pdf.set_font("Arial", size=15)
        pdf.cell(200, 10, txt="Disease Report", 
                ln=1, align='C')
        pdf.cell(200, 10, txt="A Gradio Demo.",
                ln=2, align='C')
        pdf.output(report)
    return results, report if generate_report else None

iface = gr.Interface(disease_report, 
    [
        "image", 
        gr.inputs.CheckboxGroup(["Cancer", "Rash", "Heart Failure", "Stroke", "Diabetes", "Pneumonia"]),
        "checkbox"
    ],
    [
        gr.outputs.Carousel(["text", "image"], label="Disease"),
        gr.outputs.File(label="Report")
    ],
    title="Disease Report",
    description="Upload an Xray and select the diseases to scan for.",
    theme="grass",
    flagging_options=["good", "bad", "etc"],
    allow_flagging="auto"
)

if __name__ == "__main__":
    iface.launch()