import streamlit as st
from logic2 import *
import streamlit.components.v1 as components

st.title('PyG - Movie Recommendation')
st.image('Image_graph.png', caption='Bi-Partite Graph')

st.write('Scatter Plot of embeddings')

HtmlFile = open("plot.html", 'r', encoding='utf-8')
source_code = HtmlFile.read() 
components.html(source_code)

# load Hgraph
st.write('Loading Data')
data = load_hetero_data()

# Load Model
st.write('Loading Model')
model = load_model(data)

#get user_id
user_id = st.number_input('Give User-Id')

# get recommendation
ans = get_recommendation(model,data,user_id)

st.write(ans)






#---------------------
'''
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,url,username,password,dbname = login_ArangoDB()
st.write(url)
st.write(username)
st.write(password)
st.write(dbname)
st.write('Loading Graph to ArangoDB')
movie_rec_db = load_data_to_ArangoDB(login)
st.write('Making PyG graph from ArangoDB')
global data
data,train_data, val_data, test_data = make_pyg_graph(movie_rec_db)
st.write('training graph')
model = load_model(train_data, val_data, test_data)
user_id = st.number_input('Insert a number')
ans = get_recommendation(model,data,user_id)
st.write(ans)
'''
#---------------------