Update app.py

app.py

@@ -1,30 +1,50 @@
 import gradio as gr
 from pyvis.network import Network
 import networkx as nx
-nx_graph = nx.cycle_graph(10)
-nx_graph.nodes[1]['title'] = 'Number 1'
-nx_graph.nodes[1]['group'] = 1
-nx_graph.nodes[3]['title'] = 'I belong to a different group!'
-nx_graph.nodes[3]['group'] = 10
-nx_graph.add_node(20, size=20, title='couple', group=2)
-nx_graph.add_node(21, size=15, title='couple', group=2)
-nx_graph.add_edge(20, 21, weight=5)
-nx_graph.add_node(25, size=25, label='lonely', title='lonely node', group=3)
+import numpy as np
+import pandas as pd
+from datasets import load_dataset
 
-# graph above refer to: https://pyvis.readthedocs.io/en/latest/tutorial.html#networkx-integration
 
-# nt_notebook = Network('500px', '500px', notebook=True, cdn_resources='remote')
-# populates the nodes and edges data structures
-# nt_notebook.from_nx(nx_graph)
-# nt_notebook.show('nx.html')
+Secret_token = os.getenv('token')
 
+dataset = load_dataset('FDSRashid/hadith_info',data_files = 'Basic_Edge_Information.csv', token = Secret_token, split = 'train')
+dataset2 = load_dataset('FDSRashid/hadith_info',data_files = 'Taraf_Info.csv', token = Secret_token, split = 'train')
+edge_info = dataset.to_pandas()
+taraf_info = dataset2.to_pandas()
+cities = taraf_info['City'].unique().tolist()
+min_year = int(taraf_info['Year'].min())
+max_year = int(taraf_info['Year'].max())
 
-def needs_analysis():
+def subsetEdges(city, year):
+  info = taraf_info[(taraf_info['Year'] == year) & (taraf_info['City'] == city)]
+  narrators = edge_info[edge_info['Edge_ID'].isin(info['ID'].unique())]
+  return narrators
+def splitIsnad(dataframe):
+  teacher_student =dataframe['Edge_Name'].str.split(' TO ')
+  dataframe['Teacher'] = teacher_student.apply(lambda x: x[0])
+  dataframe['Student'] = teacher_student.apply(lambda x: x[1])
+  return dataframe
+
+    
+def network_visualizer(city, year):
+    edge_15 = splitIsnad(subsetEdges(city, year)).sample(300)
     nt = Network()
-    nt.from_nx(nx_graph)
-    html = nt.generate_html()
-    #need to remove ' from HTML
+    for _, row in edge_15.iterrows():
+    source = row['Teacher']
+    target = row['Student']
+    attribute_value = row['Hadiths']
+    edge_color = value_to_hex(attribute_value)
+
+    net.add_node(source, color=value_to_hex(attribute_value), font = {'size':30, 'color': 'orange'})
+    net.add_node(target, color=value_to_hex(attribute_value) , font = {'size': 20, 'color': 'red'})
+    net.add_edge(source, target, color=edge_color, value=attribute_value)
+
+
+    net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)
+    html = net.generate_html()
     html = html.replace("'", "\"")
+
     
     return f"""<iframe style="width: 100%; height: 600px;margin:0 auto" name="result" allow="midi; geolocation; microphone; camera; 
     display-capture; encrypted-media;" sandbox="allow-modals allow-forms 
@@ -33,12 +53,9 @@ def needs_analysis():
     allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>"""
 
 
-demo = gr.Interface(
-    needs_analysis,
-    inputs=None,
-    outputs=gr.outputs.HTML(),
-    title="pyvis_in_gradio",
-    allow_flagging='never'
-)
-
-demo.launch()
+with gr.Blocks() as demo:
+  Places = gr.Dropdown(choices = cities, value = 'المدينه', label = 'Location')
+  Last_Year = gr.Slider(min_year, max_year, value = 9, label = 'End', info = 'Choose the  year to display Narrators')
+  btn = gr.Button('Submit')
+  btn.click(fn = network_visualizer, inputs = [Places, Last_Year], outputs = gr.HTML())
+  demo.launch()