Update app.py

app.py

@@ -20,6 +20,8 @@ Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder
 
 device = "cuda"
 dtype = torch.float16
+example_images = ["00000174_003.png", "00006596_000.png", "00006663_000.png", 
+                  "00012976_002.png", "00018401_000.png", "00019799_000.png"]
 
 processor = AutoProcessor.from_pretrained("StanfordAIMI/CheXagent-8b", trust_remote_code=True)
 generation_config = GenerationConfig.from_pretrained("StanfordAIMI/CheXagent-8b")
@@ -41,13 +43,12 @@ def generate(image, prompt):
 
 with gr.Blocks() as demo:
     gr.Markdown(title)
-
-    with gr.Accordion("Custom Prompt Analysis"):
+    with gr.Accordion("Custom Prompt Analysis", open=False):
         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 = gr.Button("Generate")
+            output_text_custom = gr.Textbox(label="Response")
 
         def custom_generate(image, prompt):
             if isinstance(image, str) and os.path.exists(image):
@@ -57,18 +58,24 @@ with gr.Blocks() as demo:
                 return generate(image, prompt)
 
         generate_button_custom.click(fn=custom_generate, inputs=[image_input_custom, prompt_input_custom], outputs=output_text_custom)
+        custom_prompt_examples = [
+            [os.path.join(os.path.dirname(__file__), img), "You are an expert X-Ray Analyst, describe this chest x-ray in detail focussing on the lung condition:"]
+            for img in example_images
+            ]
+    
+        # example_prompt = "65 y/m Chronic cough and weight loss x 6 months. Chest X-rays normal. Consulted multiple pulmonologists with not much benefit. One wise pulmonologist thinks of GERD and sends him to the Gastro department. Can you name the classical finding here?"
+        # example_image_path = os.path.join(os.path.dirname(__file__), "hegde.jpg")
+        with gr.Accordion("Examples", open=False):
 
-        example_prompt = "65 y/m Chronic cough and weight loss x 6 months. Chest X-rays normal. Consulted multiple pulmonologists with not much benefit. One wise pulmonologist thinks of GERD and sends him to the Gastro department. Can you name the classical finding here?"
-        example_image_path = os.path.join(os.path.dirname(__file__), "hegde.jpg")
-        gr.Examples(
-            examples=[[example_image_path, example_prompt]],
-            inputs=[image_input_custom, prompt_input_custom],
-            outputs=[output_text_custom],
-            fn=custom_generate,
-            cache_examples=True
-        )
+            gr.Examples(
+                examples=custom_prompt_examples,
+                inputs=[image_input_custom, prompt_input_custom],
+                outputs=[output_text_custom],
+                fn=custom_generate,
+                cache_examples=True
+            )
     
-    with gr.Accordion("Anatomical Feature Analysis"):
+    with gr.Accordion("Anatomical Feature Analysis", open=False):
         anatomies = [
             "Airway", "Breathing", "Cardiac", "Diaphragm",
             "Everything else (e.g., mediastinal contours, bones, soft tissues, tubes, valves, and pacemakers)"
@@ -79,8 +86,20 @@ with gr.Blocks() as demo:
         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)
+        anatomical_feature_examples = [
+            [os.path.join(os.path.dirname(__file__), img), "Airway"]
+            for img in example_images
+        ]
+        with gr.Accordion("Examples", open=False):
+            gr.Examples(
+                examples=anatomical_feature_examples,
+                inputs=[image_input_feature, prompt_select],
+                outputs=[output_text_feature],
+                fn=lambda image, feature: generate(image, f'Describe "{feature}"'),
+                cache_examples=True
+            )
 
-    with gr.Accordion("Common Abnormalities Analysis"):
+    with gr.Accordion("Common Abnormalities Analysis", open=False):
         common_abnormalities = ["Lung Nodule", "Pleural Effusion", "Pneumonia"]
         with gr.Row():
             image_input_abnormality = gr.Image(type="pil")
@@ -88,5 +107,16 @@ with gr.Blocks() as demo:
         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)
-
+        common_abnormalities_examples = [
+            [os.path.join(os.path.dirname(__file__), img), "Lung Nodule"]
+            for img in example_images
+        ]
+        with gr.Accordion("Examples", open=False):
+            gr.Examples(
+                examples=common_abnormalities_examples,
+                inputs=[image_input_abnormality, abnormality_select],
+                outputs=[output_text_abnormality],
+                fn=lambda image, abnormality: generate(image, f'Analyze for "{abnormality}"'),
+                cache_examples=True
+            )
 demo.launch()