Update app.py

app.py

@@ -1,39 +1,22 @@
-from fastapi import FastAPI, UploadFile, File
-from fastapi.responses import FileResponse
-from fastapi.openapi.docs import get_swagger_ui_html
+import gradio as gr
 import subprocess
 import uuid
 import os
 
-app = FastAPI()
-
-# -------- Healthcheck Endpoint --------
-@app.get("/")
-def health():
-    return {"status": "ok", "ffmpeg": True}
-
-# -------- Custom Swagger /docs --------
-@app.get("/docs", include_in_schema=False)
-def custom_swagger():
-    return get_swagger_ui_html(openapi_url="/openapi.json", title="FFmpeg API Docs")
-
-# -------- Trim Video Endpoint --------
-@app.post("/trim")
-async def trim(
-    file: UploadFile = File(...),
-    start: float = 0,
-    duration: float = 5
-):
+# -------- Video Trim Funktion --------
+def trim_video(file, start, duration):
+    # Pfade für temporäre Dateien
     input_path = f"/tmp/{uuid.uuid4()}_in.mp4"
     output_path = f"/tmp/{uuid.uuid4()}_out.mp4"
 
-    # Upload speichern
+    # Video speichern
     with open(input_path, "wb") as f:
-        f.write(await file.read())
+        f.write(file.read())
 
     # FFmpeg Befehl ausführen
     cmd = [
-        "ffmpeg", "-y",
+        "ffmpeg",
+        "-y",
         "-ss", str(start),
         "-i", input_path,
         "-t", str(duration),
@@ -42,9 +25,20 @@ async def trim(
     ]
     subprocess.run(cmd, check=True)
 
-    # Output zurückgeben
-    return FileResponse(
-        output_path,
-        media_type="video/mp4",
-        filename="trimmed.mp4"
-    )
+    return output_path
+
+# -------- Gradio Interface --------
+iface = gr.Interface(
+    fn=trim_video,
+    inputs=[
+        gr.File(label="Video hochladen", file_types=[".mp4"]),
+        gr.Number(label="Startzeit (Sekunden)", value=0),
+        gr.Number(label="Dauer (Sekunden)", value=5)
+    ],
+    outputs=gr.Video(label="Getrimmtes Video"),
+    title="FFmpeg Video Editor",
+    description="Trimme Videos direkt im Hugging Face Space"
+)
+
+# -------- Starten des Servers --------
+iface.launch(server_name="0.0.0.0", server_port=7860)