ai / app /main.py
gtfgffg's picture
Upload 7 files
8a93ca3 verified
from fastapi import FastAPI, Request, UploadFile, File, Form
from fastapi.responses import HTMLResponse, JSONResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from starlette.datastructures import UploadFile as StarletteUploadFile
import os, uuid, shutil
from typing import Optional
app = FastAPI(title="FlexCut — تدوین هوشمند ویدیو (وب)")
# mount static and templates
app.mount("/static", StaticFiles(directory="app/static"), name="static")
templates = Jinja2Templates(directory="app/templates")
JOBS_DIR = "/app/jobs" if os.path.exists("/app/jobs") else os.path.join(os.getcwd(), "jobs")
os.makedirs(JOBS_DIR, exist_ok=True)
@app.get("/", response_class=HTMLResponse)
async def index(request: Request):
return templates.TemplateResponse("index.html", {"request": request})
def _save_upload(dst_dir: str, name: str, file: Optional[StarletteUploadFile]) -> Optional[str]:
if not file:
return None
path = os.path.join(dst_dir, name)
with open(path, "wb") as f:
shutil.copyfileobj(file.file, f)
return path
@app.post("/create")
async def create(
request: Request,
preset: str = Form(...),
music: Optional[UploadFile] = File(None),
video: Optional[UploadFile] = File(None),
logo: Optional[UploadFile] = File(None),
):
job_id = str(uuid.uuid4())[:8]
job_dir = os.path.join(JOBS_DIR, job_id)
os.makedirs(job_dir, exist_ok=True)
# save files if provided
saved = {}
if music: saved["music"] = _save_upload(job_dir, f"music_{music.filename}", music)
if video: saved["video"] = _save_upload(job_dir, f"video_{video.filename}", video)
if logo: saved["logo"] = _save_upload(job_dir, f"logo_{logo.filename}", logo)
# NOTE: این نسخه‌ی ساده فقط فایل‌ها را ذخیره می‌کند تا Space بالا بیاید.
# بعداً می‌توانیم مرحله‌های پردازش ویدیو (librosa/moviepy) را اضافه کنیم.
return templates.TemplateResponse(
"done.html",
{
"request": request,
"job_id": job_id,
"preset": preset,
"saved": saved,
},
)