Update src/streamlit_app.py

src/streamlit_app.py

@@ -189,18 +189,22 @@ if movie_id:
 
             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)
-            if user_rating:
-                st.markdown(f"**Your Rating:** {render_star_rating(user_rating)}", unsafe_allow_html=True)
-            else:
-                st.markdown("### Rate this movie")
-                new_rating = st.radio("Your Rating:", [1, 2, 3, 4, 5], horizontal=True, key=rating_key)
-                if st.button("Submit Rating") and st.session_state[rating_key]:
+            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
+            
+            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"):
+                if new_rating:
                     save_rating_to_json({
                         "movie_id": movie_id,
-                        "rating": st.session_state[rating_key],
+                        "rating": new_rating,
                         "timestamp": datetime.now().isoformat()
                     })
+                    st.success("Rating saved.")
                     st.rerun()
+
     except Exception as e:
         st.error("Could not load movie details.")