AI.video.Saas / api /generate.py
Trkaga's picture
Create api/generate.py
cf6ec31 verified
raw
history blame contribute delete
447 Bytes
from flask import Blueprint, request
from services.runway import generate_video
from services.credit_system import has_credit, use_credit
gen_bp = Blueprint("gen", __name__)
@gen_bp.route("/generate", methods=["POST"])
def generate():
user = "demo"
if not has_credit(user):
return {"error": "No credits"}
prompt = request.json["prompt"]
video = generate_video(prompt)
use_credit(user)
return {"video": video}