Update app.py

app.py

@@ -37,27 +37,16 @@ def display_example(index):
 
     # Modify the iframe to have a fixed height
     graph_html = graph_html.replace('height: 100vh;', 'height: 500px;')
-        
-    # Wrap the graph HTML in a div with fixed height but no scrolling
-    # graph_html_with_style = f"""
-    # <div style="height: 300px; justify-content: center; align-items: center;overflow-y: auto">
-    #     <div style=max-height: 300px;">
-    #         {graph_html}
-    #     </div>
-    # </div>
-    # """
-    graph_html_with_style = graph_html
-    
+
     # Convert graph_data to a formatted JSON string
     json_data = json.dumps(transformed_graph_data, indent=2)
     
-    return img, graph_html_with_style, json_data, transformed_graph_data
+    return img, graph_html, json_data, transformed_graph_data
 
 def create_interface():
-    # with gr.Blocks(css="#graph-output {height: 200px; overflow: hidden;}") as demo:
     with gr.Blocks() as demo:
     
-        gr.Markdown("# Knowledge Graph Visualizer")
+        gr.Markdown("# Knowledge Graph Visualizer for the [Zaherrr/OOP_KG_Dataset](https://huggingface.co/datasets/Zaherrr/OOP_KG_Dataset)")
         
         with gr.Row():
             index_slider = gr.Slider(
@@ -69,7 +58,7 @@ def create_interface():
         
         with gr.Row():
             image_output = gr.Image(type="pil", label="Image", height=500)
-            graph_output = gr.HTML(label="Knowledge Graph")#, elem_id="graph-output")
+            graph_output = gr.HTML(label="Knowledge Graph")
         
         with gr.Row():
             json_output = gr.Code(language="json", label="Graph JSON Data")