gradio interface

app.py

@@ -50,39 +50,34 @@ def analyze_slide(slide_path, site_type):
 
     return aeon_results, paladin_results
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Mosaic: H&E Whole Slide Image Cancer Subtype and Biomarker Inference")
-    with gr.Row():
-        with gr.Column():
-            input_slide = gr.File(
-                label="Upload H&E Whole Slide Image",
-                file_types=[".svs", ".tiff", ".tif"],
-            )
-            input_dropdown = gr.Dropdown(
-                choices=["Primary", "Metastatic"],
-                label="Site Type",
-                value="Primary",
-            )
-            run_button = gr.Button("Run Inference")
-        with gr.Column():
-            aeon_output_table = gr.Dataframe(
-                headers=["Cancer Subtype", "Confidence"],
-                label="Cancer Subtype Inference",
-                datatype=["str", "number"],
-            )
-            paladin_output_table = gr.Dataframe(
-                headers=["Cancer Subtype", "Biomarker", "Score"],
-                label="Biomarker Inference",
-                datatype=["str", "str", "number"],
-            )
-
+if __name__ == "__main__":
+    input_slide = gr.File(
+        label="Upload H&E Whole Slide Image",
+        file_types=[".svs", ".tiff", ".tif"],
+    )
+    input_dropdown = gr.Dropdown(
+        choices=["Primary", "Metastatic"],
+        label="Site Type",
+        value="Primary",
+    )
+    aeon_output_table = gr.Dataframe(
+        headers=["Cancer Subtype", "Confidence"],
+        label="Cancer Subtype Inference",
+        datatype=["str", "number"],
+    )
+    paladin_output_table = gr.Dataframe(
+        headers=["Cancer Subtype", "Biomarker", "Score"],
+        label="Biomarker Inference",
+        datatype=["str", "str", "number"],
+    )
 
-    run_button.click(
-        analyze_slide,
+    iface = gr.Interface(
+        fn=analyze_slide,
         inputs=[input_slide, input_dropdown],
         outputs=[aeon_output_table, paladin_output_table],
-
+        title="Mosaic: H&E Whole Slide Image Cancer Subtype and Biomarker Inference",
+        description="Upload an H&E whole slide image and select the site type to infer cancer subtype and biomarkers.",
+        allow_flagging="never",
     )
 
-if __name__ == "__main__":
-    demo.launch(enable_queue=True)
+    iface.launch(enable_queue=True)