mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-27 02:30:17 +08:00
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
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="compact",
|
|
flagging_options=["good", "bad", "etc"],
|
|
allow_flagging="auto"
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
iface.launch()
|