cboettig commited on
Commit
2071015
·
1 Parent(s): 61f0fa4

old versions

Browse files
Files changed (3) hide show
  1. .gitignore +3 -0
  2. app.py +14 -13
  3. requirements.txt +0 -7
.gitignore ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ *.wal
2
+ *.ipynb_checkpoints
3
+ venv/
app.py CHANGED
@@ -4,7 +4,7 @@ from langchain.llms.openai import OpenAI
4
  from langchain.agents import create_sql_agent
5
  from langchain.sql_database import SQLDatabase
6
  from langchain.agents.agent_types import AgentType
7
- from langchain.callbacks import StreamlitCallbackHandler
8
  from langchain.agents.agent_toolkits import SQLDatabaseToolkit
9
  from sqlalchemy import create_engine
10
  import sqlite3
@@ -15,21 +15,22 @@ st.set_page_config(page_title="Protected Areas Database Chat", page_icon="🦜")
15
  st.title("🦜 Protected Areas Database Chat")
16
 
17
  db_uri = "duckdb:///:memory:"
 
18
  engine = create_engine(db_uri)
19
  from sqlalchemy import text
20
  con = engine.connect()
21
- con.execute(text("create or replace view agency_name as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-agency-name.parquet'"))
22
- con.execute(text("create or replace view agency_name as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-agency-name.parquet'"))
23
- con.execute(text("create or replace view agency_type as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-agency-type.parquet'"))
24
- con.execute(text("create or replace view category as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-category.parquet'"))
25
- con.execute(text("create or replace view designation_type as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-desgination-type.parquet'"))
26
- con.execute(text("create or replace view easement as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-easement.parquet'"))
27
- con.execute(text("create or replace view fee as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-fee.parquet'"))
28
- con.execute(text("create or replace view marine as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-marine.parquet'"))
29
- con.execute(text("create or replace view iucn as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-iucn.parquet'"))
30
- con.execute(text("create or replace view public_access as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-public-access.parquet'"))
31
- con.execute(text("create or replace view state_name as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-state-name.parquet'"))
32
- con.execute(text("create or replace view combined as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-combined.parquet'"))
33
 
34
  db = SQLDatabase(engine, view_support=True)
35
  db.get_usable_table_names()
 
4
  from langchain.agents import create_sql_agent
5
  from langchain.sql_database import SQLDatabase
6
  from langchain.agents.agent_types import AgentType
7
+ from langchain_community.callbacks import StreamlitCallbackHandler
8
  from langchain.agents.agent_toolkits import SQLDatabaseToolkit
9
  from sqlalchemy import create_engine
10
  import sqlite3
 
15
  st.title("🦜 Protected Areas Database Chat")
16
 
17
  db_uri = "duckdb:///:memory:"
18
+ db_uri = "duckdb:///pad.duckdb"
19
  engine = create_engine(db_uri)
20
  from sqlalchemy import text
21
  con = engine.connect()
22
+ #con.execute(text("create or replace view agency_name as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-agency-name.parquet'"))
23
+ #con.execute(text("create or replace view agency_name as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-agency-name.parquet'"))
24
+ #con.execute(text("create or replace view agency_type as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-agency-type.parquet'"))
25
+ #con.execute(text("create or replace view category as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-category.parquet'"))
26
+ #con.execute(text("create or replace view designation_type as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-desgination-type.parquet'"))
27
+ #con.execute(text("create or replace view easement as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-easement.parquet'"))
28
+ #con.execute(text("create or replace view fee as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-fee.parquet'"))
29
+ #con.execute(text("create or replace view marine as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-marine.parquet'"))
30
+ #con.execute(text("create or replace view iucn as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-iucn.parquet'"))
31
+ #con.execute(text("create or replace view public_access as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-public-access.parquet'"))
32
+ #con.execute(text("create or replace view state_name as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-state-name.parquet'"))
33
+ #con.execute(text("create or replace view combined as select * from 'https://huggingface.co/datasets/boettiger-lab/pad-us-3/resolve/main/parquet/pad-combined.parquet'"))
34
 
35
  db = SQLDatabase(engine, view_support=True)
36
  db.get_usable_table_names()
requirements.txt CHANGED
@@ -1,11 +1,4 @@
1
- pandas
2
- pydeck
3
- pmtiles
4
  streamlit
5
- leafmap
6
- ibis-framework[duckdb]
7
- altair
8
- openai
9
  langchain
10
  langchain-openai
11
  langchain-community
 
 
 
 
1
  streamlit
 
 
 
 
2
  langchain
3
  langchain-openai
4
  langchain-community