Integrate gui in main codebase

Author: MilesCranmer

pysr/_cli/main.py

@@ -5,6 +5,7 @@
 
 import click
 
+from ..gui import main as gui_main
 from ..test import (
     get_runtests_cli,
     runtests,
@@ -48,6 +49,11 @@ def _install(julia_project, quiet, precompile):
     )
 
 
+@pysr.command("gui", help="Start a Gradio-based GUI.")
+def _gui():
+    gui_main()
+
+
 TEST_OPTIONS = {"main", "jax", "torch", "cli", "dev", "startup"}
 
 

gui/app.py renamed to pysr/gui/app.py

@@ -1,11 +1,17 @@
-import gradio as gr
-
 from .data import test_equations
 from .plots import replot, replot_pareto
 from .processing import processing
 
 
+def get_gr():
+    import gradio as gr
+
+    return gr
+
+
 def _data_layout():
+    gr = get_gr()
+
     with gr.Tab("Example Data"):
         # Plot of the example data:
         with gr.Row():
@@ -43,6 +49,8 @@ def _data_layout():
 
 
 def _settings_layout():
+    gr = get_gr()
+
     with gr.Tab("Basic Settings"):
         binary_operators = gr.CheckboxGroup(
             choices=["+", "-", "*", "/", "^", "max", "min", "mod", "cond"],
@@ -171,6 +179,8 @@ def _settings_layout():
 
 
 def main():
+    gr = get_gr()
+
     blocks = {}
     with gr.Blocks() as demo:
         with gr.Row():
@@ -245,7 +255,3 @@ def main():
         demo.load(replot, eqn_components, blocks["example_plot"])
 
     demo.launch(debug=True)
-
-
-if __name__ == "__main__":
-    main()

gui/data.py renamed to pysr/gui/data.py

@@ -1,20 +1,30 @@
 import numpy as np
 import pandas as pd
-from matplotlib import pyplot as plt
-
-plt.ioff()
-plt.rcParams["font.family"] = [
-    "IBM Plex Mono",
-    # Fallback fonts:
-    "DejaVu Sans Mono",
-    "Courier New",
-    "monospace",
-]
 
 from .data import generate_data
 
+FIRST_LOAD = True
+
+
+def get_plt():
+    from matplotlib import pyplot as plt
+
+    if FIRST_LOAD:
+        plt.ioff()
+        plt.rcParams["font.family"] = [
+            "IBM Plex Mono",
+            # Fallback fonts:
+            "DejaVu Sans Mono",
+            "Courier New",
+            "monospace",
+        ]
+
+    FIRST_LOAD = False
+    return plt
+
 
 def replot_pareto(df: pd.DataFrame, maxsize: int):
+    plt = get_plt()
     fig, ax = plt.subplots(figsize=(6, 6), dpi=100)
 
     if len(df) == 0 or "Equation" not in df.columns:
@@ -60,6 +70,7 @@ def replot(test_equation, num_points, noise_level, data_seed):
     X, y = generate_data(test_equation, num_points, noise_level, data_seed)
     x = X["x"]
 
+    plt = get_plt()
     plt.rcParams["font.family"] = "IBM Plex Mono"
     fig, ax = plt.subplots(figsize=(6, 6), dpi=100)