fix

modules/app.py

@@ -34,7 +34,7 @@ outpath = os.path.join(os.path.expanduser("~"), OUTPUT_DIRNAME)
 # don't show access log entries for URLs that start with the given prefix
 ACCESS_LOG_SUPPRESS_PATH_PREFIXES = ['/ping', '/modifier-thumbnails']
 
-app.mount('media', StaticFiles(directory=os.path.join(SD_UI_DIR, 'media/')), name="media")
+app.mount('/media', StaticFiles(directory=os.path.join(SD_UI_DIR, 'media/')), name="media")
 
 # defaults from https://huggingface.co/blog/stable_diffusion
 class ImageRequest(BaseModel):
@@ -68,12 +68,12 @@ class ImageRequest(BaseModel):
 class SetAppConfigRequest(BaseModel):
     update_branch: str = "main"
 
-@app.get('')
+@app.get('/')
 def read_root():
     headers = {"Cache-Control": "no-cache, no-store, must-revalidate", "Pragma": "no-cache", "Expires": "0"}
     return FileResponse(os.path.join(SD_UI_DIR, 'index.html'), headers=headers)
 
-@app.get('ping')
+@app.get('/ping')
 async def ping():
     global model_loaded, model_is_loading
 
@@ -138,7 +138,7 @@ def save_model_to_config(model_name):
 
     setConfig(config)
 
-@app.post('image')
+@app.post('/image')
 def image(req : ImageRequest):
     import runtime
 
@@ -192,7 +192,7 @@ def image(req : ImageRequest):
         print(traceback.format_exc())
         return HTTPException(status_code=500, detail=str(e))
 
-@app.get('image/stop')
+@app.get('/image/stop')
 def stop():
     try:
         if model_is_loading:
@@ -206,14 +206,14 @@ def stop():
         print(traceback.format_exc())
         return HTTPException(status_code=500, detail=str(e))
 
-@app.get('image/tmp/{session_id}/{img_id}')
+@app.get('/image/tmp/{session_id}/{img_id}')
 def get_image(session_id, img_id):
     import runtime
     buf = runtime.temp_images[session_id + '/' + img_id]
     buf.seek(0)
     return StreamingResponse(buf, media_type='image/jpeg')
 
-@app.post('app_config')
+@app.post('/app_config')
 async def setAppConfig(req : SetAppConfigRequest):
     try:
         config = {
@@ -242,7 +242,7 @@ async def setAppConfig(req : SetAppConfigRequest):
         print(traceback.format_exc())
         return HTTPException(status_code=500, detail=str(e))
 
-@app.get('app_config')
+@app.get('/app_config')
 def getAppConfig():
     try:
         config_json_path = os.path.join(CONFIG_DIR, 'config.json')
@@ -277,7 +277,7 @@ def setConfig(config):
     except:
         print(traceback.format_exc())
 
-@app.get('models')
+@app.get('/models')
 def getModels():
     models = {
         'active': {
@@ -307,12 +307,12 @@ def getModels():
 
     return models
 
-@app.get('modifiers.json')
+@app.get('/modifiers.json')
 def read_modifiers():
     headers = {"Cache-Control": "no-cache, no-store, must-revalidate", "Pragma": "no-cache", "Expires": "0"}
     return FileResponse(os.path.join(SD_UI_DIR, 'modifiers.json'), headers=headers)
 
-@app.get('output_dir')
+@app.get('/output_dir')
 def read_home_dir():
     return {outpath}