Update app.py

app.py

@@ -373,14 +373,45 @@ def create_interface():
  for doc_id in topk_doc_ids:
  metadata = df_metadata[df_metadata['arxiv_id'] == doc_id].iloc[0]
  title = metadata['title'].replace('[', '').replace(']', '')
+ # Remove single quotes from title
+ title = title.replace("'", "")
+
+ url_id = doc_id.replace('_arXiv.txt', '')
+ if 'astro-ph' in url_id:
+ url_id = url_id.split('astro-ph')[1]
+ url = f"https://arxiv.org/abs/astro-ph/{url_id}"
+ else:
+ # Create the clickable link based on the doc_id
+ if '.' in doc_id:
+ url = f"https://arxiv.org/abs/{doc_id.replace('_arXiv.txt', '')}"
+ else:
+ url = f"https://arxiv.org/abs/hep-ph/{doc_id.replace('_arXiv.txt', '')}"
+ 
+ linked_title = f"[{title}]({url})"
+ 
  search_results.append([
- title,
+ linked_title,
  int(metadata['citation_count']),
  int(metadata['year'])
  ])
 
  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:
@@ -402,10 +433,17 @@ def create_interface():
 
  with gr.Row():
  with gr.Column(scale=2):
+ # df = gr.Dataframe(
+ # headers=["Title", "Citation Count", "Year"],
+ # value=search_results,
+ # label="Top 10 Search Results"
+ # )
  df = gr.Dataframe(
  headers=["Title", "Citation Count", "Year"],
  value=search_results,
- label="Top 10 Search Results"
+ label="Top 10 Search Results",
+ datatype=["markdown", "number", "number"], # Add this line
+ wrap=True # Add this line to ensure long titles don't get cut off
  )
 
  feature_search = gr.Textbox(label="Search Feature Labels")