add special mode

1 files changed, 27 insertions, 5 deletions

diff --git a/api.py b/api.py
index c9b8c5d..75d9c1f 100644
--- a/api.py
+++ b/api.py
@@ -3,6 +3,8 @@ from fastapi.responses import StreamingResponse
 from io import BytesIO
 import matplotlib.pyplot as plt
 import matplotlib.font_manager as font_manager
+from matplotlib.font_manager import FontProperties
+import matplotlib as mpl
 import numpy as np
 import pandas as pd
 
@@ -32,6 +34,7 @@ class PlotDataRequest(BaseModel):
     y_axis_scale: str = Field("linear", title="Y Axis Scale", description="Scale of the Y axis")
 
     trendline_equation: str = Field(None, title="Trendline Equation", description="Manually specify the equation for the trendline")
+    special_mode: bool = Field(False, title="Special Mode", description="Special mode")
 
     constant_line_vals: list[float] = Field([], title="Constant Line Values", description="List of values for the constant line")
     constant_line_name: list[str] = Field([], title="Constant Line Names", description="List of names for the constant line")
@@ -42,7 +45,7 @@ def plot_data(x_data, y_data, std_dev_data, color_picker, labels, df,
               title = "Plot", x_label = "X Axis", y_label = "Y Axis",
               plot_background_color="#ffffff", constant_line=[],
               enable_trendline=True, enable_grid=False,
-              trendline_color="#000000", x_axis_scale="linear", y_axis_scale="linear", trendline_equation=None):
+              trendline_color="#000000", x_axis_scale="linear", y_axis_scale="linear", trendline_equation=None, special_mode=False):
     fig, ax = plt.subplots(dpi=300)
 
     plots = []
@@ -52,10 +55,16 @@ def plot_data(x_data, y_data, std_dev_data, color_picker, labels, df,
         y = df[y_data[idx]].astype(float)
         color = color_picker[idx]
         data_series_title = labels[idx]
-        if (std_dev_data[idx] != None):
-            plot = ax.errorbar(x, y, yerr=df[std_dev_data[idx]].astype(float), fmt='o', ecolor='black', capsize=5, color=color, label=data_series_title)
+        if special_mode:
+            if (std_dev_data[idx] != None):
+                plot = ax.errorbar(x, y, yerr=df[std_dev_data[idx]].astype(float), fmt='o', ecolor='black', capsize=5, color="navy", label=data_series_title)
+            else:
+                plot = ax.plot(x, y, 'o', color="navy", label=data_series_title)
         else:
-            plot = ax.plot(x, y, 'o', color=color, label=data_series_title)
+            if (std_dev_data[idx] != None):
+                plot = ax.errorbar(x, y, yerr=df[std_dev_data[idx]].astype(float), fmt='o', ecolor='black', capsize=5, color=color, label=data_series_title)
+            else:
+                plot = ax.plot(x, y, color=color, label=data_series_title)
 
         if (type(plot) == list):
             plots.extend(plot)
@@ -122,6 +131,17 @@ def plot_data(x_data, y_data, std_dev_data, color_picker, labels, df,
     ax.set_xscale(x_axis_scale)
     ax.set_yscale(y_axis_scale)
 
+    if special_mode:
+        ax.grid(linestyle="dashed", dashes=(5,3))
+        ax.spines[['right', 'top']].set_visible(False)
+        ax.tick_params(axis='x', length=0)
+        ax.tick_params(axis='y', length=0)
+        arial_font = FontProperties(fname='./arial.ttf')
+        mpl.font_manager.fontManager.addfont('./arial.ttf')
+        with mpl.rc_context({"font.family": arial_font.get_name(), "font.size": 10}):
+            ax.legend(handles, labels, loc='best')
+            return fig
+
     return fig
 
 @app.post("/post")
@@ -158,7 +178,9 @@ async def create_plot(request: PlotDataRequest, data: Request):
                     title=request.title, x_label=request.x_label, y_label=request.y_label,
                     enable_trendline=request.enable_trendline, enable_grid=request.enable_grid,
                     constant_line=constant_line,
-                    x_axis_scale=request.x_axis_scale, y_axis_scale=request.y_axis_scale, trendline_equation=request.trendline_equation)
+                    x_axis_scale=request.x_axis_scale, y_axis_scale=request.y_axis_scale, trendline_equation=request.trendline_equation,
+                    special_mode=request.special_mode
+                    )
 
     buf = BytesIO()
     fig.savefig(buf, format="png")