Add get fooocus styles api

mrhan1993 · Oct 25, 2023 · da6d273 · da6d273
1 parent 65edfd2
commit da6d273
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -109,4 +109,7 @@ Get all filenames of base model and lora.
 #### Refresh Models
 > POST /v1/engines/refresh-models
 
-Refresh local files and get all filenames of base model and lora.
+#### Get All Fooocus Styles
+> GET /v1/engines/styles
+
+Get all legal Fooocus styles.
diff --git a/docs/openapi.json b/docs/openapi.json
@@ -497,6 +497,29 @@
                     }
                 }
             }
+        },
+        "/v1/engines/styles": {
+            "get": {
+                "summary": "All Styles",
+                "description": "Get all legal Fooocus styles",
+                "operationId": "all_styles_v1_engines_styles_get",
+                "responses": {
+                    "200": {
+                        "description": "Successful Response",
+                        "content": {
+                            "application/json": {
+                                "schema": {
+                                    "items": {
+                                        "type": "string"
+                                    },
+                                    "type": "array",
+                                    "title": "Response All Styles V1 Engines Styles Get"
+                                }
+                            }
+                        }
+                    }
+                }
+            }
         }
     },
     "components": {

diff --git a/fooocus_api_version.py b/fooocus_api_version.py
@@ -1 +1 @@
-version = '0.2.3'
+version = '0.2.4'
diff --git a/fooocusapi/api.py b/fooocusapi/api.py
@@ -164,6 +164,12 @@ def refresh_models():
     path.update_all_model_names()
     return AllModelNamesResponse(model_filenames=path.model_filenames, lora_filenames=path.lora_filenames)
 
+
+@app.get("/v1/engines/styles", response_model=List[str], description="Get all legal Fooocus styles")
+def all_styles():
+    from modules.sdxl_styles import legal_style_names
+    return legal_style_names
+
 def start_app(args):
     uvicorn.run("fooocusapi.api:app", host=args.host,
                 port=args.port, log_level=args.log_level)