Update TLDR.py

TLDR.py

@@ -46,11 +46,13 @@ def run_code(category, loaded_dict, event = None):
 def update_mainTLDR(buttons_to_add, paper_list):
     main_tldr = []
 
-    table_of_contents = []  # List to store the table of contents entries
-
     if buttons_to_add:
+        table_of_contents = []
         for i in range(len(buttons_to_add)):
             category_content = []
+            category_title = f"{buttons_to_add[i]}"
+            table_of_contents.append(f"- [{category_title}](#{category_title.lower().replace(' ', '-')})")
+
             for paper in paper_list[i]:
 
                 # Create the HTML content for both links with line breaks
@@ -69,7 +71,7 @@ def update_mainTLDR(buttons_to_add, paper_list):
                 category_content.append(paper_content)
 
             category_column = pn.Column(
-                pn.pane.Markdown(f"# [{buttons_to_add[i]}](#{buttons_to_add[i]}) <a id='{buttons_to_add[i]}'></a>"),
+                pn.pane.Markdown(f"# {category_title}"),
                 pn.layout.Spacer(height=20),
                 *category_content,
                 pn.layout.Spacer(height=10),
@@ -80,16 +82,13 @@ def update_mainTLDR(buttons_to_add, paper_list):
 
             main_tldr.append(category_column)
 
-            # Add entry to the table of contents
-            table_of_contents.append(f"- [{buttons_to_add[i]}](#{buttons_to_add[i]})")
+        # Create the table of contents Markdown text
+        toc_text = "\n".join(table_of_contents)
+
+        # Insert the table of contents at the beginning of the main content
+        main_tldr.insert(0, pn.pane.Markdown(f"## Table of Contents\n{toc_text}\n"))
 
     else:
         main_tldr.append(pn.pane.Markdown("# Please select some tags!"))
 
-    # Create the table of contents Markdown text
-    toc_text = "\n".join(table_of_contents)
-
-    # Prepend the table of contents to the main content
-    main_tldr.insert(0, pn.pane.Markdown(f"## Table of Contents\n{toc_text}\n"))
-
     return main_tldr