tweaks

app.py

@@ -188,7 +188,7 @@ with st.sidebar:
     "## Protected Areas"
 
     if st.toggle("PAD US-3", True):
-        alpha = st.slider("transparency", 0.0, 1.1, 0.5)
+        alpha = st.slider("transparency", 0.0, 1.0, 0.5)
 
         with st.expander("custom style"):
                 custom = st.text_area(
@@ -328,6 +328,7 @@ def summary_table(column = column, colors = colors):
     return df.to_pandas()
 
 df = summary_table(column, colors)
+total_percent = df.percent_protected.sum()
 
 
 base = alt.Chart(df).encode(
@@ -368,7 +369,7 @@ rsr_chart = alt.Chart(df).mark_bar().encode(
 col1, col2, col3 = st.columns(3)
 
 with col1:
-    f"#### Percent of Continental US Area"
+    f"#### {total_percent} Percent of Continental US Area Protected"
     st.altair_chart(area_chart, use_container_width=True)
     
 # -

chatmap.py

@@ -1,7 +1,16 @@
 import streamlit as st
 from openai import OpenAI
+import duckdb
 
-st.title("ChatGPT-like clone")
+import duckdb
+duckdb.install_extension("spatial")
+duckdb.load_extension("spatial")
+duckdb.install_extension("httpfs")
+duckdb.load_extension("httpfs")
+
+duckdb.sql("create or replace view pad as select * from read_parquet('https://data.source.coop/cboettig/pad-us-3/pad-mobi.parquet')")
+
+st.title("ChatGPT SQL Assistant")
 
 # Set OpenAI API key from Streamlit secrets
 client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
@@ -9,23 +18,64 @@ client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
 # Set a default model
 if "openai_model" not in st.session_state:
     st.session_state["openai_model"] = "gpt-3.5-turbo"
+# "gpt-4"
 
 # Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
+
+setup = '''
+You are a database administrator, and expert in SQL. You will be helping me write complex SQL queries. I will explain you my needs, you will generate SQL queries against my database.
+
+My application does: Conservation prioritization of protected areas to help meet US 30x30 conservation goals. 
+
+Please reply only with the SQL code that I will need to execute. Do not include an explanation of the code.
+
+The database is a POSTGIS Postgres database, please take it into consideration when generating PLSQL/SQL. Please avoid ST_Within queries if possible, because they are so slow.
+
+I will provide you with a description of the structure of my tables. You must remember them and use them for generating SQL queries. Once you read them all, just answer OK, nothing else.
+
+Here are the tables :
+
+Table "pad"
+┌─────────────┬─────────────┬─────────┬─────────┬─────────┬─────────┐
+│ column_name │ column_type │  null   │   key   │ default │  extra  │
+│   varchar   │   varchar   │ varchar │ varchar │ varchar │ varchar │
+├─────────────┼─────────────┼─────────┼─────────┼─────────┼─────────┤
+│ FID         │ INTEGER     │ YES     │ NULL    │ NULL    │ NULL    │
+│ time        │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ rsr         │ DOUBLE      │ YES     │ NULL    │ NULL    │ NULL    │
+│ richness    │ DOUBLE      │ YES     │ NULL    │ NULL    │ NULL    │
+│ bucket      │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ FeatClass   │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ Mang_Name   │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ Mang_Type   │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ Des_Tp      │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ Pub_Access  │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ GAP_Sts     │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ IUCN_Cat    │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ Unit_Nm     │ VARCHAR     │ YES     │ NULL    │ NULL    │ NULL    │
+│ area        │ DOUBLE      │ YES     │ NULL    │ NULL    │ NULL    │
+│ geometry    │ BLOB        │ YES     │ NULL    │ NULL    │ NULL    │
+├─────────────┴─────────────┴─────────┴─────────┴─────────┴─────────┤
+└───────────────────────────────────────────────────────────────────┘
+'''
+
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
 # Accept user input
-if prompt := st.chat_input("What is up?"):
+if prompt := st.chat_input("What is the total area in each GAP_Sts?"):
     # Add user message to chat history
+    st.session_state.messages.append({"role": "system", "content": setup})
+
     st.session_state.messages.append({"role": "user", "content": prompt})
     # Display user message in chat message container
-    with st.chat_message("user"):
-        st.markdown(prompt)
+    # with st.chat_message("user"):
+        #st.markdown(prompt)
 
   # Display assistant response in chat message container
     with st.chat_message("assistant"):
@@ -38,5 +88,9 @@ if prompt := st.chat_input("What is up?"):
             stream=True,
         )
         response = st.write_stream(stream)
+        st.divider()
+        df = duckdb.sql(response).df()
+        st.table(df)
     st.session_state.messages.append({"role": "assistant", "content": response})
-    
+
+

requirements.txt

@@ -5,4 +5,5 @@ streamlit
 leafmap
 ibis-framework[duckdb]
 altair
+openai