Update TLDR.py

TLDR.py

@@ -60,50 +60,42 @@ def run_code(buttons_to_add, arxiv_tags, event = None):
 
 
 def update_mainTLDR(buttons_to_add, paper_list):
-
-    
-    # Create a list to store the contents for the selected categories
-    content_list = []
+    # Create a Column to store the contents for the selected categories
+    main_tldr = pn.Column()
 
     # Check if any categories are selected
     if buttons_to_add:
         for i in range(len(buttons_to_add)):
-            # print("Used categories:", buttons_to_add, flush=True)
-
             # Generate a Markdown object for each selected category
-            first_flag = 1
+            category_content = []
             for paper in paper_list[i]:
-                # print("Paper list: ", paper.title, flush = True)
-                # paper_title = "## " + paper['title'][0]
-                # paper_tldr = "### " + paper['abstract'][0]
-                # paper_date = "### " + paper['created'][0]
-
-                paper_title = "## " + paper.title
-                paper_tldr =  paper.summary
+                paper_title = "## " + pn.pane.LaTeX(paper.title)
+                paper_tldr = paper.summary
                 paper_date = "### " + (paper.published).strftime("%d/%m/%Y %H:%M:%S")
-                
-                if first_flag:
-                    content = pn.Row(
-                        pn.pane.Markdown(f"# {buttons_to_add[i]}"),
-                        pn.Column(paper_title,
-                        paper_tldr,
-                        paper_date),
-                        sizing_mode='stretch_width'
-                    )
-                    first_flag = 0
-                else:
-                    content = pn.Row(
-                        pn.Column(
+
+                # Create a Row to hold each paper's content
+                paper_content = pn.Row(
+                    pn.Column(
                         paper_title,
                         paper_tldr,
-                        paper_date),
-                        sizing_mode='stretch_width',
-                    )
-                    
-                content_list.append(content)
-            #content_list.append(content)
+                        pn.pane.Markdown(paper_date)
+                    ),
+                    sizing_mode='stretch_width',
+                )
+                category_content.append(paper_content)
+
+            # Create a Column for each category's content
+            category_column = pn.Column(
+                pn.pane.Markdown(f"# {buttons_to_add[i]}"),
+                *category_content,  # Unpack the list of paper contents
+                sizing_mode='stretch_width'
+            )
+
+            # Add the category column to the main TLDR column
+            main_tldr.append(category_column)
+
     else:
         # If no categories are selected, display the default message
-        content_list.append(pn.pane.Markdown("# Please select some tags!"))
+        main_tldr.append(pn.pane.Markdown("# Please select some tags!"))
 
-    return content_list
+    return main_tldr