Improve formatting of plot

gui/Dockerfile

@@ -11,10 +11,17 @@ RUN apt-get update && \
         libgl1-mesa-glx \
         libglib2.0-0 \
         libpython3-dev \
+        libfreetype6-dev \
+        pkg-config \
+        libfontconfig1 \
+        fontconfig \
         && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+COPY fonts/*.ttf /usr/local/share/fonts/
+RUN fc-cache -f -v
+
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt

gui/app.py

@@ -268,6 +268,8 @@ def main():
                 blocks["df"] = gr.Dataframe(
                     headers=["complexity", "loss", "equation"],
                     datatype=["number", "number", "str"],
+                    wrap=True,
+                    column_widths=[100, 100, 300],
                 )
                 blocks["run"] = gr.Button()
 
@@ -339,5 +341,39 @@ def replot_pareto(df, maxsize):
 
     return fig
 
+def replot_pareto(df, maxsize):
+    plt.rcParams['font.family'] = 'IBM Plex Mono'
+    fig, ax = plt.subplots(figsize=(6, 6), dpi=100)
+
+    if len(df) == 0 or 'Equation' not in df.columns:
+        return fig
+
+    # Plotting the data
+    ax.loglog(df['Complexity'], df['Loss'], marker='o', linestyle='-', color='#333f48', linewidth=1.5, markersize=6)
+
+    # Set the axis limits
+    ax.set_xlim(0.5, maxsize + 1)
+    ytop = 2 ** (np.ceil(np.log2(df['Loss'].max())))
+    ybottom = 2 ** (np.floor(np.log2(df['Loss'].min() + 1e-20)))
+    ax.set_ylim(ybottom, ytop)
+
+    ax.grid(True, which="both", ls="--", linewidth=0.5, color='gray', alpha=0.5)
+    ax.spines['top'].set_visible(False)
+    ax.spines['right'].set_visible(False)
+
+    # Range-frame the plot
+    for direction in ['bottom', 'left']:
+        ax.spines[direction].set_position(('outward', 10))
+
+    # Delete far ticks
+    ax.tick_params(axis='both', which='major', labelsize=10, direction='out', length=5)
+    ax.tick_params(axis='both', which='minor', labelsize=8, direction='out', length=3)
+
+    ax.set_xlabel('Complexity')
+    ax.set_ylabel('Loss')
+    fig.tight_layout(pad=2)
+
+    return fig
+
 if __name__ == "__main__":
     main()