Improvements

fiboa/app.py

@@ -48,15 +48,17 @@ Pay close attention to use only the column names that you can see in the schema
 not query for columns that do not exist. Also, pay attention to which column is in which table.
 
 Tables include {table_info}. The data comes comes always from the table called "crops".
-Do never use the "testing" table. Pay close attention to this table schema.
+Never use the "testing" table. Pay close attention to this table schema.
 
 The column "area" is in the unit hectares, you may need to convert it to other units, e.g. square meters.
 The column "perimeter" is in the unit meters, you may need to convert it to other units, e.g. kilometers.
-The column "collection" contains the country codes for the Baltic states: "ec_lt" for Latvia, "ec_lv" for Lithuania, "ec_ee" for Estonia.
+The column "collection" contains the country codes for the Baltic states:
+"ec_lt" for Latvia, "ec_lv" for Lithuania, "ec_es" for Estonia.
 
 Question: {input}
 '''
 )
+# todo: if data get's updated, change "ec_es" to "ec_ee"
 
 llm = ChatOpenAI(model="gpt-4o-mini", temperature=0, api_key=st.secrets["OPENAI_API_KEY"])
 
@@ -69,6 +71,7 @@ Ask me about fiboa data (here: all baltic states)!
 Request "a map" to get map output, or table for tabular output, e.g.
 
 - Show a map with the 10 largest fields
+- Show a map with the largest field in Estonia
 - Show a table of the total area by crop type
 - Compute the total area of all fields in km² and compute the percentage the total area of the baltic states (175015 km²)
 

fiboa/query.py

@@ -11,6 +11,10 @@ def execute_prompt(con, chain, prompt):
     st.write(response)
     gdf = as_geopandas(con, response)
 
+    if len(gdf) == 0:
+        st.write("No results found.")
+        return
+
     if 'geometry' in gdf.columns:
         m = leafmap.Map()
         m.add_gdf(gdf)
@@ -32,7 +36,7 @@ def as_geopandas(con, response):
     if 'geometry' in gdf.columns:
         gdf = (gdf
                .cast({"geometry": "geometry"})
-               .mutate(geometry = _.geometry.convert("EPSG:4326", "EPSG:4326"))
+               # .mutate(geometry = _.geometry.convert("EPSG:4326", "EPSG:4326"))
                .to_pandas()
               ).set_crs(epsg=4326, inplace=True) 
     else: