import streamlit as st
from logic import *

st.title('ArangoDB - PyG')


#---------------------
st.write('Making Graph')
global movie_mappings, user_mapping, ratings_df, m_id, id_map, sampled_md
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)

#---------------------