Merge pull request #10194 from DumoeDss/dev

import re

import torch

from typing import Union

import scripts.api as api

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

    if added:

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

available_loras = {}

available_lora_aliases = {}

loaded_loras = []

list_available_loras()

try:

    import modules.script_callbacks as script_callbacks

    script_callbacks.on_app_started(api.api)

except:

    pass

 No newline at end of file

from fastapi import FastAPI

import gradio as gr

import json

import os

import lora

def get_lora_prompts(path):

    directory, filename = os.path.split(path)

    name_without_ext = os.path.splitext(filename)[0]

    new_filename = name_without_ext + '.civitai.info'

    try:

        new_path = os.path.join(directory, new_filename)

        if os.path.exists(new_path):

            with open(new_path, 'r') as f:

                data = json.load(f)

                trained_words = data.get('trainedWords', [])

                if len(trained_words) > 0:

                    result = ','.join(trained_words)

                    return result

                else:

                    return ''

        else:

            return ''

    except Exception as e:

        return ''

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

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

    async def get_loras():

        return [{"name": name, "path": lora.available_loras[name].filename, "prompt": get_lora_prompts(lora.available_loras[name].filename)} for name in lora.available_loras]