Update app.py

app.py

@@ -1,21 +1,31 @@
 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.plotly_chart(fig, use_container_width=True)
+
+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)