Update app.py

app.py

@@ -69,7 +69,7 @@ from datasets import load_dataset
 dataset = load_dataset('Seetha/Visualization', streaming=True)
 df = pd.DataFrame.from_dict(dataset['train'])
 
-st.write(df)
+
 def main():
 
   st.title("Text to Causal Knowledge Graph")
@@ -463,7 +463,7 @@ def main():
     writer.close()
   #df = pd.read_csv('final_data.csv', index_col=0)
 #474-515
-  # # Convert to JSON format
+  # Convert to JSON format
   json_data = []
   for row in df_tab.index:
     for col in df_tab.columns:
@@ -473,12 +473,16 @@ def main():
             'value': int(df_tab.loc[row, col])
         })
   dat = json.dumps(json_data) 
-  with open('level2.json','r+') as fi:
-      data = fi.read()
-    #st.write(data)
-      fi.seek(0)
-      fi.write(dat)
-      fi.truncate()
+  level2_df = pd.DataFrame(dat)
+  st.write(level2_df)
+  # with open('level2.json','r+') as fi:
+  #     data = fi.read()
+  #   #st.write(data)
+  #     fi.seek(0)
+  #     fi.write(dat)
+  #     fi.truncate()
+
+  updated_dataset = dataset.map(lambda example: {'new_value': level2_df['value'], 'new_source':level2_df['source'], 'new_target': level2_df['target']},remove_columns=['value','source','target'])
 
   df_final1.to_csv('predictions.csv')
   csv_file = "predictions.csv"