Update app.py

app.py

@@ -110,17 +110,22 @@ node_info = pd.DataFrame({
 st.write(f"gene_id_1: {neighbors.gene_id_1.to_list()}")
 st.write(f"gene_id_2: {neighbors.gene_id_2.to_list()}")
 st.write(f"coexp_score: {neighbors.coexp_score.to_list()}")
+st.write(f"gene_id_1: {len(neighbors.gene_id_1.to_list())}")
+st.write(f"gene_id_2: {len(neighbors.gene_id_2.to_list())}")
+st.write(f"coexp_score: {len(neighbors.coexp_score.to_list())}")
 
 
 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()])
+    [neighbors.gene_id_1.to_list()[0:3],
+     neighbors.gene_id_2.to_list()[0:3],
+     neighbors.coexp_score.to_list()[0:3]])
 layout = nx.spring_layout(G)
 
 
 
 elements = []
-for i in range(len(gene_ids)):
+for i in range(3):# range(len(gene_ids)):
     elements.append({
         "data": {
             "id": gene_ids[i],
@@ -128,7 +133,7 @@ for i in range(len(gene_ids)):
         "position": {
             "x" : layout[gene_ids[i]][0],
             "y" : layout[gene_ids[i]][1]}})
-for i in range(len(neighbors.index)):
+for i in range(3): #range(len(neighbors.index)):
     edge = neighbors.iloc[i]
     elements.append({
         "data" : {