Add County Populations Table to Database Tool

agent.py

@@ -71,7 +71,7 @@ def create_assistant_tools(cfg):
 
     db_tools = tools_factory.database_tools(
                 tool_name_prefix = "ev",
-                content_description = 'Electric Vehicles in the state of Washington',
+                content_description = 'Electric Vehicles in the state of Washington and other population information',
                 sql_database = SQLDatabase(create_engine('sqlite:///ev_database.db')),
             )
 

app.py

@@ -21,7 +21,7 @@ def setup_db():
 
     with st.spinner("Loading data... Please wait..."):
         def tables_populated() -> bool:
-            tables = ['ev_population', 'county_registrations', 'ev_registrations']        
+            tables = ['ev_population', 'county_registrations', 'ev_registrations', 'washington_population']        
             for table in tables:
                 cursor.execute(f"SELECT name FROM sqlite_master WHERE type='table' AND name='{table}'")
                 result = cursor.fetchone()
@@ -55,6 +55,10 @@ def setup_db():
         df = load_dataset("vectara/ev-dataset", data_files="Electric_Vehicle_Title_and_Registration_Activity.csv", token=hf_token)['train'].to_pandas()
         df.to_sql('ev_registrations', conn, if_exists='replace', index=False)
 
+        # Load data into washington_population table
+        df = load_dataset("vectara/ev-dataset", data_files="washington_population.csv", token=hf_token)['train'].to_pandas()
+        df.to_sql('washington_population', conn, if_exists='replace', index=False)
+
         # Commit changes and close connection
         conn.commit()
         conn.close()