Update app_web.py

app_web.py

@@ -6,7 +6,7 @@ import streamlit as st
 from src.config_loader import load_config, get_api_key
 from src.ingest import get_chroma_collection, ingest_documents
 from src.kb_meta import load_kb_meta_brief
-from src.query_engine import understand_query, categorize_query
+from src.query_engine import understand_query, categorize_query, init_query
 from src.retriever import clear_collection_cache, retrieve
 from src.verifier import verify_and_respond
 from src.llm import list_models
@@ -219,12 +219,23 @@ def render_chat():
             for m in prior_messages[-max_history:]
         ]
         try:
+            qinit_result = init_query(combined_cat, cfg, history)
+            if qinit_result.get("country", None) is None:
+                
+                with st.chat_message("assistant"):
+                    st.markdown("Please specify the country for which you'd like to run a PERI analysis")
+                return
+            if qinit_result.get("value_chain", None) is None and qinit_result.get("investment", None) is None:
+                
+                with st.chat_message("assistant"):
+                    st.markdown(f"Please specify the value chain or investment area in {qinit_result.get("country", None)} for which you'd like to run a PERI analysis")
+                return
             qcat_result = categorize_query(combined_cat, cfg, history)
         except Exception:
             qcat_result = {"category": "general", "action": "continue"}
 
         print("unresolved: ", unresolved)
-        print("result: ", qcat_result)
+        print("result: ", qinit_result)
         print("combined: ", combined_cat)
         # if qcat_result.get("category") == "contact_and_info" and qcat_result.get("action") == "contact":
         #     with st.chat_message("assistant"):