Update app.py

app.py

@@ -254,101 +254,104 @@ def collect_entities_from_records(*args):
     - **Organizations:** {sum(1 for e in builder.entities if e['type'] == 'ORGANIZATION')}
     - **Dates:** {sum(1 for e in builder.entities if e['type'] == 'DATE')}
     
-    Now define relationships between these entities below.
+    Now define relationships between these entities on the right →
     """
     
-    # Return summary and update dropdowns
-    return (
-        summary,
-        gr.update(visible=True),  # Show relationship section
-        gr.update(choices=entity_names, value=None),  # Update all relationship dropdowns
-        gr.update(choices=entity_names, value=None),
-        gr.update(choices=entity_names, value=None),
-        gr.update(choices=entity_names, value=None),
-        gr.update(choices=entity_names, value=None),
-        gr.update(choices=entity_names, value=None),
-        gr.update(choices=entity_names, value=None),
-        gr.update(choices=entity_names, value=None),
-        gr.update(choices=entity_names, value=None),
-        gr.update(choices=entity_names, value=None)
-    )
+    # Return summary and update all dropdowns (5 relationships × 2 dropdowns each = 10 updates)
+    dropdown_updates = [gr.update(choices=entity_names, value=None)] * 10
+    return [summary] + dropdown_updates
 
 def generate_network_graph(*args):
     """Generate the network graph from all inputs"""
-    builder = NetworkGraphBuilder()
-    
-    # Collect entities (first 30 args: 6 records × 5 fields)
-    num_records = 6
-    fields_per_record = 5
-    
-    for i in range(num_records):
-        record_id = i + 1
-        base_idx = i * fields_per_record
+    try:
+        builder = NetworkGraphBuilder()
         
-        person = args[base_idx] if base_idx < len(args) else ""
-        location = args[base_idx + 1] if base_idx + 1 < len(args) else ""
-        event = args[base_idx + 2] if base_idx + 2 < len(args) else ""
-        org = args[base_idx + 3] if base_idx + 3 < len(args) else ""
-        date = args[base_idx + 4] if base_idx + 4 < len(args) else ""
+        # Collect entities (first 30 args: 6 records × 5 fields)
+        num_records = 6
+        fields_per_record = 5
         
-        if person:
-            builder.add_entity(person, 'PERSON', record_id)
-        if location:
-            builder.add_entity(location, 'LOCATION', record_id)
-        if event:
-            builder.add_entity(event, 'EVENT', record_id)
-        if org:
-            builder.add_entity(org, 'ORGANIZATION', record_id)
-        if date:
-            builder.add_entity(date, 'DATE', record_id)
-    
-    # Collect relationships (next args: 5 relationships × 3 fields)
-    relationship_start = 30
-    num_relationships = 5
-    
-    for i in range(num_relationships):
-        base_idx = relationship_start + (i * 3)
-        source = args[base_idx] if base_idx < len(args) else None
-        rel_type = args[base_idx + 1] if base_idx + 1 < len(args) else None
-        target = args[base_idx + 2] if base_idx + 2 < len(args) else None
-        
-        if source and target:
-            builder.add_relationship(source, target, rel_type)
-    
-    # Get layout type (last arg)
-    layout_type = args[-1] if len(args) > relationship_start else 'spring'
-    
-    # Build graph
-    G = builder.build_graph()
-    
-    if len(G.nodes) == 0:
-        return None, "❌ No entities to display. Please add some entities first."
-    
-    # Create visualization (even if no relationships, show isolated nodes)
-    fig = builder.create_plotly_graph(G, layout_type)
-    
-    # Create statistics
-    stats = f"""
-    ### 📈 Network Statistics
-    - **Nodes (Entities):** {G.number_of_nodes()}
-    - **Edges (Relationships):** {G.number_of_edges()}
-    """
-    
-    if len(G.edges) == 0:
-        stats += "\n⚠️ **No relationships defined** - showing isolated nodes only.\n"
-    else:
-        stats += f"- **Network Density:** {nx.density(G):.3f}\n"
-        stats += f"- **Average Connections per Node:** {sum(dict(G.degree()).values()) / G.number_of_nodes():.2f}\n"
-    
-    if G.number_of_edges() > 0:
-        # Find most connected nodes
-        degrees = dict(G.degree())
-        top_nodes = sorted(degrees.items(), key=lambda x: x[1], reverse=True)[:3]
-        stats += "\n**Most Connected Entities:**\n"
-        for node, degree in top_nodes:
-            stats += f"- {node}: {degree} connections\n"
+        for i in range(num_records):
+            record_id = i + 1
+            base_idx = i * fields_per_record
+            
+            person = args[base_idx] if base_idx < len(args) else ""
+            location = args[base_idx + 1] if base_idx + 1 < len(args) else ""
+            event = args[base_idx + 2] if base_idx + 2 < len(args) else ""
+            org = args[base_idx + 3] if base_idx + 3 < len(args) else ""
+            date = args[base_idx + 4] if base_idx + 4 < len(args) else ""
+            
+            if person:
+                builder.add_entity(person, 'PERSON', record_id)
+            if location:
+                builder.add_entity(location, 'LOCATION', record_id)
+            if event:
+                builder.add_entity(event, 'EVENT', record_id)
+            if org:
+                builder.add_entity(org, 'ORGANIZATION', record_id)
+            if date:
+                builder.add_entity(date, 'DATE', record_id)
+        
+        # Collect relationships (next args: 5 relationships × 3 fields)
+        relationship_start = 30
+        num_relationships = 5
+        
+        for i in range(num_relationships):
+            base_idx = relationship_start + (i * 3)
+            source = args[base_idx] if base_idx < len(args) else None
+            rel_type = args[base_idx + 1] if base_idx + 1 < len(args) else None
+            target = args[base_idx + 2] if base_idx + 2 < len(args) else None
+            
+            if source and target:
+                builder.add_relationship(source, target, rel_type)
+        
+        # Get layout type (last arg)
+        layout_type = args[-1] if len(args) > relationship_start else 'spring'
+        
+        # Build graph
+        G = builder.build_graph()
+        
+        if len(G.nodes) == 0:
+            return None, "❌ **No entities to display.** Please enter entities in Step 1 and click 'Identify Entities' first."
+        
+        # Create visualization (even if no relationships, show isolated nodes)
+        fig = builder.create_plotly_graph(G, layout_type)
+        
+        # Create statistics
+        stats = f"""
+        ### 📈 Network Statistics
+        - **Nodes (Entities):** {G.number_of_nodes()}
+        - **Edges (Relationships):** {G.number_of_edges()}
+        """
+        
+        if len(G.edges) == 0:
+            stats += "\n⚠️ **No relationships defined** - showing isolated nodes only.\n"
+            stats += "\n*Define relationships in Step 2 to see connections between entities.*\n"
+        else:
+            stats += f"- **Network Density:** {nx.density(G):.3f}\n"
+            stats += f"- **Average Connections per Node:** {sum(dict(G.degree()).values()) / G.number_of_nodes():.2f}\n"
+        
+        if G.number_of_edges() > 0:
+            # Find most connected nodes
+            degrees = dict(G.degree())
+            top_nodes = sorted(degrees.items(), key=lambda x: x[1], reverse=True)[:3]
+            stats += "\n**Most Connected Entities:**\n"
+            for node, degree in top_nodes:
+                stats += f"- {node}: {degree} connections\n"
+        
+        return fig, stats
     
-    return fig, stats
+    except Exception as e:
+        error_msg = f"""
+        ### ❌ Error Generating Graph
+        
+        An error occurred: {str(e)}
+        
+        **Troubleshooting:**
+        1. Make sure you've clicked "Identify Entities" first
+        2. Check that you have at least one entity entered
+        3. If problem persists, try refreshing the page
+        """
+        return None, error_msg
 
 def create_interface():
     with gr.Blocks(title="Basic Network Explorer", theme=gr.themes.Soft()) as demo:
@@ -359,11 +362,11 @@ def create_interface():
         This tool demonstrates how NER can be used to visualize relationships and connections in text data.
         
         ### How to use this tool:
-        1. **📝 Enter entities** in the records below (people, locations, events, organizations, dates)
-        2. **🔗 Click "Identify Entities"** to gather all your inputs
-        3. **🤝 Define relationships** between entities in the relationship builder
-        4. **🎨 Choose a layout style** and click "Generate Network Graph"
-        5. **👁️ Explore** the interactive visualization
+        1. **📝 Enter entities** in the records on the left (people, locations, events, organizations, dates)
+        2. **🔗 Click "Identify Entities"** to populate the relationship dropdowns
+        3. **🤝 Define relationships** on the right by selecting entities and connection types
+        4. **🎨 Click "Generate Network Graph"** to visualize your network
+        5. **👁️ Explore** the interactive graph - hover over nodes and edges for details
         6. **🔄 Refresh the page** to start over with new data
         """)
         
