Update app.py

app.py

@@ -243,7 +243,9 @@ def collect_entities_from_records(
     p5, l5, e5, o5, d5,
     p6, l6, e6, o6, d6,
     p7, l7, e7, o7, d7,
-    p8, l8, e8, o8, d8
+    p8, l8, e8, o8, d8,
+    p9, l9, e9, o9, d9,
+    p10, l10, e10, o10, d10
 ):
     """Collect all entities from the input fields"""
     builder = NetworkGraphBuilder()
@@ -258,6 +260,8 @@ def collect_entities_from_records(
         (p6, l6, e6, o6, d6),
         (p7, l7, e7, o7, d7),
         (p8, l8, e8, o8, d8),
+        (p9, l9, e9, o9, d9),
+        (p10, l10, e10, o10, d10),
     ]
     
     for record_id, (person, location, event, org, date) in enumerate(records, 1):
@@ -318,7 +322,7 @@ def collect_entities_from_records(
     </div>
     '''
     
-    # Return summary and update all 16 dropdowns (8 source + 8 target)
+    # Return summary and update all 20 dropdowns (10 source + 10 target)
     return (
         summary_html,
         gr.update(choices=entity_names, value=None),  # source 1
@@ -337,6 +341,10 @@ def collect_entities_from_records(
         gr.update(choices=entity_names, value=None),  # target 7
         gr.update(choices=entity_names, value=None),  # source 8
         gr.update(choices=entity_names, value=None),  # target 8
+        gr.update(choices=entity_names, value=None),  # source 9
+        gr.update(choices=entity_names, value=None),  # target 9
+        gr.update(choices=entity_names, value=None),  # source 10
+        gr.update(choices=entity_names, value=None),  # target 10
     )
 
 
@@ -349,6 +357,8 @@ def generate_network_graph(
     p6, l6, e6, o6, d6,
     p7, l7, e7, o7, d7,
     p8, l8, e8, o8, d8,
+    p9, l9, e9, o9, d9,
+    p10, l10, e10, o10, d10,
     src1, rel1, tgt1,
     src2, rel2, tgt2,
     src3, rel3, tgt3,
@@ -356,7 +366,9 @@ def generate_network_graph(
     src5, rel5, tgt5,
     src6, rel6, tgt6,
     src7, rel7, tgt7,
-    src8, rel8, tgt8
+    src8, rel8, tgt8,
+    src9, rel9, tgt9,
+    src10, rel10, tgt10
 ):
     """Generate the network graph from all inputs"""
     try:
@@ -372,6 +384,8 @@ def generate_network_graph(
             (p6, l6, e6, o6, d6),
             (p7, l7, e7, o7, d7),
             (p8, l8, e8, o8, d8),
+            (p9, l9, e9, o9, d9),
+            (p10, l10, e10, o10, d10),
         ]
         
         for record_id, (person, location, event, org, date) in enumerate(records, 1):
@@ -386,7 +400,7 @@ def generate_network_graph(
             if date:
                 builder.add_entity(date, 'DATE', record_id)
         
-        # Process relationships (now 8 total)
+        # Process relationships (now 10 total)
         relationships = [
             (src1, rel1, tgt1),
             (src2, rel2, tgt2),
@@ -396,6 +410,8 @@ def generate_network_graph(
             (src6, rel6, tgt6),
             (src7, rel7, tgt7),
             (src8, rel8, tgt8),
+            (src9, rel9, tgt9),
+            (src10, rel10, tgt10),
         ]
         
         for source, rel_type, target in relationships:
@@ -420,7 +436,7 @@ def generate_network_graph(
         # Create visualisation
         graph_html, standalone_html = builder.create_pyvis_graph(G)
         
-        # Create statistics
+        # Create statistics with colour legend included
         stats_html = f'''
         <div style="background: #f8f9fa; padding: 15px; border-radius: 10px; border: 1px solid #ddd;">
             <h4 style="margin: 0 0 10px 0;">📈 Network Statistics</h4>
@@ -474,6 +490,38 @@ def generate_network_graph(
         
         stats_html += '</div>'
         
+        # Add colour legend below stats
+        stats_html += f'''
+        <div style="background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%); padding: 15px; border-radius: 10px; margin-top: 15px;">
+            <h4 style="color: white; margin: 0 0 12px 0; font-size: 14px;">🎨 Entity Colour Legend</h4>
+            <div style="display: flex; flex-direction: column; gap: 8px;">
+                <span style="display: flex; align-items: center; gap: 8px; color: white; font-size: 13px;">
+                    <span style="width: 16px; height: 16px; border-radius: 50%; background-color: {ENTITY_COLOURS['PERSON']}; display: inline-block; border: 2px solid white;"></span>
+                    Person
+                </span>
+                <span style="display: flex; align-items: center; gap: 8px; color: white; font-size: 13px;">
+                    <span style="width: 16px; height: 16px; border-radius: 50%; background-color: {ENTITY_COLOURS['LOCATION']}; display: inline-block; border: 2px solid white;"></span>
+                    Location
+                </span>
+                <span style="display: flex; align-items: center; gap: 8px; color: white; font-size: 13px;">
+                    <span style="width: 16px; height: 16px; border-radius: 50%; background-color: {ENTITY_COLOURS['EVENT']}; display: inline-block; border: 2px solid white;"></span>
+                    Event
+                </span>
+                <span style="display: flex; align-items: center; gap: 8px; color: white; font-size: 13px;">
+                    <span style="width: 16px; height: 16px; border-radius: 50%; background-color: {ENTITY_COLOURS['ORGANIZATION']}; display: inline-block; border: 2px solid white;"></span>
+                    Organisation
+                </span>
+                <span style="display: flex; align-items: center; gap: 8px; color: white; font-size: 13px;">
+                    <span style="width: 16px; height: 16px; border-radius: 50%; background-color: {ENTITY_COLOURS['DATE']}; display: inline-block; border: 2px solid white;"></span>
+                    Date
+                </span>
+            </div>
+            <p style="color: white; margin: 12px 0 0 0; font-size: 12px;">
+                🖱️ <strong style="color: white;">Interaction:</strong> Drag nodes • Scroll to zoom • Hover for details
+            </p>
+        </div>
+        '''
+        
         return graph_html, stats_html
     
     except Exception as e:
@@ -501,6 +549,8 @@ def export_network_graph(
     p6, l6, e6, o6, d6,
     p7, l7, e7, o7, d7,
     p8, l8, e8, o8, d8,
+    p9, l9, e9, o9, d9,
+    p10, l10, e10, o10, d10,
     src1, rel1, tgt1,
     src2, rel2, tgt2,
     src3, rel3, tgt3,
@@ -508,7 +558,9 @@ def export_network_graph(
     src5, rel5, tgt5,
     src6, rel6, tgt6,
     src7, rel7, tgt7,
-    src8, rel8, tgt8
+    src8, rel8, tgt8,
+    src9, rel9, tgt9,
+    src10, rel10, tgt10
 ):
     """Export the network graph as a standalone HTML file"""
     try:
@@ -524,6 +576,8 @@ def export_network_graph(
             (p6, l6, e6, o6, d6),
             (p7, l7, e7, o7, d7),
             (p8, l8, e8, o8, d8),
+            (p9, l9, e9, o9, d9),
+            (p10, l10, e10, o10, d10),
         ]
         
         for record_id, (person, location, event, org, date) in enumerate(records, 1):
@@ -548,6 +602,8 @@ def export_network_graph(
             (src6, rel6, tgt6),
             (src7, rel7, tgt7),
             (src8, rel8, tgt8),
+            (src9, rel9, tgt9),
+            (src10, rel10, tgt10),
         ]
         
         for source, rel_type, target in relationships:
@@ -601,6 +657,10 @@ def load_austen_example():
         "Mr. Collins", "", "Excellent Boiled Potatoes", "", "",
         # Record 8
         "Caroline Bingley", "", "", "", "",
+        # Record 9
+        "Lydia Bennet", "", "", "", "",
+        # Record 10
+        "Lady Catherine de Bourgh", "", "", "", "",
     )
 
 
@@ -623,6 +683,10 @@ def load_wwii_example():
         "", "", "", "", "",
         # Record 8
         "", "", "", "", "",
+        # Record 9
+        "", "", "", "", "",
+        # Record 10
+        "", "", "", "", "",
     )
 
 
@@ -672,47 +736,47 @@ def create_interface():
         
         gr.HTML("<hr style='margin: 15px 0;'>")
         
-        # ==================== STEP 1: ENTITY INPUT (4 per row) ====================
+        # ==================== STEP 1: ENTITY INPUT (5 per row) ====================
         gr.Markdown("## 📝 Step 1: Enter Entities")
         
         entity_inputs = []
         
-        # First row: Records 1-4 (all on one line)
+        # First row: Records 1-5
         with gr.Row():
-            with gr.Column(scale=1, min_width=200):
+            with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Record 1**")
                 gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
-                p1 = gr.Textbox(label="", placeholder="e.g., Elizabeth Bennet", show_label=False)
+                p1 = gr.Textbox(label="", placeholder="e.g. Elizabeth Bennet", show_label=False)
                 gr.HTML(create_coloured_label("Location", ENTITY_COLOURS['LOCATION'], "📍"))
-                l1 = gr.Textbox(label="", placeholder="e.g., Longbourn", show_label=False)
+                l1 = gr.Textbox(label="", placeholder="e.g. Longbourn", show_label=False)
                 gr.HTML(create_coloured_label("Event", ENTITY_COLOURS['EVENT'], "📅"))
-                e1 = gr.Textbox(label="", placeholder="e.g., Meryton Ball", show_label=False)
+                e1 = gr.Textbox(label="", placeholder="e.g. Meryton Ball", show_label=False)
                 gr.HTML(create_coloured_label("Organisation", ENTITY_COLOURS['ORGANIZATION'], "🏢"))
-                o1 = gr.Textbox(label="", placeholder="e.g., Bennet Family", show_label=False)
+                o1 = gr.Textbox(label="", placeholder="e.g. The Militia", show_label=False)
                 gr.HTML(create_coloured_label("Date", ENTITY_COLOURS['DATE'], "🗓️"))
-                d1 = gr.Textbox(label="", placeholder="e.g., 1811", show_label=False)
+                d1 = gr.Textbox(label="", placeholder="e.g. 1812", show_label=False)
                 entity_inputs.extend([p1, l1, e1, o1, d1])
             
-            with gr.Column(scale=1, min_width=200):
+            with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Record 2**")
                 gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
-                p2 = gr.Textbox(label="", placeholder="e.g., Mr. Darcy", show_label=False)
+                p2 = gr.Textbox(label="", placeholder="e.g. Mr. Darcy", show_label=False)
                 gr.HTML(create_coloured_label("Location", ENTITY_COLOURS['LOCATION'], "📍"))
-                l2 = gr.Textbox(label="", placeholder="e.g., Pemberley", show_label=False)
+                l2 = gr.Textbox(label="", placeholder="e.g. Pemberley", show_label=False)
                 gr.HTML(create_coloured_label("Event", ENTITY_COLOURS['EVENT'], "📅"))
-                e2 = gr.Textbox(label="", placeholder="e.g., Netherfield Ball", show_label=False)
+                e2 = gr.Textbox(label="", placeholder="", show_label=False)
                 gr.HTML(create_coloured_label("Organisation", ENTITY_COLOURS['ORGANIZATION'], "🏢"))
-                o2 = gr.Textbox(label="", placeholder="e.g., Darcy Estate", show_label=False)
+                o2 = gr.Textbox(label="", placeholder="", show_label=False)
                 gr.HTML(create_coloured_label("Date", ENTITY_COLOURS['DATE'], "🗓️"))
                 d2 = gr.Textbox(label="", placeholder="", show_label=False)
                 entity_inputs.extend([p2, l2, e2, o2, d2])
             
-            with gr.Column(scale=1, min_width=200):
+            with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Record 3**")
                 gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
-                p3 = gr.Textbox(label="", placeholder="e.g., Jane Bennet", show_label=False)
+                p3 = gr.Textbox(label="", placeholder="", show_label=False)
                 gr.HTML(create_coloured_label("Location", ENTITY_COLOURS['LOCATION'], "📍"))
-                l3 = gr.Textbox(label="", placeholder="e.g., Netherfield", show_label=False)
+                l3 = gr.Textbox(label="", placeholder="", show_label=False)
                 gr.HTML(create_coloured_label("Event", ENTITY_COLOURS['EVENT'], "📅"))
                 e3 = gr.Textbox(label="", placeholder="", show_label=False)
                 gr.HTML(create_coloured_label("Organisation", ENTITY_COLOURS['ORGANIZATION'], "🏢"))
@@ -721,12 +785,12 @@ def create_interface():
                 d3 = gr.Textbox(label="", placeholder="", show_label=False)
                 entity_inputs.extend([p3, l3, e3, o3, d3])
             
-            with gr.Column(scale=1, min_width=200):
+            with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Record 4**")
                 gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
-                p4 = gr.Textbox(label="", placeholder="e.g., Mr. Bingley", show_label=False)
+                p4 = gr.Textbox(label="", placeholder="", show_label=False)
                 gr.HTML(create_coloured_label("Location", ENTITY_COLOURS['LOCATION'], "📍"))
-                l4 = gr.Textbox(label="", placeholder="e.g., London", show_label=False)
+                l4 = gr.Textbox(label="", placeholder="", show_label=False)
                 gr.HTML(create_coloured_label("Event", ENTITY_COLOURS['EVENT'], "📅"))
                 e4 = gr.Textbox(label="", placeholder="", show_label=False)
                 gr.HTML(create_coloured_label("Organisation", ENTITY_COLOURS['ORGANIZATION'], "🏢"))
@@ -734,25 +798,25 @@ def create_interface():
                 gr.HTML(create_coloured_label("Date", ENTITY_COLOURS['DATE'], "🗓️"))
                 d4 = gr.Textbox(label="", placeholder="", show_label=False)
                 entity_inputs.extend([p4, l4, e4, o4, d4])
+            
+            with gr.Column(scale=1, min_width=180):
+                gr.Markdown("**Record 5**")
+                gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
+                p5 = gr.Textbox(label="", placeholder="", show_label=False)
+                gr.HTML(create_coloured_label("Location", ENTITY_COLOURS['LOCATION'], "📍"))
+                l5 = gr.Textbox(label="", placeholder="", show_label=False)
+                gr.HTML(create_coloured_label("Event", ENTITY_COLOURS['EVENT'], "📅"))
+                e5 = gr.Textbox(label="", placeholder="", show_label=False)
+                gr.HTML(create_coloured_label("Organisation", ENTITY_COLOURS['ORGANIZATION'], "🏢"))
+                o5 = gr.Textbox(label="", placeholder="", show_label=False)
+                gr.HTML(create_coloured_label("Date", ENTITY_COLOURS['DATE'], "🗓️"))
+                d5 = gr.Textbox(label="", placeholder="", show_label=False)
+                entity_inputs.extend([p5, l5, e5, o5, d5])
         
-        # Additional records 5-8
-        with gr.Accordion("➕ Additional Records (5-8)", open=False):
+        # Additional records 6-10
+        with gr.Accordion("➕ Additional Records (6-10)", open=False):
             with gr.Row():
-                with gr.Column(scale=1, min_width=200):
-                    gr.Markdown("**Record 5**")
-                    gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
-                    p5 = gr.Textbox(label="", show_label=False)
-                    gr.HTML(create_coloured_label("Location", ENTITY_COLOURS['LOCATION'], "📍"))
-                    l5 = gr.Textbox(label="", show_label=False)
-                    gr.HTML(create_coloured_label("Event", ENTITY_COLOURS['EVENT'], "📅"))
-                    e5 = gr.Textbox(label="", show_label=False)
-                    gr.HTML(create_coloured_label("Organisation", ENTITY_COLOURS['ORGANIZATION'], "🏢"))
-                    o5 = gr.Textbox(label="", show_label=False)
-                    gr.HTML(create_coloured_label("Date", ENTITY_COLOURS['DATE'], "🗓️"))
-                    d5 = gr.Textbox(label="", show_label=False)
-                    entity_inputs.extend([p5, l5, e5, o5, d5])
-                
-                with gr.Column(scale=1, min_width=200):
+                with gr.Column(scale=1, min_width=180):
                     gr.Markdown("**Record 6**")
                     gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
                     p6 = gr.Textbox(label="", show_label=False)
@@ -766,7 +830,7 @@ def create_interface():
                     d6 = gr.Textbox(label="", show_label=False)
                     entity_inputs.extend([p6, l6, e6, o6, d6])
                 
-                with gr.Column(scale=1, min_width=200):
+                with gr.Column(scale=1, min_width=180):
                     gr.Markdown("**Record 7**")
                     gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
                     p7 = gr.Textbox(label="", show_label=False)
@@ -780,7 +844,7 @@ def create_interface():
                     d7 = gr.Textbox(label="", show_label=False)
                     entity_inputs.extend([p7, l7, e7, o7, d7])
                 
-                with gr.Column(scale=1, min_width=200):
+                with gr.Column(scale=1, min_width=180):
                     gr.Markdown("**Record 8**")
                     gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
                     p8 = gr.Textbox(label="", show_label=False)
@@ -793,6 +857,34 @@ def create_interface():
                     gr.HTML(create_coloured_label("Date", ENTITY_COLOURS['DATE'], "🗓️"))
                     d8 = gr.Textbox(label="", show_label=False)
                     entity_inputs.extend([p8, l8, e8, o8, d8])
+                
+                with gr.Column(scale=1, min_width=180):
+                    gr.Markdown("**Record 9**")
+                    gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
+                    p9 = gr.Textbox(label="", show_label=False)
+                    gr.HTML(create_coloured_label("Location", ENTITY_COLOURS['LOCATION'], "📍"))
+                    l9 = gr.Textbox(label="", show_label=False)
+                    gr.HTML(create_coloured_label("Event", ENTITY_COLOURS['EVENT'], "📅"))
+                    e9 = gr.Textbox(label="", show_label=False)
+                    gr.HTML(create_coloured_label("Organisation", ENTITY_COLOURS['ORGANIZATION'], "🏢"))
+                    o9 = gr.Textbox(label="", show_label=False)
+                    gr.HTML(create_coloured_label("Date", ENTITY_COLOURS['DATE'], "🗓️"))
+                    d9 = gr.Textbox(label="", show_label=False)
+                    entity_inputs.extend([p9, l9, e9, o9, d9])
+                
+                with gr.Column(scale=1, min_width=180):
+                    gr.Markdown("**Record 10**")
+                    gr.HTML(create_coloured_label("Person", ENTITY_COLOURS['PERSON'], "👤"))
+                    p10 = gr.Textbox(label="", show_label=False)
+                    gr.HTML(create_coloured_label("Location", ENTITY_COLOURS['LOCATION'], "📍"))
+                    l10 = gr.Textbox(label="", show_label=False)
+                    gr.HTML(create_coloured_label("Event", ENTITY_COLOURS['EVENT'], "📅"))
+                    e10 = gr.Textbox(label="", show_label=False)
+                    gr.HTML(create_coloured_label("Organisation", ENTITY_COLOURS['ORGANIZATION'], "🏢"))
+                    o10 = gr.Textbox(label="", show_label=False)
+                    gr.HTML(create_coloured_label("Date", ENTITY_COLOURS['DATE'], "🗓️"))
+                    d10 = gr.Textbox(label="", show_label=False)
+                    entity_inputs.extend([p10, l10, e10, o10, d10])
         
         gr.HTML("<hr style='margin: 20px 0;'>")
         
@@ -851,8 +943,8 @@ def create_interface():
                 tgt5 = gr.Dropdown(label="To", choices=[])
                 relationship_inputs.extend([src5, rel5, tgt5])
         
-        # Additional relationships 6-8
-        with gr.Accordion("➕ Additional Relationships (6-8)", open=False):
+        # Additional relationships 6-10
+        with gr.Accordion("➕ Additional Relationships (6-10)", open=False):
             with gr.Row():
                 with gr.Column(scale=1, min_width=180):
                     gr.Markdown("**Relationship 6**")
@@ -874,6 +966,20 @@ def create_interface():
                     rel8 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to", allow_custom_value=True)
                     tgt8 = gr.Dropdown(label="To", choices=[])
                     relationship_inputs.extend([src8, rel8, tgt8])
+                
+                with gr.Column(scale=1, min_width=180):
+                    gr.Markdown("**Relationship 9**")
+                    src9 = gr.Dropdown(label="From", choices=[])
+                    rel9 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to", allow_custom_value=True)
+                    tgt9 = gr.Dropdown(label="To", choices=[])
+                    relationship_inputs.extend([src9, rel9, tgt9])
+                
+                with gr.Column(scale=1, min_width=180):
+                    gr.Markdown("**Relationship 10**")
+                    src10 = gr.Dropdown(label="From", choices=[])
+                    rel10 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to", allow_custom_value=True)
+                    tgt10 = gr.Dropdown(label="To", choices=[])
+                    relationship_inputs.extend([src10, rel10, tgt10])
         
         gr.HTML("<hr style='margin: 20px 0;'>")
         
@@ -882,7 +988,7 @@ def create_interface():
         
         generate_btn = gr.Button("🎨 Generate Network Graph", variant="primary", size="lg")
         
-        # Full-width network graph with stats sidebar
+        # Full-width network graph with stats sidebar (legend now included in stats)
         with gr.Row():
             with gr.Column(scale=3):
                 network_plot = gr.HTML(label="Interactive Network Graph")
@@ -905,38 +1011,6 @@ def create_interface():
         </div>
         """)
         
-        # Colour legend
-        gr.HTML(f"""
-        <div style="background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%); padding: 20px; border-radius: 10px; margin-top: 20px;">
-            <h4 style="color: white; margin: 0 0 15px 0;">🎨 Entity Colour Legend</h4>
-            <div style="display: flex; flex-wrap: wrap; gap: 20px;">
-                <span style="display: flex; align-items: center; gap: 8px; color: white;">
-                    <span style="width: 20px; height: 20px; border-radius: 50%; background-color: {ENTITY_COLOURS['PERSON']}; display: inline-block; border: 2px solid white;"></span>
-                    Person
-                </span>
-                <span style="display: flex; align-items: center; gap: 8px; color: white;">
-                    <span style="width: 20px; height: 20px; border-radius: 50%; background-color: {ENTITY_COLOURS['LOCATION']}; display: inline-block; border: 2px solid white;"></span>
-                    Location
-                </span>
-                <span style="display: flex; align-items: center; gap: 8px; color: white;">
-                    <span style="width: 20px; height: 20px; border-radius: 50%; background-color: {ENTITY_COLOURS['EVENT']}; display: inline-block; border: 2px solid white;"></span>
-                    Event
-                </span>
-                <span style="display: flex; align-items: center; gap: 8px; color: white;">
-                    <span style="width: 20px; height: 20px; border-radius: 50%; background-color: {ENTITY_COLOURS['ORGANIZATION']}; display: inline-block; border: 2px solid white;"></span>
-                    Organisation
-                </span>
-                <span style="display: flex; align-items: center; gap: 8px; color: white;">
-                    <span style="width: 20px; height: 20px; border-radius: 50%; background-color: {ENTITY_COLOURS['DATE']}; display: inline-block; border: 2px solid white;"></span>
-                    Date
-                </span>
-            </div>
-            <p style="color: white; margin: 15px 0 0 0; font-size: 13px;">
-                🖱️ <strong>Interaction:</strong> Drag nodes to rearrange • Scroll to zoom • Hover for details
-            </p>
-        </div>
-        """)
-        
         # ==================== WIRE UP EVENTS ====================
         
         # Example buttons
@@ -952,7 +1026,7 @@ def create_interface():
             outputs=entity_inputs
         )
         
-        # Collect entities - now updates 16 dropdowns (8 relationships x 2)
+        # Collect entities - now updates 20 dropdowns (10 relationships x 2)
         collect_btn.click(
             fn=collect_entities_from_records,
             inputs=entity_inputs,
@@ -965,7 +1039,9 @@ def create_interface():
                 src5, tgt5,
                 src6, tgt6,
                 src7, tgt7,
-                src8, tgt8
+                src8, tgt8,
+                src9, tgt9,
+                src10, tgt10
             ]
         )