from db.schema import Feedback, Response from db.crud import ingest import pandas as pd import streamlit as st from datetime import datetime import os from dotenv import load_dotenv import json from typing import Dict, List load_dotenv() class SurveyState: """Class to handle survey state management""" def __init__(self): self.state_file = "survey_states.json" def save_state(self, prolific_id: str, current_state: Dict) -> None: """Save current state to file""" try: if os.path.exists(self.state_file): with open(self.state_file, 'r') as f: states = json.load(f) else: states = {} states[prolific_id] = current_state with open(self.state_file, 'w') as f: json.dump(states, f) except Exception as e: st.error(f"Error saving state: {e}") def load_state(self, prolific_id: str) -> Dict: """Load state for a specific user""" try: if os.path.exists(self.state_file): with open(self.state_file, 'r') as f: states = json.load(f) return states.get(prolific_id, {}) except Exception as e: st.error(f"Error loading state: {e}") return {} def validate_prolific_id(prolific_id: str) -> bool: return bool(prolific_id.strip()) def initialization(): """Initialize session state variables.""" if "current_index" not in st.session_state: st.session_state.current_index = 0 if "prolific_id" not in st.session_state: st.session_state.prolific_id = None if "responses" not in st.session_state: st.session_state.responses = [] if "completed" not in st.session_state: st.session_state.completed = False if "survey_state" not in st.session_state: st.session_state.survey_state = SurveyState() def prolific_id_screen(): """Display the Prolific ID entry screen.""" st.title("Welcome to the Feedback Survey") prolific_id_input = st.text_input("Enter your Prolific ID and press ENTER:") if prolific_id_input: if validate_prolific_id(prolific_id_input): st.session_state.prolific_id = prolific_id_input # Load previous state if exists saved_state = st.session_state.survey_state.load_state(prolific_id_input) if saved_state: st.session_state.current_index = saved_state.get('current_index', 0) st.session_state.responses = saved_state.get('responses', []) st.success("Previous progress loaded! Continuing from where you left off.") else: st.success("Prolific ID recorded! Starting new survey.") st.rerun() else: st.warning("Invalid Prolific ID. Please check and try again.") def questions_screen(data): """Display the questions screen.""" current_index = st.session_state.current_index config = data.iloc[current_index] # Progress bar progress = (current_index + 1) / len(data) st.progress(progress) st.write(f"Question {current_index + 1} of {len(data)}") st.subheader(f"Config ID: {config['config_id']}") # Create tabs for better organization context_tab, questions_tab = st.tabs(["Context", "Questions"]) with context_tab: st.write(f"**Persona:** {config['persona']}") st.write(f"**Filters:** {config['filters']}") st.write(f"**Cities:** {config['city']}") # Expandable context with st.expander("Show Context", expanded=False): st.write(config['context']) options = [0, 1, 2, 3, 4, 5] with questions_tab: st.write(f"**Query_v:** {config['query_v']}") rating_v = st.radio("Rate this config:", options, key=f"rating_v_{current_index}") st.write(f"**Query_p0:** {config['query_p0']}") rating_p0 = st.radio("Rate this config:", options, key=f"rating_p0_{current_index}") st.write(f"**Query_p1:** {config['query_p1']}") rating_p1 = st.radio("Rate this config:", options, key=f"rating_p1_{current_index}") comment = st.text_area("Comments (Optional):") response = Response(config_id=config["config_id"], rating_v=rating_v, rating_p0=rating_p0, rating_p1=rating_p1, comment=comment, timestamp=datetime.now().isoformat()) print(response) if len(st.session_state.responses) > current_index: st.session_state.responses[current_index] = response else: st.session_state.responses.append(response) navigation_buttons(data, rating_v, rating_p0, rating_p1) def navigation_buttons(data, rating_v, rating_p0, rating_p1): """Display navigation buttons.""" current_index = st.session_state.current_index col1, col2, col3 = st.columns([1, 1, 2]) with col1: # Back button if st.button("Back") and current_index > 0: st.session_state.current_index -= 1 st.rerun() with col2: # Next button if st.button("Next"): if rating_v == 0 or rating_p0 == 0 or rating_p1 ==0: st.warning("Please provide a rating before proceeding.") else: if current_index < len(data) - 1: st.session_state.current_index += 1 st.rerun() else: feedback = Feedback( id=current_index + 1, user_id=st.session_state.prolific_id, time_stamp=datetime.now().isoformat(), responses=st.session_state.responses, ) try: ingest(feedback) st.session_state.completed = True st.rerun() except Exception as e: st.error(f"An error occurred while submitting feedback: {e}") def exit_screen(): """Display exit screen""" st.markdown("""

Thank you for participating!

Your responses have been saved successfully.

You can safely close this window or start a new survey.

""", unsafe_allow_html=True) if st.button("Start New Survey"): reset_survey() st.rerun() def submit_survey(): """Submit the complete survey""" try: feedback = Feedback( id=st.session_state.current_index, user_id=st.session_state.prolific_id, time_stamp=datetime.now().isoformat(), responses=st.session_state.responses, ) print(feedback) ingest(feedback) st.session_state.completed = True st.rerun() except Exception as e: st.error(f"An error occurred while submitting feedback: {e}") def reset_survey(): """Reset the survey state to start over.""" st.session_state.prolific_id = None st.session_state.current_index = 0 st.session_state.responses = [] st.session_state.completed = False def ui(): """Main function to control the survey flow.""" data = pd.read_csv("data/gemini_results_subset.csv")[:3] initialization() if st.session_state.completed: # st.title("Survey Completed") # st.success("Thank you! Your survey has been submitted successfully.") # if st.button("Start New Survey"): # reset_survey() # st.rerun() exit_screen() return if st.session_state.prolific_id is None: prolific_id_screen() else: questions_screen(data) if __name__ == "__main__": ui()