Spaces:
Running
Running
| from db.schema import Feedback, Response | |
| from db.crud import save_feedback, read | |
| import streamlit as st | |
| from datetime import datetime | |
| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| VALIDATION_CODE = os.getenv("VALIDATION_CODE") | |
| def submit_feedback(current_index): | |
| """Handles feedback submission to the database.""" | |
| feedback = Feedback( | |
| id=current_index + 1, | |
| user_id=st.session_state.username, | |
| time_stamp=datetime.now().isoformat(), | |
| responses=st.session_state.responses, | |
| ) | |
| try: | |
| save_feedback(feedback) | |
| st.session_state.completed = True | |
| st.rerun() | |
| except Exception as e: | |
| st.error(f"An error occurred while submitting feedback: {e}") | |
| def flatten_ratings(response): | |
| all_ratings = [] | |
| for model_ratings in response.model_ratings.values(): | |
| all_ratings.extend(model_ratings.query_v_ratings.values()) | |
| all_ratings.extend(model_ratings.query_p0_ratings.values()) | |
| all_ratings.extend(model_ratings.query_p1_ratings.values()) | |
| return all_ratings | |
| def navigation_buttons(data, response: Response): | |
| """Display navigation buttons.""" | |
| current_index = st.session_state.current_index | |
| col1, col2, col3, col4 = st.columns([1, 1, 1, 2]) | |
| with col1: # Back button #TODO fix: only gets ratings for the session, not from previous session | |
| if st.button("Back", disabled=st.session_state.current_index == 0): | |
| if current_index > 0: | |
| st.session_state.previous_ratings[ | |
| data.iloc[st.session_state.current_index]['config_id']] = response.model_ratings | |
| st.session_state.current_index -= 1 | |
| st.rerun() | |
| else: | |
| st.warning("You are at the beginning of the survey, can't go back.") | |
| # st.rerun() | |
| with col2: # Next button TODO might be buggy | |
| if st.button("Next", disabled=st.session_state.current_index == len(data) - 1): | |
| all_ratings = flatten_ratings(response) | |
| if any(rating == 0 for rating in all_ratings): | |
| st.warning("Please provide all ratings before proceeding.") | |
| else: | |
| if current_index < len(data) - 1: | |
| st.session_state.previous_ratings[data.iloc[st.session_state.current_index]['config_id']] = response.model_ratings | |
| st.session_state.current_index += 1 | |
| st.rerun() | |
| else: | |
| if st.button("Finish"): | |
| submit_feedback(current_index) | |
| with col3: # Submit button | |
| if st.button("Submit & Resume Later"): | |
| all_ratings = flatten_ratings(response) | |
| if any(rating == 0 for rating in all_ratings): | |
| st.warning("Please provide all ratings before submitting.") | |
| else: | |
| submit_feedback(current_index) | |
| with col4: # Save & Resume Later button | |
| if st.button("Exit & Resume Later"): | |
| submit_feedback(current_index) |