Update app_web.py

app_web.py

@@ -246,7 +246,7 @@ def render_chat():
                             Please let me know if there's anything else I can assist you with today.")
             return
             
-        if qcat_result.get("category") == "stata" and (qcat_result.get("sub-action")=="unresolved" or st.session_state.stata_options is None):
+        if qcat_result.get("category") == "stata" and ("unresolved" in qcat_result.get("sub-action") or st.session_state.stata_options is None):
             # Ask clarification — store original query and question for context
             st.session_state.unresolved_category = original_query_cat
 
@@ -289,7 +289,7 @@ def render_chat():
             return
 
         print(qcat_result.get("sub-action"))
-        if qcat_result.get("sub-action", "unresolved") != "unresolved" or isinstance(qcat_result.get("sub-action", "unresolved"), list):
+        if "unresolved" not in qcat_result.get("sub-action", "unresolved"):
             selected_options = [st.session_state.stata_options.get(s) for s in qcat_result.get("sub-action", "unresolved")]
             opts = {
                     "rewrite":    True if "rewrite" in selected_options else False,