Update app.py

app.py

@@ -1,73 +1,16 @@
 import gradio as gr
 import json
+from pathlib import Path
 
-# Include your JSON data directly in the script
-data = [
-    {
-        "Note": "The image shows a new triangular opacity in the right lung apex, which indicates an abnormality in that area. The fullness of the right hilum, which is the area where the main bronchus, blood vessels, and nerves enter and exit the lung, is also new. The remainder of the lungs appears to be clear. The heart size is within the normal range. The blunting of bilateral costophrenic angles, with the right greater than the left, may be secondary to small effusions, which are fluid collections in the pleural space surrounding the lungs.",
-        "Summary": {
-            "chest_xray_summary": [
-                {
-                    "topic": "Right Lung Apex",
-                    "description": "A new triangular opacity is observed in the right lung apex, indicating an abnormality in this area."
-                },
-                {
-                    "topic": "Right Hilum",
-                    "description": "There is fullness of the right hilum, which is a new finding. The hilum is where the main bronchus, blood vessels, and nerves enter and exit the lung."
-                },
-                {
-                    "topic": "Lung Parenchyma",
-                    "description": "The remainder of the lungs appears clear, aside from the noted abnormalities."
-                },
-                {
-                    "topic": "Heart Size",
-                    "description": "The heart size is within the normal range."
-                },
-                {
-                    "topic": "Costophrenic Angles",
-                    "description": "There is blunting of bilateral costophrenic angles, with the right side more affected than the left."
-                },
-                {
-                    "topic": "Pleural Effusions",
-                    "description": "The blunting of costophrenic angles may be secondary to small effusions in the pleural space surrounding the lungs."
-                }
-            ]
-        },
-        "MCQS": [
-            {
-                "category": "Diagnostic Accuracy",
-                "question": "What finding is observed in the right lung apex on this chest X-ray?",
-                "options": [
-                    "A) A circular opacity",
-                    "B) A triangular opacity",
-                    "C) A linear opacity",
-                    "D) No new opacity"
-                ],
-                "correct_answer": "B) A triangular opacity",
-                "location_in_text": "The image shows a new triangular opacity in the right lung apex"
-            },
-            {
-                "category": "Anatomical Detailing",
-                "question": "Which anatomical structure shows fullness in this chest X-ray?",
-                "options": [
-                    "A) Left hilum",
-                    "B) Right hilum",
-                    "C) Left lung base",
-                    "D) Right lung base"
-                ],
-                "correct_answer": "B) Right hilum",
-                "location_in_text": "The fullness of the right hilum, which is the area where the main bronchus, blood vessels, and nerves enter and exit the lung, is also new."
-            }
-        ],
-        "image": "https://huggingface.co/spaces/aaditya/demo_app/resolve/main/chest_xray.jpg"
-    }
-]
+def load_json(file_path):
+    with open(file_path, 'r') as file:
+        return json.load(file)
 
-def display_data(index):
+def display_data(index, data):
     item = data[index]
     note = item.get('Note', 'No note available')
     summary = item.get('Summary', {})
-    mcqs = item.get('MCQS', [])
+    mcqs = item.get('MCQS', {})
     image_path = item.get('image', '')
     
     summary_html = "<h3>Summary</h3>"
@@ -78,45 +21,92 @@ def display_data(index):
         summary_html += "<p>No summary available</p>"
     
     mcqs_html = "<h3>MCQs</h3>"
