annikwag commited on
Commit
da0a56f
·
verified ·
1 Parent(s): 163febb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -9
app.py CHANGED
@@ -116,21 +116,25 @@ def get_rag_answer(query, top_results):
116
  ###########################################
117
  # CRS Options using lookup (Change 7)
118
  ###########################################
119
- @st.cache_data
120
  def get_crs_options(_client, collection_name):
 
 
121
  results = hybrid_search(_client, "", collection_name)
122
  all_results = results[0] + results[1]
123
  crs_set = set()
124
  for res in all_results:
125
  metadata = res.payload.get('metadata', {})
126
- crs_key = metadata.get("crs_key", "").strip()
127
- if crs_key:
128
- new_value = lookup_crs_value(crs_key)
129
- try:
130
- new_value = str(int(float(new_value)))
131
- except Exception:
132
- new_value = new_value.replace('.0', '')
133
- crs_combined = f"{crs_key}: {new_value}"
 
 
134
  crs_set.add(crs_combined)
135
  return sorted(crs_set)
136
 
 
116
  ###########################################
117
  # CRS Options using lookup (Change 7)
118
  ###########################################
119
+ @st.cache_data(show_spinner=False)
120
  def get_crs_options(_client, collection_name):
121
+ # Clear cache if needed (uncomment next line to force a refresh)
122
+ # st.cache_data.clear()
123
  results = hybrid_search(_client, "", collection_name)
124
  all_results = results[0] + results[1]
125
  crs_set = set()
126
  for res in all_results:
127
  metadata = res.payload.get('metadata', {})
128
+ raw_crs_key = metadata.get("crs_key", "").strip()
129
+ if raw_crs_key:
130
+ # Convert crs_key to string and remove trailing ".0"
131
+ crs_key_clean = str(raw_crs_key).replace('.0', '').strip()
132
+ # Ensure the lookup receives the clean key as a string
133
+ lookup_input = crs_key_clean
134
+ new_value_raw = lookup_crs_value(lookup_input)
135
+ # Convert the returned value to a string and remove trailing ".0"
136
+ new_value_clean = str(new_value_raw).replace('.0', '').strip()
137
+ crs_combined = f"{crs_key_clean}: {new_value_clean}"
138
  crs_set.add(crs_combined)
139
  return sorted(crs_set)
140