Hide needs selector until Stage 5 - clean opening page

app.py

@@ -337,12 +337,14 @@ def process_message(user_input, history, stage, session_data, somatic_enabled, s
         if response is None:
             error_msg = "**Setup Required:** Please add ANTHROPIC_API_KEY in Space Settings > Repository secrets, then restart."
             history.append({"user": user_input, "assistant": error_msg})
-            return format_history(history), "", history, stage, session_data, f"Stage {stage}: {STAGE_NAMES[stage]}"
+            show_needs = (stage == 5)
+            return format_history(history), "", history, stage, session_data, f"Stage {stage}: {STAGE_NAMES[stage]}", gr.update(visible=show_needs)
 
     except Exception as e:
         error_msg = f"**Error:** {str(e)}"
         history.append({"user": user_input, "assistant": error_msg})
-        return format_history(history), "", history, stage, session_data, f"Stage {stage}: {STAGE_NAMES[stage]}"
+        show_needs = (stage == 5)
+        return format_history(history), "", history, stage, session_data, f"Stage {stage}: {STAGE_NAMES[stage]}", gr.update(visible=show_needs)
 
     # Update session data
     if stage == 1:
@@ -366,13 +368,17 @@ def process_message(user_input, history, stage, session_data, somatic_enabled, s
         next_stage = 1
         session_data = {}
 
+    # Show needs selector only at stage 5
+    show_needs = (next_stage == 5)
+
     return (
         format_history(history),
         "",
         history,
         next_stage,
         session_data,
-        f"Stage {next_stage}/13: {STAGE_NAMES[next_stage]}"
+        f"Stage {next_stage}/13: {STAGE_NAMES[next_stage]}",
+        gr.update(visible=show_needs)
     )
 
 def reset_session():
@@ -384,6 +390,7 @@ def reset_session():
         1,
         {},
         "Stage 1/13: Share Your Message",
+        gr.update(visible=False),
         [], [], [], []
     )
 
@@ -426,16 +433,16 @@ with gr.Blocks(
                 label="Conversation"
             )
 
-            # Needs selector (collapsible)
-            with gr.Accordion("Select Your Needs (for Stage 5)", open=False):
-                gr.Markdown("**Select needs that resonate:**")
+            # Needs selector (hidden until Stage 5)
+            with gr.Accordion("Select Your Needs", open=True, visible=False) as needs_accordion:
+                gr.Markdown("**Select 1-3 needs that resonate most deeply:**")
                 with gr.Row():
-                    needs1 = gr.CheckboxGroup(NEEDS_LIST["Emotional Needs"], label="Emotional")
-                    needs2 = gr.CheckboxGroup(NEEDS_LIST["Relational Needs"], label="Relational")
+                    needs1 = gr.CheckboxGroup(NEEDS_LIST["Emotional Needs"], label="Emotional Needs")
+                    needs2 = gr.CheckboxGroup(NEEDS_LIST["Relational Needs"], label="Relational Needs")
                 with gr.Row():
-                    needs3 = gr.CheckboxGroup(NEEDS_LIST["Value Needs"], label="Values")
-                    needs4 = gr.CheckboxGroup(NEEDS_LIST["Lifestyle Needs"], label="Lifestyle")
-                selected_needs = gr.Textbox(label="Selected Needs", interactive=False)
+                    needs3 = gr.CheckboxGroup(NEEDS_LIST["Value Needs"], label="Value Needs")
+                    needs4 = gr.CheckboxGroup(NEEDS_LIST["Lifestyle Needs"], label="Lifestyle Needs")
+                selected_needs = gr.Textbox(label="Your Selected Needs", interactive=False)
 
             # Input
             user_input = gr.Textbox(
@@ -482,13 +489,13 @@ with gr.Blocks(
     submit_btn.click(
         process_message,
         [user_input, history_state, stage_state, session_state, somatic_check, selected_needs],
-        [conversation, user_input, history_state, stage_state, session_state, stage_display]
+        [conversation, user_input, history_state, stage_state, session_state, stage_display, needs_accordion]
     )
 
     user_input.submit(
         process_message,
         [user_input, history_state, stage_state, session_state, somatic_check, selected_needs],
-        [conversation, user_input, history_state, stage_state, session_state, stage_display]
+        [conversation, user_input, history_state, stage_state, session_state, stage_display, needs_accordion]
     )
 
     # Reset handler
@@ -496,7 +503,7 @@ with gr.Blocks(
         reset_session,
         [],
         [conversation, user_input, history_state, stage_state, session_state, stage_display,
-         needs1, needs2, needs3, needs4]
+         needs_accordion, needs1, needs2, needs3, needs4]
     )
 
 if __name__ == "__main__":