fix extention arg

fix extension arg

fi

paste_this_to_webui_scripts_folder/prompt_gallery.py

@@ -17,7 +17,7 @@
 import io
 from PIL import Image
 import mimetypes
-
+from modules.paths_internal import extensions_dir
 mimetypes.init()
 mimetypes.add_type('application/javascript', '.js')
 
@@ -35,16 +35,16 @@
 else:
     root_path = os.path.abspath(shared.script_path)
 
-rela_path =  os.path.join( 'extensions')
+rela_path =  os.path.join('extensions')
 
 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']
 except:
     extension_name = "Prompt Gallery"
-
-OUTPATH_SAMPLES = os.path.join(rela_path, extension_name, 'assets', 'preview')
-OUTPATH_GRIDS =  os.path.join(rela_path, extension_name, 'assets', 'grid')
+    
+OUTPATH_SAMPLES = os.path.join(extensions_dir, extension_name, 'assets', 'preview')
+OUTPATH_GRIDS =  os.path.join(extensions_dir, extension_name, 'assets', 'grid')
 
 BATCH_SIZE = 2
 N_ITER = 2

scripts/prompt_gallery.py

@@ -7,6 +7,7 @@
 import modules.ui
 from modules.shared import opts, cmd_opts
 from modules import shared, scripts
+from modules.paths_internal import extensions_dir
 from modules import script_callbacks
 from pathlib import Path
 from typing import List, Tuple
@@ -31,11 +32,11 @@
 def on_ui_settings():
     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']
     # os.chmod('./extensions/'+name, stat.S_IRWXO)
     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)
     app.add_middleware(
         CORSMiddleware,