add spring layout

app.py

@@ -4,6 +4,7 @@ import pandas as pd
 import datasets
 import streamlit as st
 from streamlit_cytoscapejs import st_cytoscapejs
+import networkx as nx
 
 st.set_page_config(layout='wide')
 
@@ -97,6 +98,19 @@ node_info = pd.DataFrame({
     "gene_name": gene_names,
     "description": description})
 
+################################
+# Use NetworkX to layout graph #
+################################
+# note I think CytoscapeJS can layout graphs
+# but I'm unsure how to do it through the streamlit-cytoscapejs interface :(
+
+G = nx.Graph()
+G.add_weighted_edges_from(
+    [neighbors.gene_id_1.to_list(), neighbors.gene_id_2.to_list(), neighbors.coexp_score.to_list()])
+layout = nx.spring_layout(G)
+
+
+
 elements = []
 for i in range(len(gene_ids)):
     elements.append({
@@ -104,8 +118,8 @@ for i in range(len(gene_ids)):
             "id": gene_ids[i],
             "label": gene_name if gene_names[i] is not None else gene_ids[i]},
         "position": {
-            "x" : i * 10,
-            "y" : i * 10}})
+            "x" : layout[gene_ids[i]][0],
+            "y" : layout[gene_ids[i]][1]}})
 for i in range(len(neighbors.index)):
     edge = neighbors.iloc[i]
     elements.append({