Update app.py

app.py

@@ -63,15 +63,23 @@ def compute_title(metadata):
 # Load CRS lookup CSV and define a lookup function.
 crs_lookup = pd.read_csv("docStore/crs5_codes.csv")  # Assumes columns: "code" and "new_crs_value"
 def lookup_crs_value(crs_key):
-    row = crs_lookup[crs_lookup["code"] == crs_key]
+    # Ensure the input is a string and clean it.
+    key_clean = re.sub(r'\.0$', '', str(crs_key).strip())
+    # Compare against the CSV codes as strings.
+    row = crs_lookup[crs_lookup["code"].astype(str) == key_clean]
     if not row.empty:
-        # Convert to integer (drop decimals) and then to string.
-        try:
-            return str(int(float(row.iloc[0]["new_crs_value"])))
-        except:
-            return str(row.iloc[0]["new_crs_value"])
+        # If a column named "new_crs_value" exists, use it.
+        if "new_crs_value" in row.columns:
+            try:
+                return re.sub(r'\.0$', '', str(int(float(row.iloc[0]["new_crs_value"]))))
+            except Exception:
+                return re.sub(r'\.0$', '', str(row.iloc[0]["new_crs_value"]))
+        else:
+            # Otherwise, use the "name" column as the lookup value.
+            return re.sub(r'\.0$', '', str(row.iloc[0]["name"]).strip())
     return ""
 
+
 ###########################################
 # RAG Answer function (Change 1 & 2 & 3)
 ###########################################