🔀 refactor(main.py): extract get_static_files_dir() function to improve code readability and reusability

🔀 refactor(main.py): simplify setup_app() function by using get_static_files_dir() function to get static files directory
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-18 07:26:30 -03:00
commit ab61890997

View file

@ -59,14 +59,17 @@ def setup_static_files(app: FastAPI, static_files_dir: Path):
return FileResponse(path)
# app = create_app()
# setup_static_files(app, static_files_dir)
def get_static_files_dir():
"""Get the static files directory relative to Langflow's main.py file."""
frontend_path = Path(__file__).parent
return frontend_path / "frontend"
def setup_app(static_files_dir: Optional[Path] = None) -> FastAPI:
"""Setup the FastAPI app."""
# get the directory of the current file
if not static_files_dir:
frontend_path = Path(__file__).parent
static_files_dir = frontend_path / "frontend"
static_files_dir = get_static_files_dir()
app = create_app()
setup_static_files(app, static_files_dir)