Update logic2.py

logic2.py

@@ -83,8 +83,8 @@ global m_id
 with open('m_id.pkl', 'rb') as file:
     m_id = pickle.load(file)
 
-def get_movie(idx):
-    return id_map.loc[id_map['movieId'] == m_id[idx]].index
+def get_title(idx):
+    return id_map.loc[id_map['movieId'] == m_id[idx]].index[0]
 
 def get_recommendation(model,data,user_id):
 
@@ -98,7 +98,7 @@ def get_recommendation(model,data,user_id):
     # we will only select movies for the user where the predicting rating is =5
     rec_movie_ids = (pred == 5).nonzero(as_tuple=True)
     top_ten_recs = [rec_movies for rec_movies in rec_movie_ids[0].tolist()[:10]] 
-    top_ten_recs = [get_movie(movie_idx) for movie_idx in top_ten_recs]
+    top_ten_recs = [get_title(movie_idx) for movie_idx in top_ten_recs]
     return {'user': user_id, 'rec_movies': top_ten_recs}
 
 def make_1_hop_graph(data,user_id):
@@ -114,7 +114,7 @@ def make_1_hop_graph(data,user_id):
   for i in range(n):
     #print(i)
     Source = user_id
-    lab = get_title(movie_idxs[i])[0]
+    lab = get_title(movie_idxs[i])
     Target = movie_idxs[i] + 671 # Addition for sperating movie with user_id
     weight = ratings[i].item()