Add Population Data Tables

agent.py

@@ -62,7 +62,7 @@ def create_assistant_tools(cfg):
 
     db_tools = tools_factory.database_tools(
                 tool_name_prefix = "cfpb",
-                content_description = 'Customer complaints about five banks (Bank of America, Wells Fargo, Capital One, Chase, and CITI Bank)',
+                content_description = 'Customer complaints about five banks (Bank of America, Wells Fargo, Capital One, Chase, and CITI Bank) and geographic information (counties and zip codes)',
                 sql_database = SQLDatabase(create_engine('sqlite:///cfpb_database.db')),
             )
 

app.py

@@ -48,6 +48,12 @@ def setup_db():
         df = load_dataset("vectara/cfpb-complaints", data_files="cfpb_complaints.csv", token=hf_token)['train'].to_pandas()
         df.to_sql('cfpb_complaints', conn, if_exists='replace', index=False)
 
+        df = load_dataset("vectara/cfpb-complaints", data_files="cfpb_county_populations.csv", token=hf_token)['train'].to_pandas()
+        df.to_sql('cfpb_county_populations', conn, if_exists='replace', index=False)
+
+        df = load_dataset("vectara/cfpb-complaints", data_files="cfpb_zip_to_county.csv", token=hf_token)['train'].to_pandas()
+        df.to_sql('cfpb_zip_to_county', conn, if_exists='replace', index=False)
+
         # Commit changes and close connection
         conn.commit()
         conn.close()

create_table.sql

@@ -12,4 +12,17 @@ CREATE TABLE cfpb_complaints (
     report_method VARCHAR(12),
     complaint_status VARCHAR(31),
     timely_response INTEGER
+);
+
+CREATE TABLE county_populations (
+    state VARCHAR(2),
+    county VARCHAR(39),
+    year INTEGER,
+    population INTEGER
+);
+
+CREATE TABLE zip_codes_and_counties (
+    state VARCHAR(2),
+    county VARCHAR(28),
+    zip_code INTEGER
 );