add pdf link to each paper

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import pandas as pd
 from sentence_transformers import SentenceTransformer
 import streamlit as st
+from streamlit.column_config import LinkColumn
 
 import os
 
@@ -41,7 +42,11 @@ def get_retrieval_results(index, input_text, top_k, model, title_df):
         retrieved_titles.append(title_df.loc[id, "title"])
         retrieved_pids.append(title_df.loc[id, "pid"])
 
-    df = pd.DataFrame({"pids": retrieved_pids, "paper": retrieved_titles})
+    df = pd.DataFrame({
+        "pid": retrieved_pids, 
+        "paper": retrieved_titles,
+        "pdf": [f'https://www.anlp.jp/proceedings/annual_meeting/2025/pdf_dir/{pid}.pdf' for pid in retrieved_pids]
+    })
     
     return df
     
@@ -57,8 +62,14 @@ if __name__ == "__main__":
     st.markdown("## NLP2025 論文検索")
     input_text = st.text_input('query', '', placeholder='')
     top_k = st.number_input('top_k', min_value=1, value=10, step=1)
+
+    column_config = {
+        "pdf": LinkColumn(
+            display_text="🔗"
+        )
+    }
     
     if st.button('検索'):
         stripped_input_text = input_text.strip()
         df = get_retrieval_results(index, stripped_input_text, top_k, model, title_df)
-        st.table(df)
+        st.dataframe(df, column_config=column_config, width=720)