mudassir032 commited on
Commit
769923e
·
verified ·
1 Parent(s): b3978cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -24,8 +24,16 @@ movie = st.text_input("Search for movie")
24
  if 'random_movies' not in st.session_state:
25
  st.session_state['random_movies'] = random_selection(movies_df)
26
 
27
-
28
  if not movie:
29
- st.write(st.session_state['random_movies'])
30
-
 
 
 
 
 
 
 
 
31
 
 
24
  if 'random_movies' not in st.session_state:
25
  st.session_state['random_movies'] = random_selection(movies_df)
26
 
27
+ random_movies = st.session_state['random_movies']
28
  if not movie:
29
+ for i in st.session_state['random_movies']:
30
+ if st.button(i):
31
+ movie = i
32
+ st.write(movie)
33
+ movie_id = movies_df[movies_df['title'] == random_movies.iloc[movie]].index
34
+ similar_movies_idx = similarity_matrix[movie_id[0]].argsort()[::-1][1:6]
35
+ similar_movies = movies_df['title'].iloc[similar_movies_idx]
36
+ st.header("Recommended Movies")
37
+ for i in similar_movies:
38
+ st.write(i)
39