Update app.py

app.py

@@ -88,10 +88,17 @@ option_selected = st.selectbox(
     movie_list
 )
 
+genre_list = ['Action','Romance','Adventure','Science Fiction','Comedy']
+genre_selected = st.selectbox(
+    'Type or Select Genre from Dropdown',
+    genre_list
+)
+
 if st.button('Show Recommendation'):
     recommended_movie_names, recommended_movie_posters = recommend(option_selected)
+    top_popular_movies = get_popular(top_popular)
     col1, col2, col3, col4, col5 = st.columns(5)
-    st.header("Movies Based on Movie Content: Similar Movies")
+    st.header("Movies Based on Content: Similar Movies")
     with col1:
         st.image(recommended_movie_posters[0], caption=recommended_movie_names[0])
     with col2:
@@ -104,7 +111,7 @@ if st.button('Show Recommendation'):
     with col5:
         st.image(recommended_movie_posters[4], caption=recommended_movie_names[4])
 
-    st.header("Top Popular Movies")
+    st.header("Movies Based on Popularity: Top Popular")
     popular = []
     for row in top_popular_movies.loc[:,['title','movie_id']].values:
         popular.append(row)
@@ -126,8 +133,8 @@ if st.button('Show Recommendation'):
         st.image(full_path, caption=popular[4][0])
 
 
-    st.header("Top Romantic Movies")
-    top_gener_based = top_genre_based_movies('Romance').head(5)
+    st.header("Movies Based on Genre: Top "+str(genre_selected)+" Movies")
+    top_gener_based = top_genre_based_movies(genre_selected).head(5)
     genre_popular = []
     for row in top_gener_based.loc[:, ['title', 'movie_id']].values:
         genre_popular.append(row)
@@ -148,27 +155,7 @@ if st.button('Show Recommendation'):
         full_path = fetch_poster(genre_popular[4][1])
         st.image(full_path, caption=genre_popular[4][0])
 
-    st.header("Top Action Movies")
-    top_gener_based = top_genre_based_movies('Action').head(5)
-    genre_popular = []
-    for row in top_gener_based.loc[:, ['title', 'movie_id']].values:
-        genre_popular.append(row)
-    col16, col17, col18, col19, col20 = st.columns(5)
-    with col16:
-        full_path = fetch_poster(genre_popular[0][1])
-        st.image(full_path, caption=genre_popular[0][0])
-    with col17:
-        full_path = fetch_poster(genre_popular[1][1])
-        st.image(full_path, caption=genre_popular[1][0])
-    with col18:
-        full_path = fetch_poster(genre_popular[2][1])
-        st.image(full_path, caption=genre_popular[2][0])
-    with col19:
-        full_path = fetch_poster(genre_popular[3][1])
-        st.image(full_path, caption=genre_popular[3][0])
-    with col20:
-        full_path = fetch_poster(genre_popular[4][1])
-        st.image(full_path, caption=genre_popular[4][0])
+