import json from nicegui import ui # --- Configuration & Options --- FREQUENCY_OPTIONS = ["Daily", "Weekly", "Monthly", "Rarely"] FEATURE_OPTIONS = ["News Feed", "Marketplace", "Groups", "Messenger", "Stories"] OTHER_APP_OPTIONS = ["Instagram", "TikTok", "X/Twitter", "LinkedIn", "Snapchat", "Reddit"] GENDER_OPTIONS = ["Male", "Female", "Non-binary", "Prefer not to say"] class SurveyState: def __init__(self): self.frequency = None self.selected_features = [] self.feature_feedback = {} self.age = None self.gender = None self.location = "" def save_to_json(self): data = { "usage": {"frequency": self.frequency, "features_used": self.selected_features}, "feature_feedback": self.feature_feedback, "demographics": {"age": self.age, "gender": self.gender, "location": self.location}, } with open("survey_results.json", "w") as f: json.dump(data, f, indent=4) state = SurveyState() def create_survey(): container = ui.column().classes("w-full items-center q-pa-md") # --- PAGE 1: USAGE --- def show_page_1(): container.clear() with container: ui.label("Step 1: Facebook Usage").classes("text-h4 mb-4") with ui.card().classes("w-full max-w-lg p-6"): ui.label("How frequently do you use Facebook?").classes("text-bold") ui.radio(options=FREQUENCY_OPTIONS, on_change=lambda: update_p1_btn()).bind_value(state, "frequency") ui.label("What features do you use?").classes("text-bold mt-4") ui.select(options=FEATURE_OPTIONS, multiple=True, on_change=lambda: update_p1_btn()).bind_value( state, "selected_features" ).classes("w-full") next_btn = ui.button("Next", on_click=show_page_2).classes("mt-6 w-full") def update_p1_btn(): next_btn.set_enabled(state.frequency is not None and len(state.selected_features) > 0) update_p1_btn() # --- PAGE 2: RATINGS & FEEDBACK --- def show_page_2(): container.clear() # Ensure data structure exists for selected features for f in state.selected_features: if f not in state.feature_feedback: state.feature_feedback[f] = {"rating": 5, "likes": "", "other_apps": [], "improvements": ""} with container: ui.label("Step 2: Feedback").classes("text-h4 mb-4") with ui.row().classes("w-full max-w-lg justify-between mt-4"): ui.button("Back", on_click=show_page_1).props("outline") next_btn = ui.button("Next", on_click=show_page_3) ui.tooltip("Please answer all the questions (min 20 chars for improvements)").bind_visibility_from( next_btn, "enabled", backward=lambda v: not v ) def update_p2_btn(): valid = True for f in state.selected_features: f_data = state.feature_feedback[f] if f_data["rating"] < 4: if len(f_data.get("improvements", "")) < 20: valid = False next_btn.set_enabled(valid) update_p2_btn() for feature in state.selected_features: with ui.card().classes("w-full max-w-lg p-6 mb-4"): ui.label(f"Feature: {feature}").classes("text-h6 text-primary") slider = ui.slider(min=1, max=5, step=1, on_change=lambda: update_p2_btn()).bind_value( state.feature_feedback[feature], "rating" ) # High Rating Logic with ui.column().classes("w-full").bind_visibility_from(slider, "value", backward=lambda v: v >= 4): ui.textarea("What do you like about it?").classes("w-full").bind_value( state.feature_feedback[feature], "likes" ) # Low Rating Logic with ui.column().classes("w-full").bind_visibility_from(slider, "value", backward=lambda v: v < 4): ui.label("Other apps you use for this:") ui.select(options=OTHER_APP_OPTIONS, multiple=True).classes("w-full").bind_value( state.feature_feedback[feature], "other_apps" ) ui.textarea("Improvements (min 20 chars)", on_change=lambda: update_p2_btn()).classes( "w-full" ).bind_value(state.feature_feedback[feature], "improvements") # --- PAGE 3: DEMOGRAPHICS --- def show_page_3(): container.clear() with container: ui.label("Step 3: Demographics").classes("text-h4 mb-4") with ui.card().classes("w-full max-w-lg p-6"): ui.number("Age", format="%d").classes("w-full").bind_value(state, "age") ui.select(label="Gender", options=GENDER_OPTIONS).classes("w-full").bind_value(state, "gender") ui.input("Location").classes("w-full").bind_value(state, "location") with ui.row().classes("w-full justify-between mt-6"): ui.button("Back", on_click=show_page_2).props("outline") ui.button("Review Responses", on_click=show_confirmation).classes("bg-blue") # --- PAGE 4: CONFIRMATION (SUMMARY) --- def show_confirmation(): container.clear() with container: ui.label("Review Your Responses").classes("text-h4 mb-4") with ui.card().classes("w-full max-w-2xl p-6"): with ui.column().classes("gap-1"): ui.label("Usage Info").classes("text-bold text-lg") ui.label(f"Frequency: {state.frequency}") ui.label(f"Features: {', '.join(state.selected_features)}") ui.separator().classes("my-4") ui.label("Feedback").classes("text-bold text-lg") for f in state.selected_features: data = state.feature_feedback[f] ui.label(f"• {f}: Rating {data['rating']}/5") if data["rating"] >= 4: ui.label(f" Likes: {data['likes'] or 'N/A'}").classes("text-grey-7 ml-4") else: ui.label(f" Improvements: {data['improvements']}").classes("text-grey-7 ml-4") ui.separator().classes("my-4") ui.label("Demographics").classes("text-bold text-lg") ui.label(f"Age: {state.age} | Gender: {state.gender} | Location: {state.location}") with ui.row().classes("w-full justify-between mt-8"): ui.button("Back to Edit", on_click=show_page_3).props("outline") ui.button("Final Confirm & Submit", on_click=handle_final_submit).classes("bg-green") # --- PAGE 5: FINAL ACKNOWLEDGMENT --- def handle_final_submit(): state.save_to_json() container.clear() with container: with ui.card().classes("w-full max-w-lg p-12 items-center text-center"): ui.icon("check_circle", color="green").classes("text-6xl") ui.label("Thank You!").classes("text-h3 mt-4") ui.label("Your responses have been recorded successfully.").classes("text-lg text-grey-6") ui.label("This survey is now closed.").classes("mt-4 italic") show_page_1() create_survey() ui.run()