Update app.py

app.py

@@ -397,21 +397,6 @@ def create_interface():
 
  return search_results, all_values, all_indices
 
-
-
- # # Prepare search results
- # search_results = []
- # for doc_id in topk_doc_ids:
- # metadata = df_metadata[df_metadata['arxiv_id'] == doc_id].iloc[0]
- # title = metadata['title'].replace('[', '').replace(']', '')
- # search_results.append([
- # title,
- # int(metadata['citation_count']),
- # int(metadata['year'])
- # ])
-
- # return search_results, all_values, all_indices
-
  @gr.render(inputs=[input_text, search_results_state, feature_values_state, feature_indices_state, manually_added_features_state, subject])
  def show_components(text, search_results, feature_values, feature_indices, manually_added_features, current_subject):
  if len(text) == 0:
@@ -477,27 +462,56 @@ def create_interface():
  with gr.Column(scale=1):
  update_button = gr.Button("Update Results")
  sliders = []
+ # for i, (value, index) in enumerate(zip(feature_values, feature_indices)):
+ # feature = next((f for f in subject_data[current_subject]['feature_analysis'] if f['index'] == index), None)
+ # label = f"{feature['label']} ({index})" if feature else f"Feature {index}"
+ 
+ # # Add prefix and change color for manually added features
+ # if index in manually_added_features:
+ # label = f"[Custom] {label}"
+ # slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=value, label=label, key=f"slider-{index}", elem_id=f"custom-slider-{index}")
+ # else:
+ # slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=value, label=label, key=f"slider-{index}")
+ 
+ # sliders.append(slider)
  for i, (value, index) in enumerate(zip(feature_values, feature_indices)):
  feature = next((f for f in subject_data[current_subject]['feature_analysis'] if f['index'] == index), None)
  label = f"{feature['label']} ({index})" if feature else f"Feature {index}"
  
+ # Transform the value to a 0-1 range
+ transformed_value = max(0.01, min(1, value)) # Ensure value is between 0.01 and 1
+ linear_value = (np.log10(transformed_value) + 2) / 2 # Map 0.01-1 to 0-1
+ 
  # Add prefix and change color for manually added features
  if index in manually_added_features:
  label = f"[Custom] {label}"
- slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=value, label=label, key=f"slider-{index}", elem_id=f"custom-slider-{index}")
+ slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=linear_value, label=label, key=f"slider-{index}", elem_id=f"custom-slider-{index}")
  else:
- slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=value, label=label, key=f"slider-{index}")
+ slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=linear_value, label=label, key=f"slider-{index}")
  
  sliders.append(slider)
 
+ # def on_slider_change(*values):
+ # manually_added_features = values[-1]
+ # slider_values = list(values[:-1])
+ 
+ # # Reconstruct feature_indices based on the order of sliders
+ # reconstructed_indices = [int(slider.label.split('(')[-1].split(')')[0]) for slider in sliders]
+ 
+ # new_results, new_values, new_indices = update_search_results(slider_values, reconstructed_indices, manually_added_features, current_subject)
+ # return new_results, new_values, new_indices, manually_added_features
+
  def on_slider_change(*values):
  manually_added_features = values[-1]
  slider_values = list(values[:-1])
  
+ # Transform slider values back to original scale
+ transformed_values = [10 ** ((2 * float(v)) - 2) for v in slider_values]
+ 
  # Reconstruct feature_indices based on the order of sliders
  reconstructed_indices = [int(slider.label.split('(')[-1].split(')')[0]) for slider in sliders]
  
- new_results, new_values, new_indices = update_search_results(slider_values, reconstructed_indices, manually_added_features, current_subject)
+ new_results, new_values, new_indices = update_search_results(transformed_values, reconstructed_indices, manually_added_features, current_subject)
  return new_results, new_values, new_indices, manually_added_features
 
  update_button.click(