Update app.py

app.py

@@ -27,6 +27,9 @@ def reshape_json_data_to_fit_visualize_graph(graph_data):
 def display_example(index):
     example = dataset[index]
     img = example["image"]
+
+    # Get image dimensions
+    img_width, img_height = img.size
     
     # Prepare the graph data
     graph_data = {"nodes": example["nodes"], "edges": example["edges"]}
@@ -41,12 +44,12 @@ def display_example(index):
     # Convert graph_data to a formatted JSON string
     json_data = json.dumps(transformed_graph_data, indent=2)
     
-    return img, graph_html, json_data, transformed_graph_data
+    return img, graph_html, json_data, transformed_graph_data, f"Width: {img_width}px, Height: {img_height}px"
 
 def create_interface():
     with gr.Blocks() as demo:
     
-        gr.Markdown("# Knowledge Graph Visualizer for the [Zaherrr/OOP_KG_Dataset](https://huggingface.co/datasets/Zaherrr/OOP_KG_Dataset)")
+        gr.Markdown("# Knowledge Graph Visualizer for the [Zaherrr/OOP_KG_Dataset](https://huggingface.co/datasets/Zaherrr/OOP_KG_Dataset) dataset")
         
         with gr.Row():
             index_slider = gr.Slider(
@@ -59,7 +62,14 @@ def create_interface():
         with gr.Row():
             image_output = gr.Image(type="pil", label="Image", height=500)
             graph_output = gr.HTML(label="Knowledge Graph")
-        
+
+        with gr.Row():
+            dimensions_output = gr.Textbox(
+                label="Image Dimensions (pixels)",
+                placeholder="Width and Height will appear here",
+                interactive=False,
+            )
+            
         with gr.Row():
             json_output = gr.Code(language="json", label="Graph JSON Data")
             text_output = gr.Textbox(
@@ -71,7 +81,7 @@ def create_interface():
         index_slider.change(
             fn=display_example,
             inputs=[index_slider],
-            outputs=[image_output, graph_output, json_output, text_output],
+            outputs=[image_output, graph_output, json_output, text_output, dimensions_output],
         )
     
     return demo