fixed label

app.py

@@ -5,7 +5,6 @@ from streamlit_agraph import agraph, Config
 
 from src.network import make_payload, get_graph
 from src.graph import build_tree_structure, create_hierarchical_view, tree_to_dot
-from src.viz import create_pyvis_network
 
 def get_id_from_input(val: str) -> Optional[int]:
     try:
@@ -23,14 +22,17 @@ def display_tree_summary(graph: Dict[str, Any], root_id: int) -> None:
     for node in tree.values():
         depth = node["depth"]
         depth_counts[depth] = depth_counts.get(depth, 0) + 1
-    col1, col2, col3 = st.columns(3)
+    
+    # display metrics in two columns to give more space
+    col1, col2 = st.columns(2)
     with col1:
         st.metric("Total Mathematicians", total_nodes)
-    with col2:
         st.metric("Generations Back", max_depth)
-    with col3:
+    with col2:
         root_name = tree.get(root_id, {}).get("name", "Unknown")
-        st.metric("Root", root_name)
+        st.write("**Root Mathematician:**")
+        st.write(root_name)
+
 def main():
     st.title("Math Genealogy Ancestor Tree")
     st.write("Interactive visualization of academic advisor relationships from the Mathematics Genealogy Project")
@@ -157,12 +159,32 @@ def main():
         
         st.divider()
         
+        # export to pdf button
+        import io
+        from graphviz import Source
+
+        dot = tree_to_dot(graph)
+        pdf_bytes = None
+        try:
+            src = Source(dot)
+            pdf_bytes = src.pipe(format="pdf")
+        except Exception as e:
+            st.warning(f"Could not generate PDF: {e}")
+
+        if pdf_bytes:
+            st.download_button(
+                label="Download Graph as PDF",
+                data=pdf_bytes,
+                file_name="math_genealogy_tree.pdf",
+                mime="application/pdf"
+            )
+        
         # visualization options
         st.subheader("Choose Visualization")
         
         viz_option = st.radio(
             "Select visualization type:",
-            ["Interactive Hierarchical Tree", "Interactive Network", "Traditional Graph (Graphviz)"],
+            ["Interactive Hierarchical Tree", "Traditional Graph (Graphviz)"],
             help="Different views for exploring the genealogy tree"
         )
         
@@ -222,36 +244,9 @@ def main():
             else:
                 st.warning("No data to display with current filters.")
         
-        elif viz_option == "Interactive Network":
-            st.write("**Interactive Network View** - Explore with zoom, pan, and physics simulation")
-            
-            with st.spinner("Generating interactive network..."):
-                html_content = create_pyvis_network(graph, root_id)
-                st.components.v1.html(html_content, height=650)
-        
         else:  # Traditional Graph
             st.write("**Traditional Graph View** - Standard graphviz layout")
-            dot = tree_to_dot(graph)
             st.graphviz_chart(dot)
-
-            # export to pdf button
-            import io
-            from graphviz import Source
-
-            pdf_bytes = None
-            try:
-                src = Source(dot)
-                pdf_bytes = src.pipe(format="pdf")
-            except Exception as e:
-                st.warning(f"Could not generate PDF: {e}")
-
-            if pdf_bytes:
-                st.download_button(
-                    label="Download Graph as PDF",
-                    data=pdf_bytes,
-                    file_name="math_genealogy_tree.pdf",
-                    mime="application/pdf"
-                )
         
         # search functionality
         st.divider()