mit REST API. Bilder müssen per POST als Binärdatei übergeben werden
This commit is contained in:
Fedja Windows
parent
e313edd8ab
commit
88fc7017e6
150
api_server.py
Normal file
150
api_server.py
Normal file
@ -0,0 +1,150 @@
|
||||
# api_server.py - REST-API für den Videogenerator
|
||||
# Im webui-Verzeichnis speichern
|
||||
|
||||
from fastapi import FastAPI, File, UploadFile, BackgroundTasks, Form
|
||||
from fastapi.responses import FileResponse
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
import uvicorn
|
||||
import os
|
||||
import subprocess
|
||||
import uuid
|
||||
import shutil
|
||||
from typing import Optional
|
||||
|
||||
app = FastAPI(title="Hunyuan Video Generator API")
|
||||
|
||||
# CORS für n8n-Zugriff erlauben
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=["*"], # Im Produktivbetrieb einschränken auf n8n-Server-IP
|
||||
allow_credentials=True,
|
||||
allow_methods=["*"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
|
||||
# Verzeichnisse für temporäre Dateien und Ausgaben
|
||||
TEMP_DIR = os.path.join(os.path.dirname(__file__), "temp_uploads")
|
||||
OUTPUT_DIR = os.path.join(os.path.dirname(__file__), "outputs")
|
||||
PYTHON_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "system", "python", "python.exe")
|
||||
|
||||
os.makedirs(TEMP_DIR, exist_ok=True)
|
||||
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
||||
|
||||
def generate_video_task(image_path: str, prompt: str, n_prompt: str,
|
||||
seed: int, length: float, steps: int,
|
||||
use_teacache: bool, output_filename: str,
|
||||
job_id: str):
|
||||
"""Führt die Videogenerierung als Hintergrundaufgabe aus"""
|
||||
|
||||
# Befehlszeile erstellen
|
||||
cmd = [
|
||||
PYTHON_PATH,
|
||||
"hunyuan_cli.py",
|
||||
"--image", image_path,
|
||||
"--prompt", prompt,
|
||||
"--seed", str(seed),
|
||||
"--length", str(length),
|
||||
"--steps", str(steps),
|
||||
"--output", f"{job_id}_{output_filename}"
|
||||
]
|
||||
|
||||
if n_prompt:
|
||||
cmd.extend(["--n_prompt", n_prompt])
|
||||
|
||||
if use_teacache:
|
||||
cmd.append("--teacache")
|
||||
|
||||
# Befehl ausführen
|
||||
try:
|
||||
subprocess.run(cmd, check=True, cwd=os.path.dirname(__file__))
|
||||
print(f"Video generation completed for job {job_id}")
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"Error generating video: {e}")
|
||||
|
||||
@app.post("/generate/")
|
||||
async def generate_video(
|
||||
background_tasks: BackgroundTasks,
|
||||
image: UploadFile = File(...),
|
||||
prompt: str = Form(...),
|
||||
n_prompt: Optional[str] = Form(""),
|
||||
seed: Optional[int] = Form(31337),
|
||||
length: Optional[float] = Form(5.0),
|
||||
steps: Optional[int] = Form(25),
|
||||
use_teacache: Optional[bool] = Form(True),
|
||||
output_filename: Optional[str] = Form("output.mp4")
|
||||
):
|
||||
"""
|
||||
Generiert ein Video basierend auf den angegebenen Parametern
|
||||
"""
|
||||
# Eindeutige Job-ID generieren
|
||||
job_id = str(uuid.uuid4())
|
||||
|
||||
# Temporären Dateipfad für das Bild erstellen
|
||||
temp_image_path = os.path.join(TEMP_DIR, f"{job_id}_{image.filename}")
|
||||
|
||||
# Bild speichern
|
||||
with open(temp_image_path, "wb") as buffer:
|
||||
shutil.copyfileobj(image.file, buffer)
|
||||
|
||||
# Videogenerierung als Hintergrundaufgabe starten
|
||||
background_tasks.add_task(
|
||||
generate_video_task,
|
||||
temp_image_path,
|
||||
prompt,
|
||||
n_prompt,
|
||||
seed,
|
||||
length,
|
||||
steps,
|
||||
use_teacache,
|
||||
output_filename,
|
||||
job_id
|
||||
)
|
||||
|
||||
return {
|
||||
"status": "processing",
|
||||
"job_id": job_id,
|
||||
"message": "Video generation started in background",
|
||||
"result_url": f"/result/{job_id}_{output_filename}"
|
||||
}
|
||||
|
||||
@app.get("/result/{filename}")
|
||||
async def get_result(filename: str):
|
||||
"""
|
||||
Gibt das generierte Video zurück, wenn es verfügbar ist
|
||||
"""
|
||||
file_path = os.path.join(OUTPUT_DIR, filename)
|
||||
|
||||
if os.path.exists(file_path):
|
||||
return FileResponse(
|
||||
file_path,
|
||||
media_type="video/mp4",
|
||||
filename=filename.split("_", 1)[1] # Entferne Job-ID vom Dateinamen
|
||||
)
|
||||
else:
|
||||
return {"status": "not_found", "message": "Requested video not found or still processing"}
|
||||
|
||||
@app.get("/status/{job_id}")
|
||||
async def check_status(job_id: str):
|
||||
"""
|
||||
Prüft den Status einer Videogenerierung
|
||||
"""
|
||||
# Suche nach Dateien, die mit der Job-ID beginnen
|
||||
result_files = [f for f in os.listdir(OUTPUT_DIR) if f.startswith(job_id)]
|
||||
|
||||
if result_files:
|
||||
return {
|
||||
"status": "completed",
|
||||
"job_id": job_id,
|
||||
"files": result_files,
|
||||
"download_urls": [f"/result/{file}" for file in result_files]
|
||||
}
|
||||
else:
|
||||
# Prüfen, ob das Eingabebild noch existiert (Verarbeitung läuft noch)
|
||||
input_files = [f for f in os.listdir(TEMP_DIR) if f.startswith(job_id)]
|
||||
if input_files:
|
||||
return {"status": "processing", "job_id": job_id}
|
||||
else:
|
||||
return {"status": "not_found", "job_id": job_id}
|
||||
|
||||
if __name__ == "__main__":
|
||||
uvicorn.run(app, host="0.0.0.0", port=8000)
|
||||
207
n8n_workflow.json
Normal file
207
n8n_workflow.json
Normal file
@ -0,0 +1,207 @@
|
||||
{
|
||||
"nodes": [
|
||||
{
|
||||
"parameters": {
|
||||
"httpMethod": "POST",
|
||||
"path": "generate-video",
|
||||
"options": {}
|
||||
},
|
||||
"name": "Webhook",
|
||||
"type": "n8n-nodes-base.webhook",
|
||||
"typeVersion": 1,
|
||||
"position": [
|
||||
250,
|
||||
300
|
||||
]
|
||||
},
|
||||
{
|
||||
"parameters": {
|
||||
"url": "=http://{{$node[\"Webhook\"].json[\"server_ip\"]}}/generate/",
|
||||
"options": {
|
||||
"formData": {
|
||||
"values": {
|
||||
"image": {
|
||||
"property": "=data:application/octet-stream;base64,{{$node[\"Webhook\"].json[\"image_base64\"]}}",
|
||||
"fileName": "=input.jpg"
|
||||
},
|
||||
"prompt": "={{$node[\"Webhook\"].json[\"prompt\"]}}",
|
||||
"n_prompt": "={{$node[\"Webhook\"].json[\"n_prompt\"]}}",
|
||||
"seed": "={{$node[\"Webhook\"].json[\"seed\"]}}",
|
||||
"length": "={{$node[\"Webhook\"].json[\"length\"]}}",
|
||||
"steps": "={{$node[\"Webhook\"].json[\"steps\"]}}",
|
||||
"use_teacache": "={{$node[\"Webhook\"].json[\"use_teacache\"]}}",
|
||||
"output_filename": "={{$node[\"Webhook\"].json[\"output_filename\"]}}"
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"name": "HTTP Request",
|
||||
"type": "n8n-nodes-base.httpRequest",
|
||||
"typeVersion": 4.1,
|
||||
"position": [
|
||||
500,
|
||||
300
|
||||
]
|
||||
},
|
||||
{
|
||||
"parameters": {
|
||||
"rule": {
|
||||
"interval": [
|
||||
{
|
||||
"field": "seconds",
|
||||
"secondsInterval": 10
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"name": "Polling Trigger",
|
||||
"type": "n8n-nodes-base.scheduleTrigger",
|
||||
"typeVersion": 1.1,
|
||||
"position": [
|
||||
250,
|
||||
500
|
||||
]
|
||||
},
|
||||
{
|
||||
"parameters": {
|
||||
"url": "=http://{{$node[\"HTTP Request\"].json[\"job_id\"]}}/status/{{$node[\"HTTP Request\"].json[\"job_id\"]}}",
|
||||
"options": {}
|
||||
},
|
||||
"name": "Check Status",
|
||||
"type": "n8n-nodes-base.httpRequest",
|
||||
"typeVersion": 4.1,
|
||||
"position": [
|
||||
500,
|
||||
500
|
||||
]
|
||||
},
|
||||
{
|
||||
"parameters": {
|
||||
"conditions": {
|
||||
"string": [
|
||||
{
|
||||
"value1": "={{$node[\"Check Status\"].json[\"status\"]}}",
|
||||
"operation": "equals",
|
||||
"value2": "completed"
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"name": "IF",
|
||||
"type": "n8n-nodes-base.if",
|
||||
"typeVersion": 1,
|
||||
"position": [
|
||||
700,
|
||||
500
|
||||
]
|
||||
},
|
||||
{
|
||||
"parameters": {
|
||||
"url": "=http://{{$node[\"Webhook\"].json[\"server_ip\"]}}/result/{{$node[\"HTTP Request\"].json[\"job_id\"]}}_{{$node[\"Webhook\"].json[\"output_filename\"]}}",
|
||||
"options": {
|
||||
"response": {
|
||||
"response": {
|
||||
"fullResponse": true,
|
||||
"responseFormat": "file"
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"name": "Download Result",
|
||||
"type": "n8n-nodes-base.httpRequest",
|
||||
"typeVersion": 4.1,
|
||||
"position": [
|
||||
900,
|
||||
450
|
||||
]
|
||||
},
|
||||
{
|
||||
"parameters": {
|
||||
"values": {
|
||||
"string": [
|
||||
{
|
||||
"name": "jobId",
|
||||
"value": "={{$node[\"HTTP Request\"].json[\"job_id\"]}}"
|
||||
},
|
||||
{
|
||||
"name": "status",
|
||||
"value": "={{$node[\"Check Status\"].json[\"status\"]}}"
|
||||
}
|
||||
]
|
||||
},
|
||||
"options": {}
|
||||
},
|
||||
"name": "Continue Polling",
|
||||
"type": "n8n-nodes-base.set",
|
||||
"typeVersion": 2,
|
||||
"position": [
|
||||
900,
|
||||
550
|
||||
]
|
||||
}
|
||||
],
|
||||
"connections": {
|
||||
"Webhook": {
|
||||
"main": [
|
||||
[
|
||||
{
|
||||
"node": "HTTP Request",
|
||||
"type": "main",
|
||||
"index": 0
|
||||
}
|
||||
]
|
||||
]
|
||||
},
|
||||
"HTTP Request": {
|
||||
"main": [
|
||||
[
|
||||
{
|
||||
"node": "Polling Trigger",
|
||||
"type": "main",
|
||||
"index": 0
|
||||
}
|
||||
]
|
||||
]
|
||||
},
|
||||
"Polling Trigger": {
|
||||
"main": [
|
||||
[
|
||||
{
|
||||
"node": "Check Status",
|
||||
"type": "main",
|
||||
"index": 0
|
||||
}
|
||||
]
|
||||
]
|
||||
},
|
||||
"Check Status": {
|
||||
"main": [
|
||||
[
|
||||
{
|
||||
"node": "IF",
|
||||
"type": "main",
|
||||
"index": 0
|
||||
}
|
||||
]
|
||||
]
|
||||
},
|
||||
"IF": {
|
||||
"main": [
|
||||
[
|
||||
{
|
||||
"node": "Download Result",
|
||||
"type": "main",
|
||||
"index": 0
|
||||
}
|
||||
],
|
||||
[
|
||||
{
|
||||
"node": "Continue Polling",
|
||||
"type": "main",
|
||||
"index": 0
|
||||
}
|
||||
]
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
12
start_api_server.bat
Normal file
12
start_api_server.bat
Normal file
@ -0,0 +1,12 @@
|
||||
@echo off
|
||||
|
||||
call environment.bat
|
||||
|
||||
cd %~dp0webui
|
||||
|
||||
"%DIR%\python\python.exe" -m pip install fastapi uvicorn python-multipart
|
||||
|
||||
"%DIR%\python\python.exe" api_server.py
|
||||
|
||||
:done
|
||||
pause
|
||||
Loading…
x
Reference in New Issue
Block a user