Add api method to get LoRA models with prompt

1 files changed, 3 insertions, 10 deletions

diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py
index 8fc1ddca..05162e41 100644
--- a/extensions-builtin/Lora/lora.py
+++ b/extensions-builtin/Lora/lora.py
@@ -2,9 +2,8 @@ import glob
 import os

 import re

 import torch

-from typing import Union, List, Optional

-from fastapi import FastAPI

-import gradio as gr

+from typing import Union

+import scripts.api as api

 

 from modules import shared, devices, sd_models, errors, scripts

 

@@ -445,12 +444,6 @@ def infotext_pasted(infotext, params):
     if added:

         params["Prompt"] += "\n" + "".join(added)

 

-def api(_: gr.Blocks, app: FastAPI):

-    @app.get("/sdapi/v1/loras")

-    async def getloras():

-        return [{"name": name, "path": available_loras[name].filename, "prompt": ""} for name in available_loras]

-

-

 available_loras = {}

 available_lora_aliases = {}

 loaded_loras = []

@@ -458,6 +451,6 @@ loaded_loras = []
 list_available_loras()

 try:

     import modules.script_callbacks as script_callbacks

-    script_callbacks.on_app_started(api)

+    script_callbacks.on_app_started(api.api)

 except:

     pass
\ No newline at end of file