Update src/streamlit_app.py

src/streamlit_app.py

@@ -13,8 +13,8 @@ RATINGS_JSON_PATH = "/tmp/ratings.json"
 POSTER_PLACEHOLDER = "https://via.placeholder.com/300x450.png?text=No+Poster"
 TMDB_API_KEY = "d15fc170483ad01d6b3d59561432fefc"
 
-
-def get_poster_url(title, year=None):
+@st.cache_data(show_spinner=False)
+def get_tmdb_data(title, year=None):
     url = "https://api.themoviedb.org/3/search/movie"
     params = {
         "api_key": TMDB_API_KEY,
@@ -25,12 +25,15 @@ def get_poster_url(title, year=None):
     try:
         response = requests.get(url, params=params)
         if response.status_code == 200 and response.json()["results"]:
-            poster_path = response.json()["results"][0].get("poster_path")
-            if poster_path:
-                return f"https://image.tmdb.org/t/p/w500{poster_path}"
+            result = response.json()["results"][0]
+            poster_path = result.get("poster_path")
+            movie_id = result.get("id")
+            poster_url = f"https://image.tmdb.org/t/p/w500{poster_path}" if poster_path else POSTER_PLACEHOLDER
+            tmdb_link = f"https://www.themoviedb.org/movie/{movie_id}" if movie_id else None
+            return poster_url, tmdb_link
     except Exception:
         pass
-    return POSTER_PLACEHOLDER
+    return POSTER_PLACEHOLDER, None
 
 def load_ratings():
     if os.path.exists(RATINGS_JSON_PATH):
@@ -51,7 +54,6 @@ def load_movies():
     df["year"] = df["title"].str.extract(r'\((\d{4})\)').fillna("0").astype(int)
     df["clean_title"] = df["title"].str.replace(r'\(\d{4}\)', '', regex=True).str.strip()
     df["genres"] = df["genres"].fillna("Unknown")
-    df["poster"] = df.apply(lambda row: get_poster_url(row["clean_title"], row["year"]), axis=1)
     return df
 
 movie_df = load_movies()
@@ -167,13 +169,16 @@ if movie_id:
         movie_id = int(movie_id)
         movie_info = movie_df[movie_df["movieId"] == movie_id].iloc[0]
         st.title(movie_info["clean_title"])
+        poster_url, tmdb_link = get_tmdb_data(movie_info["clean_title"], movie_info["year"])
         col1, col2 = st.columns([1, 3])
         with col1:
-            st.image(movie_info["poster"], use_column_width=True)
+            st.image(poster_url, use_column_width=True)
         with col2:
             st.subheader("Details")
             st.write(f"**Genres:** {movie_info['genres']}")
             st.write(f"**Year:** {movie_info['year']}")
+            if tmdb_link:
+                st.markdown(f"[📎 View on TMDb]({tmdb_link})")
 
             user_rating = next((r["rating"] for r in all_ratings_data if r["movie_id"] == movie_id), None)
             if user_rating:
@@ -184,63 +189,37 @@ if movie_id:
         st.error("Could not load movie details.")
 
 elif page:
-    st.title("🎲 Rate Random Movies")
-
-    # Filter UI
-    genres_all = sorted(set("|".join(movie_df["genres"]).split("|")))
-    selected_genres = st.multiselect("Filter by genres", genres_all)
-
-    min_year = movie_df["year"].replace(0, pd.NA).dropna().min()
-    max_year = movie_df["year"].max()
-    selected_year_range = st.slider("Filter by release year", min_year, max_year, (min_year, max_year))
-
-    # Filter movie list
-    filtered_df = movie_df.copy()
-    if selected_genres:
-        genre_filter = filtered_df["genres"].apply(lambda g: any(gen in g for gen in selected_genres))
-        filtered_df = filtered_df[genre_filter]
-    filtered_df = filtered_df[(filtered_df["year"] >= selected_year_range[0]) & (filtered_df["year"] <= selected_year_range[1])]
-
-    filtered_titles = filtered_df["title"].tolist()
-
-    if "queue" not in st.session_state or st.session_state.get("filter_hash") != hash((tuple(selected_genres), selected_year_range)):
-        random.shuffle(filtered_titles)
-        st.session_state.queue = filtered_titles
-        st.session_state.index = 0
-        st.session_state.rated = []
-        st.session_state.filter_hash = hash((tuple(selected_genres), selected_year_range))
-
-    if st.session_state.index >= len(st.session_state.queue):
-        st.subheader("No movies based on your filters in the dataset")
-        st.session_state.quiz_history.append(st.session_state.rated.copy())
-        for r in st.session_state.rated:
-            save_rating_to_json(r)
-
-    else:
-        movie = st.session_state.queue[st.session_state.index]
-        movie_id_val = title_to_movie_id.get(movie)
-        movie_row = movie_df[movie_df["movieId"] == movie_id_val].iloc[0]
-
-        st.markdown(f"## **{movie_row['clean_title']}**")
-        st.caption(f"Genres: {movie_row['genres']}  |  Year: {movie_row['year']}")
-
-        col1, col2, col3, col4, col5 = st.columns(5)
-        selected = None
-        for i, col in enumerate([col1, col2, col3, col4, col5], 1):
-            if col.button("★" * i, key=f"rate_{i}"):
-                selected = i
-
-        if selected:
-            entry = {"movie_id": movie_id_val, "rating": selected, "timestamp": datetime.now().isoformat(), "source": "quiz"}
-            st.session_state.rated.append(entry)
-            save_rating_to_json(entry)
-            st.session_state.index += 1
-            st.rerun()
-
-        st.markdown("<br>", unsafe_allow_html=True)
-        if st.button("Didn't Watch"):
-            st.session_state.index += 1
-            st.rerun()
+    st.title("🍿 Rate Random Movies")
+    num_questions = 1
+    movie_pool = movie_df.sample(frac=1).reset_index(drop=True)
+    index = st.session_state.get("random_index", 0)
+
+    if index < len(movie_pool):
+        movie = movie_pool.iloc[index]
+        st.session_state["random_index"] = index + 1
+        poster_url, tmdb_link = get_tmdb_data(movie["clean_title"], movie["year"])
+
+        st.image(poster_url, width=300)
+        st.subheader(movie["clean_title"])
+        st.markdown(f"**Genres:** {movie['genres']}")
+        st.markdown(f"**Year:** {movie['year']}")
+        if tmdb_link:
+            st.markdown(f"[📎 View on TMDb]({tmdb_link})")
+
+        st.markdown("### Your Rating")
+        rating = st.radio("Rate this movie:", [1, 2, 3, 4, 5], horizontal=True, key=f"rating_{movie['movieId']}")
+        col1, col2 = st.columns([1, 1])
+        with col1:
+            if st.button("Submit Rating"):
+                save_rating_to_json({
+                    "movie_id": int(movie["movieId"]),
+                    "rating": rating,
+                    "timestamp": datetime.now().isoformat()
+                })
+                st.rerun()
+        with col2:
+            if st.button("Didn't Watch"):
+                st.rerun()
             
 elif search_query:
     # Movie Search Page