Commit

Pages/Recommender App.py

@@ -51,11 +51,27 @@ def fetch_poster(anime_id):
 
 
 # Recommender System
-def recommend(anime):
-    index = anime_data[anime_data["title"] == anime].index[0]
+def recommend(anime, genre=None):
+    if genre == None:
+        index = (
+            anime_data[anime_data["title"] == anime]
+            .sort_values("score", ascending=False)
+            .index[0]
+        )
+    elif genre != None:
+        index = (
+            anime_data[
+                (anime_data["title"] == anime)
+                | (anime_data["genres"].str.contains(genre))
+            ]
+            .sort_values("score", ascending=False)
+            .index[0]
+        )
+    # index = anime_data[anime_data["title"] == anime].index[0]
     distances = sorted(
         list(enumerate(similarity[index])), reverse=True, key=lambda x: x[1]
     )
+
     recommended_anime_names = []
     recommended_anime_posters = []
     recommended_anime_urls = []
@@ -185,7 +201,42 @@ def top_animes_custom(anime_select):
 
 # Function to display the top 8 animes with user given anime name and genre
 def top_animes_custom_genres(anime_select, genre_select):
-    pass
+    (
+        recommended_anime_names,
+        recommended_anime_posters,
+        recommended_anime_urls,
+    ) = recommend(anime_select, genre_select)
+    with st.container():
+        col1, col2, col3, col4 = st.columns(4)
+        with col1:
+            st.write(f"[{recommended_anime_names[0]}]({recommended_anime_urls[0]})")
+            st.image(recommended_anime_posters[0])
+        with col2:
+            st.write(f"[{recommended_anime_names[1]}]({recommended_anime_urls[1]})")
+            st.image(recommended_anime_posters[1])
+        with col3:
+            st.write(f"[{recommended_anime_names[2]}]({recommended_anime_urls[2]})")
+            st.image(recommended_anime_posters[2])
+        with col4:
+            st.write(f"[{recommended_anime_names[3]}]({recommended_anime_urls[3]})")
+            st.image(recommended_anime_posters[3])
+
+    st.divider()
+
+    with st.container():
+        col5, col6, col7, col8 = st.columns(4)
+        with col5:
+            st.write(f"[{recommended_anime_names[4]}]({recommended_anime_urls[4]})")
+            st.image(recommended_anime_posters[4])
+        with col6:
+            st.write(f"[{recommended_anime_names[5]}]({recommended_anime_urls[5]})")
+            st.image(recommended_anime_posters[5])
+        with col7:
+            st.write(f"[{recommended_anime_names[6]}]({recommended_anime_urls[6]})")
+            st.image(recommended_anime_posters[6])
+        with col8:
+            st.write(f"[{recommended_anime_names[7]}]({recommended_anime_urls[7]})")
+            st.image(recommended_anime_posters[7])
 
 
 # Recommender Page
@@ -212,12 +263,19 @@ def recommender_page():
     genre_select = st.selectbox("Select a Genre", get_genres(), key="genre_select")
 
     if st.button("Recommendation"):
+        st.divider()
         if anime_select == "Top 8 Animes" and genre_select == "All Genres":
             top_animes()
+            st.divider()
         elif anime_select == "Top 8 Animes" and genre_select != "All Genres":
             top_animes_genres(genre_select)
-        else:
+            st.divider()
+        elif anime_select != "Top 8 Animes" and genre_select == "All Genres":
             top_animes_custom(anime_select)
+            st.divider()
+        elif anime_select != "Top 8 Animes" and genre_select != "All Genres":
+            top_animes_custom_genres(anime_select, genre_select)
+            st.divider()
 
 
 if __name__ == "__main__":

app.py

@@ -2,17 +2,6 @@ import streamlit as st
 from st_pages import Page, show_pages
 from PIL import Image
 
-# Configuring Pages
-
-show_pages(
-    [
-        Page(r"app.py", "Homepage", "🏠"),
-        Page(r"Pages/Recommender App.py", "Anime Recommender", "📺"),
-        Page(r"Pages/About.py", "About", "👋"),
-    ]
-)
-
-
 # Make the page full width
 im = Image.open(r"ninja.png")
 st.set_page_config(
@@ -23,6 +12,16 @@ st.set_page_config(
     menu_items={"About": "This Anime Recommender App is made by Harshit Singh."},
 )
 
+# Configuring Pages
+
+show_pages(
+    [
+        Page(r"app.py", "Homepage", "🏠"),
+        Page(r"Pages/Recommender App.py", "Anime Recommender", "📺"),
+        Page(r"Pages/About.py", "About", "👋"),
+    ]
+)
+
 
 # Home Page
 def home_page():