Update app.py

app.py

@@ -42,14 +42,33 @@ def generate(image, prompt):
 with gr.Blocks() as demo:
     gr.Markdown(title)
 
-    with gr.Row():
-        with gr.Column():
-            image_input = gr.Image(type="pil")
-            prompt_input = gr.Textbox(label="Prompt")
-        with gr.Column():
-            output_text = gr.Textbox(label="Response")
-
-    generate_button = gr.Button("Generate")
-    generate_button.click(fn=generate, inputs=[image_input, prompt_input], outputs=output_text)
+    with gr.Accordion("Custom Prompt Analysis"):
+        with gr.Row():
+            image_input_custom = gr.Image(type="pil")
+            prompt_input_custom = gr.Textbox(label="Enter your custom prompt")
+        generate_button_custom = gr.Button("Generate")
+        output_text_custom = gr.Textbox(label="Response")
+        generate_button_custom.click(fn=generate, inputs=[image_input_custom, prompt_input_custom], outputs=output_text_custom)
+
+    with gr.Accordion("Anatomical Feature Analysis"):
+        anatomies = [
+            "Airway", "Breathing", "Cardiac", "Diaphragm",
+            "Everything else (e.g., mediastinal contours, bones, soft tissues, tubes, valves, and pacemakers)"
+        ]
+        with gr.Row():
+            image_input_feature = gr.Image(type="pil")
+            prompt_select = gr.Dropdown(label="Select an anatomical feature", choices=anatomies)
+        generate_button_feature = gr.Button("Analyze Feature")
+        output_text_feature = gr.Textbox(label="Response")
+        generate_button_feature.click(fn=lambda image, feature: generate(image, f'Describe "{feature}"'), inputs=[image_input_feature, prompt_select], outputs=output_text_feature)
+
+    with gr.Accordion("Common Abnormalities Analysis"):
+        common_abnormalities = ["Lung Nodule", "Pleural Effusion", "Pneumonia"]
+        with gr.Row():
+            image_input_abnormality = gr.Image(type="pil")
+            abnormality_select = gr.Dropdown(label="Select a common abnormality", choices=common_abnormalities)
+        generate_button_abnormality = gr.Button("Analyze Abnormality")
+        output_text_abnormality = gr.Textbox(label="Response")
+        generate_button_abnormality.click(fn=lambda image, abnormality: generate(image, f'Analyze for "{abnormality}"'), inputs=[image_input_abnormality, abnormality_select], outputs=output_text_abnormality)
 
 demo.launch()