From 94ce9f9fa162ddf19653377e68794f86d682a2c9 Mon Sep 17 00:00:00 2001
From: Kenshin Abe <abe.kenshin@gmail.com>
Date: Fri, 26 Jan 2024 19:12:24 +0900
Subject: [PATCH] Implement API

---
 optuna_dashboard/_app.py         | 8 ++++++++
 optuna_dashboard/ts/apiClient.ts | 4 ++++
 2 files changed, 12 insertions(+)

diff --git a/optuna_dashboard/_app.py b/optuna_dashboard/_app.py
index 1c9c0119..5057353d 100644
--- a/optuna_dashboard/_app.py
+++ b/optuna_dashboard/_app.py
@@ -271,6 +271,14 @@ def create_app(
         )
         if plot_type == "contour":
             fig = optuna.visualization.plot_contour(study)
+        elif plot_type == "slice":
+            fig = optuna.visualization.plot_slice(study)
+        elif plot_type == "parallel_coordinate":
+            fig = optuna.visualization.plot_parallel_coordinate(study)
+        elif plot_type == "rank":
+            fig = optuna.visualization.plot_rank(study)
+        elif plot_type == "edf":
+            fig = optuna.visualization.plot_edf(study)
         else:
             response.status = 404  # Not found
             return {"reason": f"plot_type={plot_type} is not supported."}
diff --git a/optuna_dashboard/ts/apiClient.ts b/optuna_dashboard/ts/apiClient.ts
index 3798f602..46fe3d0e 100644
--- a/optuna_dashboard/ts/apiClient.ts
+++ b/optuna_dashboard/ts/apiClient.ts
@@ -448,6 +448,10 @@ type PlotResponse = {
 }
 export enum PlotType {
   Contour = "contour",
+  Slice = "slice",
+  ParallelCoordinate = "parallel_coordinate",
+  Rank = "rank",
+  EDF = "edf",
 }
 export const getPlotAPI = (
   studyId: number,