annikwag commited on
Commit
6a4a0e2
·
verified ·
1 Parent(s): da0a56f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -118,26 +118,26 @@ def get_rag_answer(query, top_results):
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
 
 
141
  @st.cache_data
142
  def load_project_data():
143
  # Load your full project DataFrame using your processing function.
 
118
  ###########################################
119
  @st.cache_data(show_spinner=False)
120
  def get_crs_options(_client, collection_name):
121
+ # Optionally clear the cache if needed: st.cache_data.clear()
 
122
  results = hybrid_search(_client, "", collection_name)
123
  all_results = results[0] + results[1]
124
  crs_set = set()
125
  for res in all_results:
126
  metadata = res.payload.get('metadata', {})
127
+ raw_crs_key = metadata.get("crs_key", "")
128
+ # Convert to string and remove trailing ".0"
129
+ crs_key_clean = re.sub(r'\.0$', '', str(raw_crs_key).strip())
130
+ if crs_key_clean:
131
+ # Ensure lookup input is clean
132
  lookup_input = crs_key_clean
133
  new_value_raw = lookup_crs_value(lookup_input)
134
+ # Convert lookup return value to string and remove trailing ".0"
135
+ new_value_clean = re.sub(r'\.0$', '', str(new_value_raw).strip())
136
  crs_combined = f"{crs_key_clean}: {new_value_clean}"
137
  crs_set.add(crs_combined)
138
  return sorted(crs_set)
139
 
140
+
141
  @st.cache_data
142
  def load_project_data():
143
  # Load your full project DataFrame using your processing function.