jyshbgde commited on
Commit
4a150c8
·
1 Parent(s): 394ff69

changed zip to csv and converted recommended_movies into array of objects

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -5,8 +5,8 @@ import ast
5
 
6
 
7
  from datasets import load_dataset, concatenate_datasets
8
- movies = load_dataset("jyshbgde/cinescopeDataset", data_files = "tmdb_5000_movies.zip")
9
- credits = load_dataset("jyshbgde/cinescopeDataset", data_files = "tmdb_5000_credits.zip")
10
 
11
  movies = movies['train']
12
  credits = credits['train']
@@ -94,7 +94,9 @@ similarity = cosine_similarity(vectors)
94
  def recommend(movie):
95
  movie = movie.lower()
96
  if movie not in new_movies_df['lower_title'].values:
97
- return [["Check movie name again"]]
 
 
98
  else:
99
  movie_index = new_movies_df[new_movies_df['lower_title'] == movie].index[0]
100
  distances = similarity[movie_index]
@@ -102,9 +104,18 @@ def recommend(movie):
102
 
103
  recommended_movies = []
104
  for i in movies_list:
 
 
 
 
 
 
 
 
 
 
105
 
106
-
107
- recommended_movies.append([str(movies.iloc[i[0]]["movie_id"]) ,movies.iloc[i[0]]["title"], movies.iloc[i[0]]["overview"], movies.iloc[i[0]]["genres"], movies.iloc[i[0]]["cast"], movies.iloc[i[0]]["crew"] ])
108
 
109
  return recommended_movies
110
 
 
5
 
6
 
7
  from datasets import load_dataset, concatenate_datasets
8
+ movies = load_dataset("jyshbgde/cinescopeDataset", data_files = "tmdb_5000_movies.csv")
9
+ credits = load_dataset("jyshbgde/cinescopeDataset", data_files = "tmdb_5000_credits.csv")
10
 
11
  movies = movies['train']
12
  credits = credits['train']
 
94
  def recommend(movie):
95
  movie = movie.lower()
96
  if movie not in new_movies_df['lower_title'].values:
97
+ return [{
98
+ "error": "movie not found"
99
+ }]
100
  else:
101
  movie_index = new_movies_df[new_movies_df['lower_title'] == movie].index[0]
102
  distances = similarity[movie_index]
 
104
 
105
  recommended_movies = []
106
  for i in movies_list:
107
+
108
+ obj = {
109
+ "_id":str(movies.iloc[i[0]]["movie_id"]),
110
+ "title":movies.iloc[i[0]]["title"],
111
+ "overview":movies.iloc[i[0]]["overview"],
112
+ "genres": movies.iloc[i[0]]["genres"],
113
+ "casts":movies.iloc[i[0]]["cast"],
114
+ "crew": movies.iloc[i[0]]["crew"],
115
+ }
116
+ recommended_movies.append(obj)
117
 
118
+
 
119
 
120
  return recommended_movies
121