split links by column

app.py

@@ -48,24 +48,52 @@ hf_llm_diagrams = [img for img in imgs if 'hf_llm_diagram' in os.path.basename(i
 # Getting the remaining images
 remaining_imgs = [img for img in imgs if 'hf_llm_diagram' not in os.path.basename(img)]
 
-def print_model_list(file_name, st):
+def print_model_list(file_name, st, split_into_two=False):
     file_path = file_name[:-4] + '.json'
     # Read the list from the JSON file
     with open(file_path, 'r') as file:
         model_id_list_loaded = json.load(file)
-        model_str = "<ul>"
-        for model_id in model_id_list_loaded:
-            model_id_trunc = model_id
-            if len(model_id) > 35:
-                model_id_trunc = '...' + model_id[-35:]
-            model_str += f'<li><a href="https://huggingface.co/{model_id}">{model_id_trunc}</a></li>'
-        model_str += "</ul>"
-        st.write(model_str, unsafe_allow_html=True)
+        midpoint = len(model_id_list_loaded) // 2 + (len(model_id_list_loaded) % 2)  # Calculate the midpoint
+
+        # Split the list into two parts
+        left_list = model_id_list_loaded[:midpoint]
+        right_list = model_id_list_loaded[midpoint:]
+
+        # Generate HTML for the left column
+        left_html = ""
+        for model_id in left_list:
+            model_id_trunc = model_id if len(model_id) <= 35 else '...' + model_id[-35:]
+            left_html += f'<li><a href="https://huggingface.co/{model_id}">{model_id_trunc}</a></li>'
+
+
+        # Generate HTML for the right column
+        right_html = ""
+        for model_id in right_list:
+            model_id_trunc = model_id if len(model_id) <= 35 else '...' + model_id[-35:]
+            right_html += f'<li><a href="https://huggingface.co/{model_id}">{model_id_trunc}</a></li>'
+
+        final_html = ""
+        if(split_into_two):
+            final_html  = "<ul>"
+            final_html  += left_html
+            final_html  += "</ul>"
+            cols = st.columns(2)
+            cols[0].write(final_html, unsafe_allow_html=True)
+            final_html  = "<ul>"
+            final_html  += right_html
+            final_html  += "</ul>"
+            cols[1].write(final_html, unsafe_allow_html=True)
+        else:
+            final_html  = "<ul>"
+            final_html  += left_html
+            final_html  += right_html
+            final_html  += "</ul>"
+            st.write(final_html, unsafe_allow_html=True)
 
 st.write("HuggingFace Open LLM leaderboard by Model Size")
 st.image(hf_llm_diagrams[0],use_column_width="auto")
 
-print_model_list(hf_llm_diagrams[0],st)
+print_model_list(hf_llm_diagrams[0],st,True)
 
 cols = st.columns(2)
 cols[0].image(hf_llm_diagrams[1],caption="Other or commercially permissive licenses only", use_column_width="auto")