-    for i, q in enumerate(mcqs, 1):
-        mcqs_html += f"<p><strong>Q{i}: {q['question']}</strong></p>"
-        mcqs_html += "<ul>"
-        for option in q['options']:
-            mcqs_html += f"<li>{option}</li>"
-        mcqs_html += "</ul>"
-        mcqs_html += f"<p>Correct Answer: {q['correct_answer']}</p>"
-        if 'location_in_text' in q:
-            mcqs_html += f"<p><em>Location in text: {q['location_in_text']}</em></p>"
-        mcqs_html += "<hr>"
+    if isinstance(mcqs, dict) and 'questions' in mcqs:
+        for i, q in enumerate(mcqs['questions'], 1):
+            mcqs_html += f"<p><strong>Q{i}: {q['question']}</strong></p>"
+            mcqs_html += "<ul>"
+            for option in q['options']:
+                mcqs_html += f"<li>{option}</li>"
+            mcqs_html += "</ul>"
+            mcqs_html += f"<p>Correct Answer: {q['correct_answer']}</p>"
+            if 'location_in_text' in q:
+                mcqs_html += f"<p><em>Location in text: {q['location_in_text']}</em></p>"
+            mcqs_html += "<hr>"
+    elif isinstance(mcqs, list):
+        for i, q in enumerate(mcqs, 1):
+            mcqs_html += f"<p><strong>Q{i}: {q['question']}</strong></p>"
+            mcqs_html += "<ul>"
+            for option in q['options']:
+                mcqs_html += f"<li>{option}</li>"
+            mcqs_html += "</ul>"
+            mcqs_html += f"<p>Correct Answer: {q['correct_answer']}</p>"
+            if 'location_in_text' in q:
+                mcqs_html += f"<p><em>Location in text: {q['location_in_text']}</em></p>"
+            mcqs_html += "<hr>"
+    else:
+        mcqs_html += "<p>No MCQs available for this item.</p>"
     
-    return note, summary_html, mcqs_html, image_path
+    return note, summary_html, mcqs_html, image_path, f"Item {index + 1} of {len(data)}"
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Chest X-Ray Analysis Viewer")
-    
-    with gr.Row():
-        with gr.Column(scale=1):
-            image = gr.Image(label="X-Ray Image")
-        with gr.Column(scale=2):
-            note = gr.Textbox(label="Note", lines=4)
-            summary = gr.HTML(label="Summary")
-            mcqs = gr.HTML(label="MCQs")
-    
-    index = gr.Slider(0, len(data) - 1, step=1, label="Item Index")
-    
-    def update_display(index):
-        return display_data(int(index))
+def next_item(index, data):
+    return min(index + 1, len(data) - 1)
+
+def prev_item(index, data):
+    return max(index - 1, 0)
+
+def create_app(json_path):
+    data = load_json(json_path)
     
-    index.change(
-        update_display,
-        inputs=[index],
-        outputs=[note, summary, mcqs, image]
-    )
+    with gr.Blocks() as app:
+        gr.Markdown("# Chest X-Ray Analysis Viewer")
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                image = gr.Image(label="X-Ray Image")
+            with gr.Column(scale=2):
+                note = gr.Textbox(label="Note", lines=4)
+                summary = gr.HTML(label="Summary")
+                mcqs = gr.HTML(label="MCQs")
+        
+        index = gr.State(0)
+        item_count = gr.Textbox(label="Item Count")
+        
+        with gr.Row():
+            prev_btn = gr.Button("Previous")
+            next_btn = gr.Button("Next")
+        
+        def update_display(index):
+            return display_data(index, data)
+        
+        next_btn.click(
+            next_item,
+            inputs=[index, gr.State(data)],
+            outputs=[index]
+        ).then(
+            update_display,
+            inputs=[index],
+            outputs=[note, summary, mcqs, image, item_count]
+        )
+        
+        prev_btn.click(
+            prev_item,
+            inputs=[index, gr.State(data)],
+            outputs=[index]
+        ).then(
+            update_display,
+            inputs=[index],
+            outputs=[note, summary, mcqs, image, item_count]
+        )
+        
+        app.load(
+            update_display,
+            inputs=[index],
+            outputs=[note, summary, mcqs, image, item_count]
+        )
     
-    demo.load(
-        update_display,
-        inputs=[index],
-        outputs=[note, summary, mcqs, image]
-    )
+    return app
+
 
-demo.launch()
+json_path = Path("mimic_data.json")
+app = create_app(json_path)
+app.launch()