Spaces:
Sleeping
Sleeping
| 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 = st.columns([1, 1, 2]) | |
| with col1: # Back button #TODO fix | |
| if st.button("Back") and current_index > 0: | |
| st.session_state.current_index -= 1 | |
| st.rerun() | |
| with col2: # Next button | |
| if st.button("Next"): | |
| all_ratings = flatten_ratings(response) | |
| if any(rating == 0 for rating in all_ratings): | |
| st.warning("Please provide a rating before proceeding.") | |
| else: | |
| if current_index < len(data) - 1: | |
| st.session_state.current_index += 1 | |
| st.rerun() | |
| else: | |
| submit_feedback(current_index) | |
| with col3: # Save & Resume Later button | |
| if st.button("Exit & Resume Later"): | |
| last_response = st.session_state.responses[-1] | |
| all_last_ratings = flatten_ratings(last_response) | |
| if not any(rating == 0 for rating in all_last_ratings): | |
| submit_feedback(current_index) | |
| else: | |
| st.session_state.completed = True | |
| st.rerun() | |