Update app.py

app.py

@@ -78,39 +78,25 @@ def masking(text, task):
     image = return_species_image(new_species)
     return text, image
 
-def plantbert(text, typology, confidence, task):
-    if task == "classification":
-        formatted_output, image_output = classification(text, typology, confidence, task)
-    else:
-        formatted_output, image_output = masking(text, task)
-    return formatted_output, image_output
-
-inputs=[
-    gr.Textbox(lines=2, label="Species", placeholder="Enter a list of comma-separated binomial names here."),
-    gr.Dropdown(["EUNIS"], value="EUNIS", label="Typology", info="Will add more typologies later!"),
-    gr.Slider(0, 100, value=90, label="Confidence", info="Choose the level of confidence for the prediction."),
-    gr.Radio(["classification", "masking"], value="classification", label="Task", info="Which task to choose?")
-]
-
-outputs=[
-    gr.Textbox(lines=2, label="Vegetation Plot Classification Result"),
-    "image"
-]
-
-title="Pl@ntBERT"
-
-description="Vegetation Plot Classification: enter the species found in a vegetation plot and see its EUNIS habitat!"
-
-examples=[
-    ["sparganium erectum, calystegia sepium, persicaria amphibia", "EUNIS", 90, "classification"],
-    ["vaccinium myrtillus, dryopteris dilatata, molinia caerulea", "EUNIS", 90, "masking"]
-]
+with gr.Blocks() as demo:
+    gr.Markdown("Classify vegetation plots or find the missing species.")
+    with gr.Tab("Vegetation plot classification"):
+        species = gr.Textbox(lines=2, label="Species", placeholder="Enter a list of comma-separated binomial names here.")
+        typology = gr.Dropdown(["EUNIS"], value="EUNIS", label="Typology", info="Will add more typologies later!")
+        confidence = gr.Slider(0, 100, value=90, label="Confidence", info="Choose the level of confidence for the prediction.")
+        task = gr.Radio(["classification", "masking"], value="classification", label="Task", info="Which task to choose?")
+        text_output_1 = gr.Textbox()
+        text_output_2 = gr.Image()
+        text_button = gr.Button("Classify")
+    with gr.Tab("Missing species finding"):
+        with gr.Row():
+            species_2 = gr.Textbox(lines=2, label="Species", placeholder="Enter a list of comma-separated binomial names here.")
+            task_2 = gr.Radio(["classification", "masking"], value="classification", label="Task", info="Which task to choose?")
+            image_output_1 = gr.Textbox()
+            image_output_2 = gr.Image()
+        image_button = gr.Button("Find")
 
-io = gr.Interface(fn=plantbert, 
-                         inputs=inputs, 
-                         outputs=outputs,
-                         title=title,
-                         description=description,
-                         examples=examples)
+    text_button.click(classification, inputs=[species, typology, confidence, task], outputs=[text_output_1, text_output_2]])
+    image_button.click(masking, inputs=[species_2, task_2], outputs=[image_output_1, image_output_2]])
 
-io.launch()
+demo.launch()