Update app.py

app.py

@@ -50,7 +50,7 @@ accepted_venues = ['ICLR 2023 poster', 'ICLR 2023 notable top 5%', 'ICLR 2023 no
 df = df_raw[df_raw["content.venue"].isin(accepted_venues)]
 st.write("Number of submissions accepted at ICLR 2023:", len(df))
 
-df_filtered = df[['id', 'content.title', 'content.keywords', 'content.abstract']]
+df_filtered = df[['content.title', 'content.keywords', 'content.abstract', 'content.venue']]
 df = df_filtered
 if "CO_API_KEY" not in os.environ:
     raise KeyError("CO_API_KEY not found in st.secrets or os.environ. Please set it in "
@@ -58,7 +58,24 @@ if "CO_API_KEY" not in os.environ:
 
 co = cohere.Client(os.environ["CO_API_KEY"])
 
+def to_html(df: pd.DataFrame, table_header: str) -> str:
+        table_data = ''.join(df.html_table_content)
+        html = f'''
+        <table>
+            {table_header}
+            {table_data}
+        </table>'''
+        return html
+
+
 def get_visualizations():
+    table_header = '''
+            <tr>
+                <td width="25%">Title</td>
+                <td width="15%">Keywords</td>
+                <td width="10%">Venue</td>
+                <td width="50%">Abstract</td>
+            </tr>'''
     list_of_titles = list(df["content.title"].values)
     embeds = co.embed(texts=list_of_titles,                  				
   					model="small").embeddings
@@ -87,5 +104,8 @@ def get_visualizations():
     fig2 = topic_model.visualize_barchart(custom_labels=True)
     st.plotly_chart(fig1)
     st.plotly_chart(fig2)
+    st.write("Full List of Papers")
+    to_html(df,table_header)
+
 
 st.button("Run Visualization", on_click=get_visualizations)