Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,12 +4,16 @@ import random
 import os
 
 # Load movie data
-MOVIES_PATH = "/mnt/data/movies.csv"
+BASE_DIR = os.path.dirname(__file__)
+MOVIES_PATH = os.path.join(BASE_DIR, "src", "movies.csv")
 
 @st.cache_data
 def load_movies():
     df = pd.read_csv(MOVIES_PATH)
-    return df.dropna(subset=["title"])
+    df["year"] = df["title"].str.extract(r'\((\d{4})\)').fillna("Unknown")
+    df["clean_title"] = df["title"].str.replace(r'\(\d{4}\)', '', regex=True).str.strip()
+    df["genres"] = df["genres"].fillna("Unknown")
+    return df
 
 movie_df = load_movies()
 movie_titles = movie_df["title"].unique().tolist()
@@ -55,8 +59,24 @@ st.markdown("""
 query_params = st.query_params
 page = query_params.get("recommend")
 search_query = query_params.get("search")
+movie_id = query_params.get("movie_id")
 
-if page:
+if movie_id:
+    # Detail Page
+    try:
+        movie_id = int(movie_id)
+        movie_row = movie_df[movie_df["movieId"] == movie_id].iloc[0]
+        st.title(movie_row["clean_title"])
+        st.caption(f"Genres: {movie_row['genres']}")
+        st.caption(f"Release Year: {movie_row['year']}")
+        rating = st.radio("Rate this movie", [1, 2, 3, 4, 5], key=f"detail_{movie_id}")
+        if st.button("Submit Rating"):
+            st.session_state.rated.append({"movie": movie_row["title"], "rating": rating})
+            st.success(f"Thanks for rating '{movie_row['title']}' with ⭐{rating}")
+    except:
+        st.error("Movie not found.")
+
+elif page:
     # Quiz Flow
     if "queue" not in st.session_state:
         random.shuffle(movie_titles)
@@ -87,16 +107,34 @@ if page:
 elif search_query:
     # Movie Search Page
     st.title(f"🔍 Search Results for '{search_query}'")
-    matches = movie_df[movie_df["title"].str.contains(search_query, case=False, na=False)]
+    matches = movie_df[movie_df["clean_title"].str.contains(search_query, case=False, na=False)]
+
+    # Filters
+    genres_all = sorted(set("|".join(movie_df["genres"]).split("|")))
+    selected_genre = st.selectbox("Filter by genre", ["All"] + genres_all)
+    if selected_genre != "All":
+        matches = matches[matches["genres"].str.contains(selected_genre)]
+
+    sort_order = st.selectbox("Sort by", ["Title A-Z", "Year Desc", "Year Asc"])
+    if sort_order == "Year Desc":
+        matches = matches.sort_values(by="year", ascending=False)
+    elif sort_order == "Year Asc":
+        matches = matches.sort_values(by="year", ascending=True)
+    else:
+        matches = matches.sort_values(by="clean_title")
+
     if matches.empty:
         st.warning("No matching movies found.")
     for _, row in matches.iterrows():
         with st.expander(row["title"]):
             st.write(f"**Genres:** {row['genres']}")
+            st.write(f"**Year:** {row['year']}")
+            st.markdown(f"[➡️ Go to Detail Page](?movie_id={row['movieId']})")
             rating = st.radio(f"Rate '{row['title']}'", [1, 2, 3, 4, 5], key=row["title"])
             if st.button(f"Submit rating for {row['title']}", key=f"submit_{row['title']}"):
                 st.session_state.rated.append({"movie": row["title"], "rating": rating})
                 st.success(f"Thanks for rating '{row['title']}' with ⭐{rating}")
+
 else:
     # Home Page
     st.title("🏠 Welcome to MovieMatch")