Skip to content

MERGE #705

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 18 commits into
base: main
Choose a base branch
from
14 changes: 9 additions & 5 deletions app/asgi.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,12 @@
"""Application implementation - ASGI."""

from os import path
from fastapi import FastAPI, Request
from fastapi.exceptions import RequestValidationError
from fastapi.responses import JSONResponse
from loguru import logger
from fastapi.staticfiles import StaticFiles

# from starlette.routing import Router
from app.config import config
from app.models.exception import HttpException
from app.router import root_api_router
Expand Down Expand Up @@ -39,16 +40,19 @@ def get_application() -> FastAPI:
version=config.project_version,
debug=False,
)

# instance.mount("/", StaticFiles(directory=utils.public_dir(), html=True), name="")

instance.include_router(root_api_router)
instance.add_exception_handler(HttpException, exception_handler)
instance.add_exception_handler(RequestValidationError, validation_exception_handler)
return instance


app = get_application()
public_dir = utils.public_dir()
app.mount("/", StaticFiles(directory=public_dir, html=True), name="")

sub_app = FastAPI()
video_dir = utils.video_dir()
sub_app.mount("/videos", StaticFiles(directory=video_dir), name="videos")
app.mount("/subapp", sub_app)

@app.on_event("shutdown")
def shutdown_event():
Expand Down
3 changes: 1 addition & 2 deletions app/controllers/ping.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,8 @@
from fastapi import APIRouter
from fastapi import Request

router = APIRouter()


@router.get("/ping", tags=["Health Check"], description="检查服务可用性", response_description="pong")
def ping(request: Request) -> str:
return "pong"
return "pong"
19 changes: 17 additions & 2 deletions app/utils/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,9 +102,24 @@ def song_dir(sub_dir: str = ""):
os.makedirs(d)
return d


def public_dir(sub_dir: str = ""):
d = resource_dir(f"public")
d = os.path.join(root_dir(), "public")
if sub_dir:
d = os.path.join(d, sub_dir)
if not os.path.exists(d):
os.makedirs(d)
return d

# def public_dir(sub_dir: str = ""):
# d = resource_dir(f"public")
# if sub_dir:
# d = os.path.join(d, sub_dir)
# if not os.path.exists(d):
# os.makedirs(d)
# return d

def video_dir(sub_dir: str = ""):
d = os.path.join(root_dir(), "videos")
if sub_dir:
d = os.path.join(d, sub_dir)
if not os.path.exists(d):
Expand Down
4 changes: 2 additions & 2 deletions webui/Main.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import streamlit as st

st.set_page_config(page_title="MoneyPrinterTurbo", page_icon="🤖", layout="wide",
st.set_page_config(page_title="Text to Video", page_icon="🤖", layout="wide",
initial_sidebar_state="auto")
import sys
import os
Expand All @@ -16,7 +16,7 @@
<style>#root > div:nth-child(1) > div > div > div > div > section > div {padding-top: 0rem;}</style>
"""
st.markdown(hide_streamlit_style, unsafe_allow_html=True)
st.title("MoneyPrinterTurbo")
st.title("Text to Video")

root_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
font_dir = os.path.join(root_dir, "resource", "fonts")
Expand Down