Update app.py

app.py

@@ -313,7 +313,7 @@ def collect_entities_from_records(
     </div>
     '''
     
-    # Return summary and update all 10 dropdowns (5 source + 5 target)
+    # Return summary and update all 16 dropdowns (8 source + 8 target)
     return (
         summary_html,
         gr.update(choices=entity_names, value=None),  # source 1
@@ -326,6 +326,12 @@ def collect_entities_from_records(
         gr.update(choices=entity_names, value=None),  # target 4
         gr.update(choices=entity_names, value=None),  # source 5
         gr.update(choices=entity_names, value=None),  # target 5
+        gr.update(choices=entity_names, value=None),  # source 6
+        gr.update(choices=entity_names, value=None),  # target 6
+        gr.update(choices=entity_names, value=None),  # source 7
+        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
     )
 
 
@@ -342,7 +348,10 @@ def generate_network_graph(
     src2, rel2, tgt2,
     src3, rel3, tgt3,
     src4, rel4, tgt4,
-    src5, rel5, tgt5
+    src5, rel5, tgt5,
+    src6, rel6, tgt6,
+    src7, rel7, tgt7,
+    src8, rel8, tgt8
 ):
     """Generate the network graph from all inputs"""
     try:
@@ -372,13 +381,16 @@ def generate_network_graph(
             if date:
                 builder.add_entity(date, 'DATE', record_id)
         
-        # Process relationships
+        # Process relationships (now 8 total)
         relationships = [
             (src1, rel1, tgt1),
             (src2, rel2, tgt2),
             (src3, rel3, tgt3),
             (src4, rel4, tgt4),
             (src5, rel5, tgt5),
+            (src6, rel6, tgt6),
+            (src7, rel7, tgt7),
+            (src8, rel8, tgt8),
         ]
         
         for source, rel_type, target in relationships:
@@ -394,7 +406,7 @@ def generate_network_graph(
                 <div style="text-align: center; color: white;">
                     <div style="font-size: 48px; margin-bottom: 20px;">📊</div>
                     <h3>No entities to display</h3>
-                    <p style="color: #888;">Enter entities above and click "Identify Entities" first</p>
+                    <p style="color: #888;">Enter entities above and click "Process Entities" first</p>
                 </div>
             </div>
             '''
@@ -479,24 +491,30 @@ def load_austen_example():
     """Load the Jane Austen Pride and Prejudice example"""
     return (
         # Record 1
-        "Elizabeth Bennet", "Longbourn", "Meryton Ball", "Bennet Family", "1811",
+        "Elizabeth Bennet", "Longbourn", "Meryton Ball", "", "",
         # Record 2
-        "Mr. Darcy", "Pemberley", "Netherfield Ball", "Darcy Estate", "",
+        "Mr. Darcy", "Pemberley", "Netherfield Ball", "", "",
         # Record 3
         "Jane Bennet", "Netherfield", "", "", "",
         # Record 4
         "Mr. Bingley", "London", "", "", "",
         # Record 5
-        "Mr. Wickham", "Meryton", "", "Militia", "",
+        "Mr. Wickham", "", "", "Militia", "",
         # Record 6
-        "Charlotte Lucas", "Hunsford", "", "", "",
+        "Charlotte Lucas", "", "", "", "",
         # Record 7
-        "", "", "", "", "",
+        "Mr. Collins", "", "Excellent Boiled Potatoes", "", "",
         # Record 8
-        "", "", "", "", "",
+        "Caroline Bingley", "", "", "", "",
     )
 
 
+def load_austen_example_extra():
+    """Return additional Austen characters for records 5-8 area - but we handle in main example"""
+    # This is handled by main example now
+    pass
+
+
 def load_wwii_example():
     """Load a WWII history example"""
     return (
@@ -544,7 +562,7 @@ def create_interface():
         
         ### How to use this tool:
         1. **📝 Enter entities** in the records below (or load an example to get started)
-        2. **🔍 Click "Identify Entities"** to collect and list all entities
+        2. **⚙️ Click "Process Entities"** to collect and prepare all entities for relationships
         3. **🤝 Define relationships** between entities using the dropdowns
         4. **🎨 Click "Generate Network Graph"** to visualise
         5. **👁️ Explore** - drag nodes to rearrange, scroll to zoom, hover for details
@@ -686,19 +704,25 @@ def create_interface():
                     d8 = gr.Textbox(label="", show_label=False)
                     entity_inputs.extend([p8, l8, e8, o8, d8])
         
-        # Identify button
-        collect_btn = gr.Button("🔍 Identify Entities", variant="primary", size="lg")
+        gr.HTML("<hr style='margin: 20px 0;'>")
+        
+        # ==================== STEP 2: PROCESS ENTITIES ====================
+        gr.Markdown("## ⚙️ Step 2: Process Entities")
+        gr.Markdown("*Click the button below to collect all entities before defining relationships*")
+        
+        # Process button
+        collect_btn = gr.Button("⚙️ Process Entities", variant="primary", size="lg")
         
         # Full-width entity summary
         entity_summary = gr.HTML()
         
         gr.HTML("<hr style='margin: 20px 0;'>")
         
-        # ==================== STEP 2: RELATIONSHIPS (Grid) ====================
-        gr.Markdown("## 🤝 Step 2: Define Relationships")
-        gr.Markdown("*Select entities from the dropdowns to create connections (click 'Identify Entities' first)*")
+        # ==================== STEP 3: RELATIONSHIPS (Grid) ====================
+        gr.Markdown("## 🤝 Step 3: Define Relationships")
+        gr.Markdown("*Select entities from the dropdowns to create connections (click 'Process Entities' first)*")
         
-        # Relationship inputs in a grid (5 columns)
+        # Relationship inputs in a grid (5 columns for first 5)
         relationship_inputs = []
         
         with gr.Row():
@@ -737,10 +761,34 @@ 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):
+            with gr.Row():
+                with gr.Column(scale=1, min_width=180):
+                    gr.Markdown("**Relationship 6**")
+                    src6 = gr.Dropdown(label="From", choices=[])
+                    rel6 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to")
+                    tgt6 = gr.Dropdown(label="To", choices=[])
+                    relationship_inputs.extend([src6, rel6, tgt6])
+                
+                with gr.Column(scale=1, min_width=180):
+                    gr.Markdown("**Relationship 7**")
+                    src7 = gr.Dropdown(label="From", choices=[])
+                    rel7 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to")
+                    tgt7 = gr.Dropdown(label="To", choices=[])
+                    relationship_inputs.extend([src7, rel7, tgt7])
+                
+                with gr.Column(scale=1, min_width=180):
+                    gr.Markdown("**Relationship 8**")
+                    src8 = gr.Dropdown(label="From", choices=[])
+                    rel8 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to")
+                    tgt8 = gr.Dropdown(label="To", choices=[])
+                    relationship_inputs.extend([src8, rel8, tgt8])
+        
         gr.HTML("<hr style='margin: 20px 0;'>")
         
-        # ==================== STEP 3: GENERATE & VIEW ====================
-        gr.Markdown("## 🎨 Step 3: Generate Network Graph")
+        # ==================== STEP 4: GENERATE & VIEW ====================
+        gr.Markdown("## 🎨 Step 4: Generate Network Graph")
         
         generate_btn = gr.Button("🎨 Generate Network Graph", variant="primary", size="lg")
         
@@ -777,7 +825,7 @@ def create_interface():
                     Date
                 </span>
             </div>
-            <p style="color: #888; margin: 15px 0 0 0; font-size: 13px;">
+            <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>
@@ -798,7 +846,7 @@ def create_interface():
             outputs=entity_inputs
         )
         
-        # Collect entities
+        # Collect entities - now updates 16 dropdowns (8 relationships x 2)
         collect_btn.click(
             fn=collect_entities_from_records,
             inputs=entity_inputs,
@@ -808,7 +856,10 @@ def create_interface():
                 src2, tgt2,
                 src3, tgt3,
                 src4, tgt4,
-                src5, tgt5
+                src5, tgt5,
+                src6, tgt6,
+                src7, tgt7,
+                src8, tgt8
             ]
         )