@@ -382,7 +385,8 @@ def create_interface():
             # LEFT COLUMN: Entity Inputs
             with gr.Column(scale=1):
                 with gr.Accordion("📚 Step 1: Enter Entities from Your Records", open=True):
-                    for i in range(6):
+                    # First 4 records (always visible)
+                    for i in range(4):
                         with gr.Group():
                             gr.Markdown(f"### Record {i+1}")
                             with gr.Row():
@@ -394,42 +398,55 @@ def create_interface():
                             date = gr.Textbox(label="🗓️ Date", placeholder="e.g., 1940")
                             
                             entity_inputs.extend([person, location, event, org, date])
+                    
+                    # Additional records (collapsible)
+                    with gr.Accordion("➕ Additional Records (5-6)", open=False):
+                        for i in range(4, 6):
+                            with gr.Group():
+                                gr.Markdown(f"### Record {i+1}")
+                                with gr.Row():
+                                    person = gr.Textbox(label="👤 Person", placeholder="e.g., Winston Churchill")
+                                    location = gr.Textbox(label="📍 Location", placeholder="e.g., London")
+                                with gr.Row():
+                                    event = gr.Textbox(label="📅 Event", placeholder="e.g., Battle of Britain")
+                                    org = gr.Textbox(label="🏢 Organization", placeholder="e.g., Royal Air Force")
+                                date = gr.Textbox(label="🗓️ Date", placeholder="e.g., 1940")
+                                
+                                entity_inputs.extend([person, location, event, org, date])
                 
                 collect_btn = gr.Button("🔍 Identify Entities", variant="primary", size="lg")
                 entity_summary = gr.Markdown()
             
