update gradio output

app.py

@@ -55,39 +55,40 @@ IMAGE_PROMPT="Are these cells healthy or cancerous?"
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(INTRO_TEXT)
-    with gr.Tab("Agentic Detection"):
+    with gr.Tab("Segment/Detect"):
         with gr.Row():
             with gr.Column():
-                image = gr.Image(type="numpy")
+                image = gr.Image(type="pil")
+                seg_input = gr.Text(label="Entities to Segment/Detect")
+        
             with gr.Column():
-                text_input = gr.Text(label="Input Text")
-                text_output = gr.Text(label="Text Output")
-        chat_btn = gr.Button()
+                annotated_image = gr.AnnotatedImage(label="Output")
 
-        chat_inputs = [
-            image
-        ]
-        chat_outputs = [
-            text_output
-        ]
-        chat_btn.click(
-            fn=detect_brain_tumor,
-            inputs=chat_inputs,
-            outputs=chat_outputs,
-        )
-        
-        examples = [["./examples/194_jpg.rf.3e3dd592d034bb5ee27a978553819f42.jpg"],
-                    ["./examples/239_jpg.rf.3dcc0799277fb78a2ab21db7761ccaeb.jpg"],
-                    ["./examples/1385_jpg.rf.3c67cb92e2922dba0e6dba86f69df40b.jpg"],
-                    ["./examples/1491_jpg.rf.3c658e83538de0fa5a3f4e13d7d85f12.jpg"],
-                    ["./examples/1550_jpg.rf.3d067be9580ec32dbee5a89c675d8459.jpg"],
-                    ["./examples/2256_jpg.rf.3afd7903eaf3f3c5aa8da4bbb928bc19.jpg"],
-                    ["./examples/2871_jpg.rf.3b6eadfbb369abc2b3bcb52b406b74f2.jpg"],
-                    ["./examples/2921_jpg.rf.3b952f91f27a6248091e7601c22323ad.jpg"],
+        seg_btn = gr.Button("Submit")    
+        examples = [["./examples/194_jpg.rf.3e3dd592d034bb5ee27a978553819f42.jpg", "detect brain tumor"],
+                    ["./examples/239_jpg.rf.3dcc0799277fb78a2ab21db7761ccaeb.jpg", "detect brain tumor"],
+                    ["./examples/1385_jpg.rf.3c67cb92e2922dba0e6dba86f69df40b.jpg", "detect brain tumor"],
+                    ["./examples/1491_jpg.rf.3c658e83538de0fa5a3f4e13d7d85f12.jpg", "detect brain tumor"],
+                    ["./examples/1550_jpg.rf.3d067be9580ec32dbee5a89c675d8459.jpg", "detect brain tumor"],
+                    ["./examples/2256_jpg.rf.3afd7903eaf3f3c5aa8da4bbb928bc19.jpg", "detect brain tumor"],
+                    ["./examples/2871_jpg.rf.3b6eadfbb369abc2b3bcb52b406b74f2.jpg", "detect brain tumor"],
+                    ["./examples/2921_jpg.rf.3b952f91f27a6248091e7601c22323ad.jpg", "detect brain tumor"],
                     ]
         gr.Examples(
             examples=examples,
-            inputs=chat_inputs,
+            inputs=[image, seg_input],
+        )
+        seg_inputs = [
+            image,
+            seg_input
+            ]
+        seg_outputs = [
+            annotated_image
+        ]
+        seg_btn.click(
+            fn=detect_brain_tumor,
+            inputs=seg_inputs,
+            outputs=seg_outputs,
         )
 
 if __name__ == "__main__":