Update app.py

app.py

@@ -1,4 +1,19 @@
 import streamlit as st
 from logic import *
 
-st.title('ArangoDB - PyG')
+st.title('ArangoDB - PyG')
+
+
+#---------------------
+st.write('Making Graph')
+movie_mappings, user_mapping, ratings_df, m_id, id_map, sampled_md = make_graph()
+st.write('Login to ArangoDB')
+login = login_ArangoDB()
+st.write('Loading Graph to ArangoDB')
+movie_rec_db = load_data_to_ArangoDB(login)
+st.write('Making PyG graph from ArangoDB')
+train_data, val_data, test_data = make_pyg_graph(movie_rec_db)
+st.write('training graph')
+train(train_data, val_data, test_data)
+
+#---------------------