mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
import os
|
|
import tempfile
|
|
|
|
import numpy as np
|
|
from fpdf import FPDF
|
|
|
|
import gradio as gr
|
|
|
|
|
|
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()
|