Update app_web.py

app_web.py

@@ -46,6 +46,8 @@ def init_session():
         st.session_state.pending_clarification_question = None
     if "unresolved_category_question" not in st.session_state:
         st.session_state.unresolved_category_question = None
+    if "stata_options" not in st.session_state:
+        st.session_state.stata_options = None
 
 
 # ── Sidebar ──────────────────────────────────────────────────────────────────
@@ -271,7 +273,8 @@ def render_chat():
             if qcat_result.get("action") == "error":
                 options = {"explain the error":"explain", 
                            "suggest fix(es) for the error":"suggestfix"}        
-                        
+
+            st.session_state.stata_options = options
             st.session_state.unresolved_category_question = f"Please let me know what you need help with: {' or '.join(options.keys())}. List all options that apply."
             st.session_state.resolution_rounds += 1
             resolution_msg = f"**Before I search, could you clarify?** Please let me know what you need help with: {' or '.join(options.keys())}. List all options that apply."
@@ -285,7 +288,7 @@ def render_chat():
         print("selection: ", selection)
         if selection != "unresolved" or isinstance(selection, list):
             qu_enabled = False
-            selected_options = [options.get(s) for s in selection]
+            selected_options = [st.session_state.stata_options.get(s) for s in selection]
             opts = {
                     "rewrite":    True if "rewrite" in selected_options else False,
                     "explain":    True if "explain" in selected_options else False,