Update app.py

app.py

@@ -13,28 +13,28 @@ ENTITY_COLOURS = {
     'DATE': '#FF6B6B'         # Red
 }
 
-# Relationship types for dropdown
+# Relationship types for dropdown (alphabetical, capitalised, no underscores)
 RELATIONSHIP_TYPES = [
-    'related to',
-    'knows',
-    'works with',
-    'located in',
-    'participated in',
-    'member of',
-    'occurred at',
-    'employed by',
-    'founded',
-    'attended',
-    'collaborates with',
-    'married to',
-    'sibling of',
-    'parent of',
-    'lives at',
-    'wrote',
-    'visited',
-    'born in',
-    'died in',
-    'other'
+    'Attended',
+    'Born in',
+    'Collaborates with',
+    'Died in',
+    'Employed by',
+    'Founded',
+    'Knows',
+    'Lives at',
+    'Located in',
+    'Married to',
+    'Member of',
+    'Occurred at',
+    'Other',
+    'Parent of',
+    'Participated in',
+    'Related to',
+    'Sibling of',
+    'Visited',
+    'Works with',
+    'Wrote'
 ]
 
 
@@ -61,7 +61,7 @@ class NetworkGraphBuilder:
             self.relationships.append({
                 'source': source.strip(),
                 'target': target.strip(),
-                'type': rel_type if rel_type else 'related_to'
+                'type': rel_type if rel_type else 'Related to'
             })
     
     def build_graph(self):
@@ -409,11 +409,11 @@ def generate_network_graph(
             <h4 style="margin: 0 0 10px 0;">📈 Network Statistics</h4>
             <table style="width: 100%; border-collapse: collapse;">
                 <tr>
-                    <td style="padding: 8px; border-bottom: 1px solid #eee;"><strong>Nodes (Entities)</strong></td>
+                    <td style="padding: 8px; border-bottom: 1px solid #eee;"><strong>Nodes</strong></td>
                     <td style="padding: 8px; border-bottom: 1px solid #eee; text-align: right;">{G.number_of_nodes()}</td>
                 </tr>
                 <tr>
-                    <td style="padding: 8px; border-bottom: 1px solid #eee;"><strong>Edges (Relationships)</strong></td>
+                    <td style="padding: 8px; border-bottom: 1px solid #eee;"><strong>Edges</strong></td>
                     <td style="padding: 8px; border-bottom: 1px solid #eee; text-align: right;">{G.number_of_edges()}</td>
                 </tr>
         '''
@@ -423,7 +423,10 @@ def generate_network_graph(
             avg_degree = sum(dict(G.degree()).values()) / G.number_of_nodes()
             stats_html += f'''
                 <tr>
-                    <td style="padding: 8px; border-bottom: 1px solid #eee;"><strong>Density</strong></td>
+                    <td style="padding: 8px; border-bottom: 1px solid #eee;">
+                        <strong>Connectedness (Density)</strong>
+                        <div style="font-size: 11px; color: #888; font-weight: normal;">0 = isolated, 1 = fully linked</div>
+                    </td>
                     <td style="padding: 8px; border-bottom: 1px solid #eee; text-align: right;">{density:.3f}</td>
                 </tr>
                 <tr>
@@ -702,35 +705,35 @@ def create_interface():
             with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Relationship 1**")
                 src1 = gr.Dropdown(label="From", choices=[])
-                rel1 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="related_to")
+                rel1 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to")
                 tgt1 = gr.Dropdown(label="To", choices=[])
                 relationship_inputs.extend([src1, rel1, tgt1])
             
             with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Relationship 2**")
                 src2 = gr.Dropdown(label="From", choices=[])
-                rel2 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="related_to")
+                rel2 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to")
                 tgt2 = gr.Dropdown(label="To", choices=[])
                 relationship_inputs.extend([src2, rel2, tgt2])
             
             with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Relationship 3**")
                 src3 = gr.Dropdown(label="From", choices=[])
-                rel3 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="related_to")
+                rel3 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to")
                 tgt3 = gr.Dropdown(label="To", choices=[])
                 relationship_inputs.extend([src3, rel3, tgt3])
             
             with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Relationship 4**")
                 src4 = gr.Dropdown(label="From", choices=[])
-                rel4 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="related_to")
+                rel4 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to")
                 tgt4 = gr.Dropdown(label="To", choices=[])
                 relationship_inputs.extend([src4, rel4, tgt4])
             
             with gr.Column(scale=1, min_width=180):
                 gr.Markdown("**Relationship 5**")
                 src5 = gr.Dropdown(label="From", choices=[])
-                rel5 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="related_to")
+                rel5 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to")
                 tgt5 = gr.Dropdown(label="To", choices=[])
                 relationship_inputs.extend([src5, rel5, tgt5])