Update app.py

app.py

@@ -1,113 +1,130 @@
-import streamlit as st
-import pandas as pd
-from agent import agent, AgentState 
-
-# =========================================
-# STREAMLIT SETTINGS
-# =========================================
-st.set_page_config(page_title="Weather AI Assistant", page_icon="🌤️", layout="wide")
-
-# Custom CSS
-st.markdown("""
-<style>
-body { background-color: #f0f4ff; }
-.chat-message {
-    padding: 12px; border-radius: 12px; margin-bottom: 12px;
-    max-width: 80%; line-height: 1.5;
-}
-.user-msg {
-    background: #e3edff; color: #0f1e46;
-    align-self: flex-end; text-align: right; margin-left: auto;
-}
-.assistant-msg {
-    background: #d5e8ff; color: #0b1b33;
-    align-self: flex-start; margin-right: auto;
-}
-.chat-container {
-    display: flex; flex-direction: column; gap: 10px;
-}
-</style>
-""", unsafe_allow_html=True)
-
-# =========================================
-# SESSION STATE
-# =========================================
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-if "last_details" not in st.session_state:
-    st.session_state.last_details = None
-
-# =========================================
-# MAIN TITLE
-# =========================================
-st.title("🌤️ Weather Data Chat Assistant")
-st.write("Ask questions about weather data — I will generate SQL, run it, and answer.")
-
-# =========================================
-# CHAT MESSAGES RENDER
-# =========================================
-# Спочатку малюємо історію чату
-st.markdown("<div class='chat-container'>", unsafe_allow_html=True)
-for role, msg in st.session_state.messages:
-    if role == "user":
-        st.markdown(f"<div class='chat-message user-msg'><b>You:</b> {msg}</div>", unsafe_allow_html=True)
-    else:
-        st.markdown(f"<div class='chat-message assistant-msg'><b>Assistant:</b> {msg}</div>", unsafe_allow_html=True)
-st.markdown("</div>", unsafe_allow_html=True)
-
-# =========================================
-# USER INPUT & LOGIC
-# =========================================
-user_input = st.chat_input("Type your question here...")
-
-if user_input:
-    st.session_state.messages.append(("user", user_input))
-    st.rerun() 
-
-if st.session_state.messages and st.session_state.messages[-1][0] == "user":
-    last_user_msg = st.session_state.messages[-1][1]
-    
-    with st.spinner("Thinking and querying database..."):
-        try:
-            raw_state = agent.invoke({"question": last_user_msg})
-            
-            answer = raw_state.get("answer", "No answer generated.")
-            sql_query = raw_state.get("sql_query")
-            rows = raw_state.get("rows")
-            reasoning = raw_state.get("reasoning")
-
-            st.session_state.messages.append(("assistant", answer))
-            
-            st.session_state.last_details = {
-                "sql": sql_query,
-                "rows": rows,
-                "reasoning": reasoning
-            }
-            
-            st.rerun()
-
-        except Exception as e:
-            st.session_state.messages.append(("assistant", f"❌ Error: {e}"))
-            st.rerun()
-
-# =========================================
-# DEBUG / DETAILS SECTION
-# =========================================
-if st.session_state.last_details:
-    with st.expander("🔍 See Technical Details (SQL & Data)", expanded=False):
-        details = st.session_state.last_details
-        
-        if details["reasoning"]:
-            st.write("**Reasoning:**")
-            st.info(details["reasoning"])
-            
-        if details["sql"]:
-            st.write("**Generated SQL:**")
-            st.code(details["sql"], language="sql")
-        
-        if details["rows"]:
-            st.write(f"**Data Found ({len(details['rows'])} rows):**")
-            df = pd.DataFrame(details["rows"])
-            st.dataframe(df)
-        else:
+import streamlit as st
+import pandas as pd
+from agent import agent, AgentState 
+
+# =========================================
+# STREAMLIT SETTINGS
+# =========================================
+st.set_page_config(page_title="Weather AI Assistant", page_icon="🌤️", layout="wide")
+
+# Custom CSS
+st.markdown("""
+<style>
+body { background-color: #f0f4ff; }
+.chat-message {
+    padding: 12px; border-radius: 12px; margin-bottom: 12px;
+    max-width: 80%; line-height: 1.5;
+}
+.user-msg {
+    background: #e3edff; color: #0f1e46;
+    align-self: flex-end; text-align: right; margin-left: auto;
+}
+.assistant-msg {
+    background: #d5e8ff; color: #0b1b33;
+    align-self: flex-start; margin-right: auto;
+}
+.chat-container {
+    display: flex; flex-direction: column; gap: 10px;
+}
+</style>
+""", unsafe_allow_html=True)
+
+# =========================================
+# SESSION STATE
+# =========================================
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+if "last_details" not in st.session_state:
+    st.session_state.last_details = None
+
+# =========================================
+# MAIN TITLE
+# =========================================
+st.title("🌤️ Weather Data Chat Assistant")
+st.write("Ask questions about weather data — I will generate SQL, run it, and answer.")
+
+# =========================================
+# RULES / CONSTRAINTS (EN)
+# =========================================
+ALLOWED_COUNTRIES = ["Austria", "Belgium", "Bulgaria", "Switzerland", "Czechia"]
+START_DATE = "1980-01-01"
+END_DATE = "2019-12-31"
+
+st.info(
+    f"""
+**Data constraints**
+- **Allowed countries only:** {", ".join(ALLOWED_COUNTRIES)}
+- **Allowed time range (UTC):** from **{START_DATE}** to **{END_DATE}**
+
+If your question mentions another country or a date outside this range, I may not be able to answer from the database.
+""".strip()
+)
+
+# =========================================
+# CHAT MESSAGES RENDER
+# =========================================
+# Спочатку малюємо історію чату
+st.markdown("<div class='chat-container'>", unsafe_allow_html=True)
+for role, msg in st.session_state.messages:
+    if role == "user":
+        st.markdown(f"<div class='chat-message user-msg'><b>You:</b> {msg}</div>", unsafe_allow_html=True)
+    else:
+        st.markdown(f"<div class='chat-message assistant-msg'><b>Assistant:</b> {msg}</div>", unsafe_allow_html=True)
+st.markdown("</div>", unsafe_allow_html=True)
+
+# =========================================
+# USER INPUT & LOGIC
+# =========================================
+user_input = st.chat_input("Type your question here...")
+
+if user_input:
+    st.session_state.messages.append(("user", user_input))
+    st.rerun() 
+
+if st.session_state.messages and st.session_state.messages[-1][0] == "user":
+    last_user_msg = st.session_state.messages[-1][1]
+    
+    with st.spinner("Typing..."):
+        try:
+            raw_state = agent.invoke({"question": last_user_msg})
+            
+            answer = raw_state.get("answer", "No answer generated.")
+            sql_query = raw_state.get("sql_query")
+            rows = raw_state.get("rows")
+            reasoning = raw_state.get("reasoning")
+
+            st.session_state.messages.append(("assistant", answer))
+            
+            st.session_state.last_details = {
+                "sql": sql_query,
+                "rows": rows,
+                "reasoning": reasoning
+            }
+            
+            st.rerun()
+
+        except Exception as e:
+            st.session_state.messages.append(("assistant", f"❌ Error: {e}"))
+            st.rerun()
+
+# =========================================
+# DEBUG / DETAILS SECTION
+# =========================================
+if st.session_state.last_details:
+    with st.expander("🔍 See Technical Details (SQL & Data)", expanded=False):
+        details = st.session_state.last_details
+        
+        if details["reasoning"]:
+            st.write("**Reasoning:**")
+            st.info(details["reasoning"])
+            
+        if details["sql"]:
+            st.write("**Generated SQL:**")
+            st.code(details["sql"], language="sql")
+        
+        if details["rows"]:
+            st.write(f"**Data Found ({len(details['rows'])} rows):**")
+            df = pd.DataFrame(details["rows"])
+            st.dataframe(df)
+        else:
             st.warning("No data returned from SQL.")