From 8f1918d2b5ae3ea304bc2054680f890ff8185443 Mon Sep 17 00:00:00 2001
From: Roland Li <roland@arkavian.com>
Date: Wed, 21 Feb 2024 15:55:49 -0800
Subject: [PATCH 1/2] add name to args

---
 plugin.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/plugin.py b/plugin.py
index 5f96941..4f1a317 100644
--- a/plugin.py
+++ b/plugin.py
@@ -37,21 +37,22 @@ def set_config(update: dict):
     sd_plugin.set_config(update) # TODO: Validate config dict are all valid keys
     return sd_plugin.get_config()
 
-@app.on_event("startup")
-async def startup_event():
+@app.get("/startup/{plugin_name}")
+async def startup_event(plugin_name: str):
     print("Starting up")
     # A slight delay to ensure the app has started up.
     try:
-        set_model()
+        set_model(plugin_name)
         print("Successfully started up")
+        print(sd_plugin.plugin_name)
         sd_plugin.notify_main_system_of_startup("True")
-    except:
+    except Exception as e:
         sd_plugin.notify_main_system_of_startup("False")
 
 @app.get("/set_model/")
-def set_model():
+def set_model(plugin_name):
     global sd_plugin
-    args = {"plugin": plugin, "config": config, "endpoints": endpoints}
+    args = {"plugin": plugin, "config": config, "endpoints": endpoints, "name": plugin_name}
     sd_plugin = SD(Namespace(**args))
     # try:
     # sd_plugin.set_model(args["model_name"], dtype=args["model_dtype"])
@@ -106,7 +107,6 @@ class SD(Plugin):
     """
     def __init__(self, arguments: "Namespace") -> None:
         super().__init__(arguments)
-        self.plugin_name = "Diffusers"
         self.set_model()
 
     def load_lora_weights(self, pipeline, checkpoint_path, multiplier=1):

From 30dd05b5d917bc761334bc6bf2471f42aed18b82 Mon Sep 17 00:00:00 2001
From: Roland Li <roland@arkavian.com>
Date: Fri, 26 Apr 2024 09:12:50 -0700
Subject: [PATCH 2/2] status

---
 plugin.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/plugin.py b/plugin.py
index 4f1a317..5d1bcb4 100644
--- a/plugin.py
+++ b/plugin.py
@@ -46,6 +46,8 @@ async def startup_event(plugin_name: str):
         print("Successfully started up")
         print(sd_plugin.plugin_name)
         sd_plugin.notify_main_system_of_startup("True")
+        return {"status": "Success", "detail": "Plugin started successfully"}
+
     except Exception as e:
         sd_plugin.notify_main_system_of_startup("False")