1580158
/
PromptGallery-stable-diffusion-webui
mirror of https://github.com/dr413677671/PromptGallery-stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix extention arg 

fix extension arg

fi
main
dr413677671 2023-08-06 01:27:06 +08:00
parent 89b20a5887
commit ef9aabc847
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
paste_this_to_webui_scripts_folder/prompt_gallery.py
View File

@ -17,7 +17,7 @@ import base64
import io import io
from PIL import Image from PIL import Image
import mimetypes import mimetypes
from modules.paths_internal import extensions_dir
mimetypes.init() mimetypes.init()
mimetypes.add_type('application/javascript', '.js') mimetypes.add_type('application/javascript', '.js')
@ -35,16 +35,16 @@ if '__file__' in locals().keys():
else: else:
root_path = os.path.abspath(shared.script_path) root_path = os.path.abspath(shared.script_path)
rela_path = os.path.join( 'extensions') rela_path = os.path.join('extensions')
try: try:
with open( os.path.join( rela_path, 'prompt_gallery_name.json') ) as fd: with open(os.path.join(extensions_dir, 'prompt_gallery_name.json')) as fd:
extension_name = json.load(fd)['name'] extension_name = json.load(fd)['name']
except: except:
extension_name = "Prompt Gallery" extension_name = "Prompt Gallery"
OUTPATH_SAMPLES = os.path.join(rela_path, extension_name, 'assets', 'preview') OUTPATH_SAMPLES = os.path.join(extensions_dir, extension_name, 'assets', 'preview')
OUTPATH_GRIDS = os.path.join(rela_path, extension_name, 'assets', 'grid') OUTPATH_GRIDS = os.path.join(extensions_dir, extension_name, 'assets', 'grid')
BATCH_SIZE = 2 BATCH_SIZE = 2
N_ITER = 2 N_ITER = 2

5
scripts/prompt_gallery.py
View File

@ -7,6 +7,7 @@ import modules.extras
import modules.ui import modules.ui
from modules.shared import opts, cmd_opts from modules.shared import opts, cmd_opts
from modules import shared, scripts from modules import shared, scripts
from modules.paths_internal import extensions_dir
from modules import script_callbacks from modules import script_callbacks
from pathlib import Path from pathlib import Path
from typing import List, Tuple from typing import List, Tuple
@ -31,11 +32,11 @@ pg_port = 5173
def on_ui_settings(): def on_ui_settings():
global pg_ip global pg_ip
with open("./extensions/prompt_gallery_name.json") as fd: with open(os.path.join(extensions_dir, 'prompt_gallery_name.json')) as fd:
name = json.load(fd)['name'] name = json.load(fd)['name']
# os.chmod('./extensions/'+name, stat.S_IRWXO) # os.chmod('./extensions/'+name, stat.S_IRWXO)
app = FastAPI() app = FastAPI()
app.mount('/', StaticFiles(directory='./extensions/'+name,html=True)) app.mount('/', StaticFiles(directory=extensions_dir+"/"+name,html=True))
config = Config(app=app, host=pg_ip,port=pg_port, log_level="info", loop="asyncio", limit_max_requests=1) config = Config(app=app, host=pg_ip,port=pg_port, log_level="info", loop="asyncio", limit_max_requests=1)
app.add_middleware( app.add_middleware(
CORSMiddleware, CORSMiddleware,