Loading/saving of previous configuration for plugins.

Author: bryanlyon

main.py

@@ -284,7 +284,14 @@ def get_plugin_info(plugin_name: str):
 
 @app.get("/plugins/get_config/{plugin_name}")
 def get_plugin_config(plugin_name: str):
-    if plugin_name in plugin_list: 
+    if plugin_name in plugin_list:
+        sleep = 0
+        while plugin_states[plugin] != "RUNNING":
+            start_plugin(plugin)
+            time.sleep(5)
+            sleep += 5
+            if sleep > 120:
+                return {"status": "failed", "error": "Plugin too slow to start"}
         if plugin_name in port_mapping.keys():
             port = port_mapping[plugin_name]
             r = client.get("http://127.0.0.1:" + port + "/get_config")

plugin/__init__.py

@@ -11,6 +11,7 @@
 from tqdm import tqdm
 import shutil
 import threading
+import storage_db
 
 if sys.platform == "win32":
     storage_folder = os.path.join(os.getenv('APPDATA'),"DeepMake")
@@ -78,8 +79,9 @@ class Plugin():
     Generic plugin class
     """
 
-    def __init__(self, arguments={}):
-        self.plugin_name = "default"
+    def __init__(self, arguments={}, plugin_name="default"):
+        self.plugin_name = plugin_name
+        self.db = storage_db.storage_db()
         if arguments == {}:
             self.plugin = {}
             self.config = {}
@@ -88,7 +90,9 @@ def __init__(self, arguments={}):
             self.plugin = arguments.plugin
             self.config = arguments.config
             self.endpoints = arguments.endpoints
-            
+            config = self.db.retrieve_data(f"plugin_config.{self.plugin_name}")
+            if config:
+                self.config = config
 
         # Create a plugin-specific storage path
         self.plugin_storage_path = os.path.join(storage_folder, self.plugin_name)
@@ -103,10 +107,9 @@ def get_config(self):
 
     def set_config(self, update: dict):
         self.config.update(update) # TODO: Validate config dict are all valid keys
-        if "model_name" in update or "scheduler" in update or "loras" in update or "inverters" in update:
+        self.db.store_data(f"plugin_config.{self.plugin_name}", self.config)
+        if "model_name" in update:
             self.set_model()
-            # if response["status"] == "Failed":
-            #     return response
         return self.config
 
     def progress_callback(self, progress, stage):