updates

app.py

@@ -1,9 +1,5 @@
 import streamlit as st
-from pprint import pprint
-import streamlit.components.v1 as components
 from widgets import *
-from pyecharts.charts import Bar
-from pyecharts import options as opts
 from lrt_instance import *
 
 

lrt/lrt.py

@@ -12,6 +12,7 @@ class LiteratureResearchTool:
         self.literature_search = AcademicQuery
         self.cluster_pipeline = ClusterPipeline(cluster_config)
 
+
     def __postprocess_clusters__(self, clusters: ClusterList) ->ClusterList:
         '''
         add top-5 keyphrases to each cluster

widgets/body.py

@@ -1,7 +1,10 @@
 import streamlit as st
 from api_ import ArxivQuery, IEEEQuery, PaperWithCodeQuery
 from lrt_instance import *
-
+from pyecharts.charts import Bar
+from pyecharts import options as opts
+import streamlit.components.v1 as st_render
+from .utils import generate_html_pyecharts
 
 def __preview__(platforms, num_papers, num_papers_preview, query_input,start_year,end_year):
     with st.spinner('Searching...'):
@@ -60,11 +63,23 @@ def render_body(platforms, num_papers, num_papers_preview, query_input, show_pre
 
 
         # lrt results
-        generator =  baseline_lrt(query_input,num_papers,start_year,end_year,platforms, best_k=k,
-                                  # loading_ctx_manager= st.spinner,
-                                  )
-        for plat in platforms:
+        generator =  baseline_lrt(query_input,num_papers,start_year,end_year,platforms, best_k=k)
+        for i,plat in enumerate(platforms):
+            st.markdown(f'''# {plat} Results''')
             clusters, articles = next(generator)
-            print(clusters)
+            clusters.sort()
+
+            st.markdown(f'''## Clusters Overview''')
+            st.markdown(f'''Here we show the overview of the clusters, more specifically,''')
+            st.markdown(f'''\n- the number of papers in each cluster\n- the number of keyphrases of each cluster''')
+            bar = (
+                Bar()
+                .add_xaxis([f'Cluster {i + 1}' for i in range(len(clusters))])
+                .add_yaxis("number of papers", [len(c) for c in clusters])
+                .add_yaxis("number of keyphrases", [len(c.get_keyphrases()) for c in clusters])
+            )
+            html = generate_html_pyecharts(bar, 'tmp.html')
+            st_render.html(html, height=500, width=1000)
+