Added colorscales. Added script

Author: lucasimi

pyproject.toml

@@ -52,6 +52,15 @@ dev = [
     "flake8",
     "nbformat>=4.2.0",
 ]
+app = [
+    "pandas<3.0.0",
+    "scikit-learn<1.6.0",
+    "nicegui>=2.18.0,<3.0.0",
+    "umap-learn<0.6.0",
+]
+
+[project.scripts]
+tda-mapper-app = "tdamapper.app:main"
 
 [project.urls]
 Homepage = "https://github.com/lucasimi/tda-mapper-python"

src/tdamapper/app.py

@@ -199,6 +199,19 @@ def run_mapper(df, labels, **kwargs):
     ).plot_plotly(
         colors=df_colors.to_numpy(),
         title=df_colors.columns.to_list(),
+        cmap=[
+            "Viridis",
+            "Cividis",
+            "Jet",
+            "Plasma",
+            "Inferno",
+            "Magma",
+            "Turbo",
+            "RdBu",
+            "BrBG",
+            "PiYG",
+            "PuOr",
+        ],
         height=800,
         node_size=[i * 0.125 for i in range(17)],
     )
@@ -235,7 +248,7 @@ def __init__(self, storage):
                 self._init_clustering()
 
             ui.button(
-                "Run Mapper",
+                "✨ Run Mapper",
                 on_click=self.async_run_mapper,
                 color="primary",
             ).classes("w-full")
@@ -568,10 +581,15 @@ async def async_run_mapper(self):
 
 
 @ui.page("/")
-def main():
+def main_page():
     ui.query(".nicegui-content").classes("p-0")
     storage = app.storage.client
     App(storage=storage)
 
 
-ui.run(storage_secret="secret")
+def main():
+    ui.run(storage_secret="secret", title="tda-mapper-app", favicon=ICON_URL)
+
+
+if __name__ in ["__main__", "__mp_main__", "tdamapper.app"]:
+    main()