Update app.py

app.py

@@ -9,8 +9,6 @@ st.header("List Ollama model downloads")
 
 HOST = os.getenv("HOST", "https://1tsnakers-ollamasearchapi.hf.space")
 
-robot_mode = st.checkbox("Robot mode (outputs unparsed data)")
-
 def is_api_up():
     
     r = requests.get(f"{HOST}/ping")
@@ -46,48 +44,25 @@ if st.button("Make list"):
         for model in library:
             date.append(datetime.fromisoformat(model["last_updated_iso"]))
 
-        date_iso = []
-        for model in library:
-            date_iso.append(model["last_updated_iso"])
-
-        if robot_mode:
-            df = pd.DataFrame({
-                "model": models,
-                "pull_count": pull_count,
-                "date_iso": date_iso
-            })
-        else:
-            df = pd.DataFrame({
-                "model": models,
-                "pull_count_str": pull_count_str,
-                "date": date
-            })
+        df = pd.DataFrame({
+            "model": models,
+            "pull_count": pull_count,
+            "pull_count_str": pull_count_str,
+            "date": date
+        })
         end_time = time.time()
         elapsed_time = end_time - start_time
         
         st.write(f"{len(models)} models counted in {elapsed_time:.4f} seconds")
 
-        if robot_mode:
-            st.dataframe(
-                df,
-                hide_index=True,
-                column_config={
-                    "model": "AI Model",
-                    "pull_count": st.column_config.NumberColumn(
-                        "Pulls"
-                    ),
-                    "date_iso": "Last Updated"
-                }
-            )
-        else:
-            st.dataframe(
-                df,
-                hide_index=True,
-                column_config={
-                    "name": "AI Model",
-                    "pull_count_str": "Pulls",
-                    "date": st.column_config.DateColumn(
-                        "Last Updated"
-                    )
-                }
-            )
+        st.dataframe(
+            dfstyle.format({"pull_count": lambda x, i=None: pull_count_str[i]}),
+            hide_index=True,
+            column_config={
+                "name": "AI Model",
+                "pull_count_str": "Pulls",
+                "date": st.column_config.DateColumn(
+                    "Last Updated"
+                )
+            }
+        )