testing-patch

fiboa/query.py

@@ -8,7 +8,7 @@ from ibis import _
 
 def execute_prompt(con, chain, prompt):
     response = chain.invoke({"question": prompt})
-    st.write(response)
+    st.write(response.replace("testing", "crops"))
     gdf = as_geopandas(con, response)
 
     if len(gdf) == 0:
@@ -28,7 +28,8 @@ def execute_prompt(con, chain, prompt):
         })
 
 def as_geopandas(con, response):
-    response = re.sub(";$", "", response)
+    #import code; code.interact(local=locals())
+    response = re.sub(";$", "", response).replace("testing", "crops")
     sql_query = f"CREATE OR REPLACE VIEW testing AS ({response})"
     con.raw_sql(sql_query)
     gdf = con.table("testing")
@@ -47,4 +48,4 @@ def as_geopandas(con, response):
         if dtype.startswith("datetime64"):
             gdf[col] = gdf[col].astype(str)
     
-    return gdf
+    return gdf