extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-05-22 01:00:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

adjust server

Browse Source
This commit is contained in:
caranicas 2022-09-14 12:57:03 -04:00
parent eefa7d53c5
commit a1ad1147e6
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ui/server.py
View File

@ -17,9 +17,10 @@ OUTPUT_DIRNAME = "Stable Diffusion UI" # in the user's home folder
from fastapi import FastAPI, HTTPException
from starlette.responses import FileResponse, StreamingResponse
from pydantic import BaseModel
import logging
# this is needed for development.
from fastapi.middleware.cors import CORSMiddleware
import logging
from sd_internal import Request, Response
app = FastAPI()
@ -71,7 +72,6 @@ class SetAppConfigRequest(BaseModel):
@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)
return FileResponse(os.path.join(SD_UI_DIR,'frontend/dist/index.html'), headers=headers)
# then get the js files
@ -84,6 +84,7 @@ def read_scripts():
def read_styles():
return FileResponse(os.path.join(SD_UI_DIR, 'frontend/dist/index.css'))
@app.get('/ping')
async def ping():
global model_loaded, model_is_loading
@ -216,7 +217,7 @@ def read_modifiers():
@app.get('/modifiers.json')
def read_modifiers():
return FileResponse(os.path.join(SD_UI_DIR, 'modifiers.json'))
return FileResponse(os.path.join(SD_UI_DIR, 'frontend/dist/modifiers.json'))
@app.get('/output_dir')
def read_home_dir():