updated legend layout

visualize_kg.py

@@ -274,9 +274,14 @@ def visualize_protein_subgraph(data, protein_id, prediction_df, limit=10):
             border: 1px solid #ddd;
             border-radius: 5px;
             font-family: Arial, sans-serif;
+            display: flex;
+            gap: 20px;
+        }
+        .legend-section-nodes {
+            flex: 2;  /* Takes up 2/3 of the space */
         }
-        .legend-section {
-            margin-bottom: 20px;
+        .legend-section-edges {
+            flex: 1;  /* Takes up 1/3 of the space */
         }
         .legend-title {
             margin-bottom: 15px;
@@ -284,9 +289,14 @@ def visualize_protein_subgraph(data, protein_id, prediction_df, limit=10):
             font-size: 16px;
             font-weight: bold;
         }
-        .legend-grid {
+        .nodes-grid {
+            display: grid;
+            grid-template-columns: repeat(2, 1fr);
+            gap: 12px;
+        }
+        .edges-grid {
             display: grid;
-            grid-template-columns: repeat(auto-fit, minmax(180px, 1fr));
+            grid-template-columns: 1fr;
             gap: 12px;
         }
         .legend-item {
@@ -312,11 +322,11 @@ def visualize_protein_subgraph(data, protein_id, prediction_df, limit=10):
         }
     </style>
     <div class="kg-legend">
-        <div class="legend-section">
+        <div class="legend-section-nodes">
             <div class="legend-title">Node Types</div>
-            <div class="legend-grid">"""
+            <div class="nodes-grid">"""
 
-    # Node types
+    # Node types in 2 columns
     for node_type, color in NODE_TYPE_COLORS.items():
         legend_html += f"""
                 <div class="legend-item">
@@ -324,13 +334,13 @@ def visualize_protein_subgraph(data, protein_id, prediction_df, limit=10):
                     <span class="legend-label">{node_type}</span>
                 </div>"""
 
-    # Edge types
+    # Edge types in 1 column
     legend_html += """
             </div>
         </div>
-        <div class="legend-section">
+        <div class="legend-section-edges">
             <div class="legend-title">Edge Colors</div>
-            <div class="legend-grid">
+            <div class="edges-grid">
                 <div class="legend-item">
                     <div class="edge-indicator" style="background-color: #8338ec;"></div>
                     <span class="legend-label">Validated GO Prediction</span>