Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import networkx as nx
 from pyvis.network import Network
 import base64
+import tempfile
+import os
 
 # Entity type colours (matching your NER tool)
 # Updated for better distinction between Person, Location, and Event
@@ -88,9 +90,9 @@ class NetworkGraphBuilder:
         return G
     
     def create_pyvis_graph(self, G):
-        """Create interactive PyVis visualisation"""
+        """Create interactive PyVis visualisation. Returns (iframe_html, standalone_html)"""
         if len(G.nodes) == 0:
-            return None
+            return None, None
         
         # Create PyVis network with dark theme
         net = Network(
@@ -213,6 +215,9 @@ class NetworkGraphBuilder:
         # Generate HTML
         html = net.generate_html()
         
+        # Store standalone HTML for export
+        standalone_html = html
+        
         # Encode as base64 data URI for iframe src
         html_bytes = html.encode('utf-8')
         b64_html = base64.b64encode(html_bytes).decode('utf-8')
@@ -227,7 +232,7 @@ class NetworkGraphBuilder:
         ></iframe>
         '''
         
-        return iframe_html
+        return iframe_html, standalone_html
 
 
 def collect_entities_from_records(
@@ -413,7 +418,7 @@ def generate_network_graph(
             return empty_html, "❌ **No entities to display.** Please enter entities in Step 1 first."
         
         # Create visualisation
-        graph_html = builder.create_pyvis_graph(G)
+        graph_html, standalone_html = builder.create_pyvis_graph(G)
         
         # Create statistics
         stats_html = f'''
@@ -487,6 +492,96 @@ def generate_network_graph(
         return error_html, f"❌ Error: {str(e)}"
 
 
+def export_network_graph(
+    p1, l1, e1, o1, d1,
+    p2, l2, e2, o2, d2,
+    p3, l3, e3, o3, d3,
+    p4, l4, e4, o4, d4,
+    p5, l5, e5, o5, d5,
+    p6, l6, e6, o6, d6,
+    p7, l7, e7, o7, d7,
+    p8, l8, e8, o8, d8,
+    src1, rel1, tgt1,
+    src2, rel2, tgt2,
+    src3, rel3, tgt3,
+    src4, rel4, tgt4,
+    src5, rel5, tgt5,
+    src6, rel6, tgt6,
+    src7, rel7, tgt7,
+    src8, rel8, tgt8
+):
+    """Export the network graph as a standalone HTML file"""
+    try:
+        builder = NetworkGraphBuilder()
+        
+        # Process each record
+        records = [
+            (p1, l1, e1, o1, d1),
+            (p2, l2, e2, o2, d2),
+            (p3, l3, e3, o3, d3),
+            (p4, l4, e4, o4, d4),
+            (p5, l5, e5, o5, d5),
+            (p6, l6, e6, o6, d6),
+            (p7, l7, e7, o7, d7),
+            (p8, l8, e8, o8, d8),
+        ]
+        
+        for record_id, (person, location, event, org, date) in enumerate(records, 1):
+            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)
+        
+        # Process relationships
+        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:
+            if source and target:
+                builder.add_relationship(source, target, rel_type)
+        
+        # Build graph
+        G = builder.build_graph()
+        
+        if len(G.nodes) == 0:
+            return None
+        
+        # Create visualisation
+        _, standalone_html = builder.create_pyvis_graph(G)
+        
+        # Save standalone HTML to a temporary file for download
+        if standalone_html:
+            export_file = tempfile.NamedTemporaryFile(
+                mode='w',
+                suffix='.html',
+                prefix='network_graph_',
+                delete=False,
+                encoding='utf-8'
+            )
+            export_file.write(standalone_html)
+            export_file.close()
+            return export_file.name
+        
+        return None
+    
+    except Exception as e:
+        return None
+
+
 def load_austen_example():
     """Load the Jane Austen Pride and Prejudice example"""
     return (
@@ -509,12 +604,6 @@ def load_austen_example():
     )
 
 
-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 (
@@ -563,9 +652,10 @@ def create_interface():
         ### How to use this tool:
         1. **📝 Enter entities** in the records below (or load an example to get started)
         2. **⚙️ Click "Process Entities"** to collect and prepare all entities for relationships
-        3. **🤝 Define relationships** between entities using the dropdowns
+        3. **🤝 Define relationships** between entities using the dropdowns (or type your own relationship type)
         4. **🎨 Click "Generate Network Graph"** to visualise
         5. **👁️ Explore** - drag nodes to rearrange, scroll to zoom, hover for details
+        6. **💾 Export (optional)** - click "Export as HTML" to download your graph as an interactive file
         """)
         
         gr.HTML("""
@@ -720,7 +810,7 @@ def create_interface():
         
         # ==================== STEP 3: RELATIONSHIPS (Grid) ====================
         gr.Markdown("## 🤝 Step 3: Define Relationships")
-        gr.Markdown("*Select entities from the dropdowns to create connections (click 'Process Entities' first)*")
+        gr.Markdown("*Select entities from the dropdowns to create connections (click 'Process Entities' first). You can also type your own relationship type.*")
         
         # Relationship inputs in a grid (5 columns for first 5)
         relationship_inputs = []
@@ -729,35 +819,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", allow_custom_value=True)
                 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", allow_custom_value=True)
                 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", allow_custom_value=True)
                 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", allow_custom_value=True)
                 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", allow_custom_value=True)
                 tgt5 = gr.Dropdown(label="To", choices=[])
                 relationship_inputs.extend([src5, rel5, tgt5])
         
@@ -767,21 +857,21 @@ def create_interface():
                 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")
+                    rel6 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to", allow_custom_value=True)
                     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")
+                    rel7 = gr.Dropdown(label="Type", choices=RELATIONSHIP_TYPES, value="Related to", allow_custom_value=True)
                     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")
+                    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])
         
@@ -799,6 +889,22 @@ def create_interface():
             with gr.Column(scale=1):
                 network_stats = gr.HTML()
         
+        # Export section
+        gr.Markdown("### 💾 Export Your Graph")
+        gr.Markdown("*Generate a graph first, then click export to download*")
+        with gr.Row():
+            export_btn = gr.Button("💾 Export as HTML", variant="secondary", size="sm")
+            export_file = gr.File(
+                label="Download Interactive HTML",
+                file_types=[".html"],
+                interactive=False
+            )
+        gr.HTML("""
+        <div style="background-color: #e8f4f8; border: 1px solid #bee5eb; border-radius: 8px; padding: 12px; margin: 10px 0;">
+            <span style="color: #0c5460;">💡 The exported HTML file is fully interactive — open it in any web browser to explore your network!</span>
+        </div>
+        """)
+        
         # Colour legend
         gr.HTML(f"""
         <div style="background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%); padding: 20px; border-radius: 10px; margin-top: 20px;">
@@ -863,7 +969,7 @@ def create_interface():
             ]
         )
         
-        # Generate graph
+        # Generate graph - outputs the visualisation and stats
         all_inputs = entity_inputs + relationship_inputs
         generate_btn.click(
             fn=generate_network_graph,
@@ -871,6 +977,13 @@ def create_interface():
             outputs=[network_plot, network_stats]
         )
         
+        # Export graph - separate button for downloading
+        export_btn.click(
+            fn=export_network_graph,
+            inputs=all_inputs,
+            outputs=[export_file]
+        )
+        
         # Model Information & Documentation section
         gr.HTML("""
         <hr style="margin: 40px 0 20px 0;">