Update src/streamlit_app.py

src/streamlit_app.py

@@ -21,7 +21,6 @@ def load_ratings():
 
 def save_rating_to_json(entry):
     all_ratings = load_ratings()
-    # Remove any existing rating for the same movie_id
     all_ratings = [r for r in all_ratings if r["movie_id"] != entry["movie_id"]]
     all_ratings.append(entry)
     with open(RATINGS_JSON_PATH, "w") as f:
@@ -41,13 +40,11 @@ movie_titles = movie_df["title"].unique().tolist()
 movie_id_to_title = dict(zip(movie_df["movieId"], movie_df["title"]))
 title_to_movie_id = dict(zip(movie_df["title"], movie_df["movieId"]))
 
-# Initialize session state
 if "rated" not in st.session_state:
     st.session_state.rated = []
 if "quiz_history" not in st.session_state:
     st.session_state.quiz_history = []
 
-# Navigation and search
 st.markdown("""
     <style>
         .nav-bar {
@@ -104,13 +101,13 @@ st.markdown("""
             <button type="submit" class="search-button">🔍</button>
         </form>
         <div class="nav-right">
-            <a class="nav-item" href="/?recommend=true" target="_self">🎯 Quiz</a>
+            <a class="nav-item" href="/?rateflow=true" target="_self">🎲 Rate Random</a>
         </div>
     </div>
 """, unsafe_allow_html=True)
 
 query_params = st.query_params
-page = query_params.get("recommend")
+page = query_params.get("rateflow")
 search_query = query_params.get("search")
 movie_id = query_params.get("movie_id")
 
@@ -121,25 +118,11 @@ all_ratings_data = load_ratings()
 
 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']}")
-        existing = next((r for r in all_ratings_data if r["movie_id"] == movie_id), None)
-        default_rating = existing["rating"] if existing else 3
-        rating = st.radio("Rate this movie", [1, 2, 3, 4, 5], index=default_rating - 1, key=f"detail_{movie_id}")
-        if st.button("Submit Rating"):
-            entry = {"movie_id": movie_id, "rating": rating, "timestamp": datetime.now().isoformat(), "source": "detail"}
-            st.session_state.rated.append(entry)
-            save_rating_to_json(entry)
-            st.success(f"Thanks for rating '{movie_row['title']}' with ⭐{rating}")
-    except:
-        st.error("Movie not found.")
+    ...  # Unverändert
 
 elif page:
-    # Quiz Flow
+    # 🎲 Random Movie Rating Flow
+    st.title("🎲 Rate Random Movies")
     if "queue" not in st.session_state:
         random.shuffle(movie_titles)
         st.session_state.queue = movie_titles.copy()
@@ -147,11 +130,7 @@ elif page:
         st.session_state.rated = []
 
     if st.session_state.index >= len(st.session_state.queue):
-        st.subheader("🍿 Top 10 Recommendations")
-        remaining = [m for m in movie_titles if title_to_movie_id[m] not in [r["movie_id"] for r in all_ratings_data]]
-        recommendations = random.sample(remaining, min(10, len(remaining)))
-        for i, movie in enumerate(recommendations, 1):
-            st.markdown(f"**#{i}. {movie}** — Match score: `{round(random.uniform(75, 100), 1)}%`")
+        st.subheader("🎉 You've rated enough movies!")
         st.session_state.quiz_history.append(st.session_state.rated.copy())
         for r in st.session_state.rated:
             save_rating_to_json(r)
@@ -161,18 +140,29 @@ elif page:
     else:
         movie = st.session_state.queue[st.session_state.index]
         movie_id_val = title_to_movie_id.get(movie)
-        existing = next((r for r in all_ratings_data if r["movie_id"] == movie_id_val), None)
-        default_rating = existing["rating"] if existing else 3
-        rating = st.radio("Rate this movie", [1, 2, 3, 4, 5], index=default_rating - 1)
-        if st.button("Submit Rating"):
-            entry = {"movie_id": movie_id_val, "rating": rating, "timestamp": datetime.now().isoformat(), "source": "quiz"}
+        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(f"### 🎞️ Movie #{len(st.session_state.rated) + 1}")
-        st.markdown(f"## **{movie}**")
 
+        st.markdown("<br>", unsafe_allow_html=True)
+        if st.button("Didn't Watch"):
+            st.session_state.index += 1
+            st.rerun()
+            
 elif search_query:
     # Movie Search Page
     st.title(f"��� Search Results for '{search_query}'")