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(