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")