Update src/streamlit_app.py

src/streamlit_app.py

@@ -190,15 +190,16 @@ all_ratings_data = load_ratings_cached()
 if movie_id:
     try:
         movie_id = int(movie_id)
-
         match = movie_df[movie_df["movieId"] == movie_id]
+
         if match.empty:
             st.error(f"Movie with ID {movie_id} not found.")
             st.stop()
-        movie_info = match.iloc[0]
 
+        movie_info = match.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:
             if poster_url and "placeholder.com" not in poster_url:
@@ -209,6 +210,7 @@ if movie_id:
                     No picture available
                 </div>
                 """, unsafe_allow_html=True)
+
         with col2:
             st.subheader("Details")
             st.write(f"**Genres:** {movie_info['genres']}")
@@ -216,12 +218,11 @@ if movie_id:
             if tmdb_link:
                 st.markdown(f"<a href='{tmdb_link}' target='_blank'>View on TMDb</a>", unsafe_allow_html=True)
 
-            rating_key = f"detail_rating_{movie_id}"
-            user_rating = next((r["rating"] for r in all_ratings_data if r["movie_id"] == movie_id), None)
             st.markdown("### Your Rating")
             existing_rating = next((r["rating"] for r in all_ratings_data if r["movie_id"] == movie_id), None)
-            initial_index = existing_rating - 1 if existing_rating else None
+            initial_index = (existing_rating - 1) if existing_rating else 0
 
+            rating_key = f"detail_rating_{movie_id}"
             new_rating = st.radio("Rate this movie:", [1, 2, 3, 4, 5], horizontal=True, index=initial_index, key=rating_key)
 
             if st.button("Submit Rating", key=f"submit_rating_btn_{movie_id}"):
@@ -232,17 +233,15 @@ if movie_id:
                 })
 
                 st.success("Rating saved.")
-                st.experimental_set_query_params(movie_id=movie_info["movieId"])
-                st.experimental_rerun()
+                st.rerun()  # KEIN set_query_params notwendig
 
     except Exception as e:
-        st.error("Could not load movie details.")
+        st.error(f"Could not load movie details: {e}")
 
 
 elif page:
     st.title("Rate Random Movies")
 
-    # Genres extrahieren
     all_genres = sorted(set(
         genre.strip()
         for sublist in movie_df["genres"].dropna().str.split("|")
@@ -251,7 +250,6 @@ elif page:
 
     selected_genres = st.multiselect("Select genres:", options=all_genres, placeholder="Choose a genre")
 
-    # Initialisieren (einmal)
     if "rating_mode_state" not in st.session_state:
         st.session_state.rating_mode_state = {
             "movie_pool": pd.DataFrame(),
@@ -260,13 +258,11 @@ elif page:
             "active": False
         }
 
-    # Session starten nur per Button
     if st.button("Start Rating Session"):
         if not selected_genres:
             st.warning("Please select at least one genre.")
             st.stop()
 
-        # Filter anwenden
         genre_filter = movie_df["genres"].apply(
             lambda g: any(genre in g.split("|") for genre in selected_genres)
         )
@@ -282,7 +278,6 @@ elif page:
         st.session_state.rating_mode_state["active"] = True
         st.rerun()
 
-    # Zugriff auf Zustand
     mode = st.session_state.rating_mode_state
     if not mode["active"]:
         st.info("Please select genres and start a session.")
@@ -292,8 +287,8 @@ elif page:
     index = mode["index"]
 
     if movie_pool.empty or index >= len(movie_pool):
-        st.success("You've rated all movies in this session.")
-        if st.button("Start New Session"):
+        st.success("✅ You've rated all movies in this session.")
+        if st.button("🔁 Start New Session"):
             mode["active"] = False
             st.rerun()
         st.stop()
@@ -320,8 +315,19 @@ elif page:
             st.markdown(f"<a href='{tmdb_link}' target='_blank'>View on TMDb</a>", unsafe_allow_html=True)
 
         st.markdown(f"### Movie {index + 1} of {len(movie_pool)}")
+
+        # Bestehende Bewertung laden
+        existing_rating = next((r["rating"] for r in all_ratings_data if r["movie_id"] == movie["movieId"]), None)
+        default_rating = existing_rating if existing_rating else 1
+
         rating_key = f"rating_{movie['movieId']}_{index}"
-        rating = st.radio("Rate this movie:", [1, 2, 3, 4, 5], horizontal=True, key=rating_key)
+        rating = st.radio(
+            "Rate this movie:",
+            options=[1, 2, 3, 4, 5],
+            horizontal=True,
+            key=rating_key,
+            index=[1, 2, 3, 4, 5].index(default_rating)
+        )
 
         col_submit, col_skip = st.columns([1, 1])
         with col_submit:
@@ -344,6 +350,7 @@ elif page:
 
 
 
+
 elif search_query:
     st.title(f"Search Results for '{search_query}'")