-            # RIGHT COLUMN: Relationship Builder
+            # RIGHT COLUMN: Relationship Builder (ALWAYS VISIBLE)
             with gr.Column(scale=1):
-                # Relationship section (initially hidden)
-                with gr.Column(visible=False) as relationship_section:
-                    with gr.Accordion("🤝 Step 2: Define Relationships Between Entities", open=True):
-                        gr.Markdown("Select entities and specify how they're connected:")
+                with gr.Accordion("🤝 Step 2: Define Relationships Between Entities", open=True):
+                    gr.Markdown("*First identify entities, then define relationships below:*")
+                
+                    relationship_inputs = []
                     
-                        relationship_inputs = []
+                    for i in range(5):
+                        with gr.Row():
+                            source = gr.Dropdown(label=f"From", choices=[], interactive=True, scale=2)
+                            rel_type = gr.Dropdown(
+                                label="Type",
+                                choices=RELATIONSHIP_TYPES,
+                                value="related_to",
+                                interactive=True,
+                                scale=2
+                            )
+                            target = gr.Dropdown(label=f"To", choices=[], interactive=True, scale=2)
                         
-                        for i in range(5):
-                            with gr.Row():
-                                source = gr.Dropdown(label=f"From", choices=[], interactive=True, scale=2)
-                                rel_type = gr.Dropdown(
-                                    label="Type",
-                                    choices=RELATIONSHIP_TYPES,
-                                    value="related_to",
-                                    interactive=True,
-                                    scale=2
-                                )
-                                target = gr.Dropdown(label=f"To", choices=[], interactive=True, scale=2)
-                            
-                            relationship_inputs.extend([source, rel_type, target])
+                        relationship_inputs.extend([source, rel_type, target])
+                
+                with gr.Accordion("🎨 Step 3: Customize and Generate", open=True):
+                    layout_type = gr.Dropdown(
+                        label="Graph Layout",
+                        choices=['spring', 'circular', 'kamada_kawai', 'shell'],
+                        value='spring',
+                        info="Choose how nodes are arranged"
+                    )
                     
-                    with gr.Accordion("🎨 Step 3: Customize and Generate", open=True):
-                        layout_type = gr.Dropdown(
-                            label="Graph Layout",
-                            choices=['spring', 'circular', 'kamada_kawai', 'shell'],
-                            value='spring',
-                            info="Choose how nodes are arranged"
-                        )
-                        
-                        generate_btn = gr.Button("🔍 Generate Network Graph", variant="primary", size="lg")
+                    generate_btn = gr.Button("🔍 Generate Network Graph", variant="primary", size="lg")
         
         # Output section
         gr.HTML("<hr style='margin: 30px 0;'>")
@@ -527,14 +544,11 @@ def create_interface():
         """)
         
         # Wire up the interface
-        # Collect entities button
+        # Collect entities button - updates the relationship dropdowns
         collect_btn.click(
             fn=collect_entities_from_records,
             inputs=entity_inputs,
-            outputs=[
-                entity_summary,
-                relationship_section
-            ] + relationship_inputs[::3] + relationship_inputs[2::3]  # Update source and target dropdowns
+            outputs=[entity_summary] + relationship_inputs[::3] + relationship_inputs[2::3]  # Update source and target dropdowns
         )
         
         # Generate graph button