u

app.py

@@ -1,51 +1,65 @@
-import gradio as gr, json, os, subprocess, tempfile
+import gradio as gr
+import json, os, subprocess, tempfile, shlex, mimetypes
 
-# Load tất cả effect từ assets/
-effects = {}
-for f in os.listdir("assets"):
-    if f.endswith(".json"):
-        with open(f"assets/" + f, encoding="utf-8") as j:
-            for e in json.load(j).get("effects", []):
-                effects[e["name"]] = e["command"]
+def load_effects(path="assets"):
+    effects = {}
+    for file in os.listdir(path):
+        if file.endswith(".json"):
+            with open(os.path.join(path, file), encoding="utf-8") as f:
+                for e in json.load(f).get("effects", []):
+                    effects[e["name"]] = e["command"]
+    return effects
 
-def update_command(name):
-    return effects.get(name, "")
+EFFECTS = load_effects()
 
-def apply_effect(file, command, output_type):
-    if file is None:
-        return "❌ Chưa chọn file", None, ""
-
-    ext = ".gif" if output_type == "GIF" else ".mp4"
-    out_path = tempfile.NamedTemporaryFile(suffix=ext, delete=False).name
+def get_command(name): return EFFECTS.get(name, "")
 
-    # Build command
-    cmd = ["ffmpeg", "-y", "-i", file.name] + command.split() + ["-preset", "ultrafast", out_path]
-    log = " ".join(cmd)
+def run_ffmpeg(input_path, command_str, output_ext):
+    out = tempfile.NamedTemporaryFile(suffix=output_ext, delete=False).name
+    args = ["ffmpeg", "-y", "-i", input_path] + shlex.split(command_str) + ["-preset", "ultrafast", out]
+    try:
+        subprocess.run(args, check=True)
+        return "✅ Hiệu ứng đã tạo", out, " ".join(args)
+    except subprocess.CalledProcessError as e:
+        return f"❌ Lỗi FFmpeg: {e}", None, " ".join(args)
 
-    subprocess.run(cmd, check=True)
+def apply_effect(file, command, out_type):
+    if not file:
+        return "❌ Chưa chọn file", None, ""
+    ext = ".gif" if out_type == "GIF" else ".mp4"
+    return run_ffmpeg(file.name, command, ext)
 
-    return f"✅ Đã tạo hiệu ứng", out_path, log
+def preview_media(file):
+    if not file:
+        return None, None
+    mime, _ = mimetypes.guess_type(file.name)
+    if mime and mime.startswith("image"):
+        return file.name, None
+    elif mime and mime.startswith("video"):
+        return None, file.name
+    return None, None
 
 with gr.Blocks() as app:
-    gr.Markdown("🎞️ **Ứng dụng hiệu ứng video từ JSON (FFmpeg)**")
+    gr.Markdown("🎞️ **Hiệu ứng video bằng FFmpeg từ file JSON**")
 
-    with gr.Row():
-        f = gr.File(label="Ảnh hoặc Video")
-        output_type = gr.Radio(["MP4", "GIF"], value="MP4", label="Loại đầu ra")
+    f = gr.File(label="Tệp ảnh/video", file_types=["image", "video"])
+    t = gr.Radio(["MP4", "GIF"], value="MP4", label="Loại đầu ra")
+    name_input = gr.Textbox(label="Nhập tên hiệu ứng")
+    name_select = gr.Dropdown(choices=list(EFFECTS), label="Hoặc chọn hiệu ứng")
+    cmd_box = gr.Textbox(label="Lệnh FFmpeg", lines=2)
+    log_box = gr.Textbox(label="Log", lines=2, interactive=False)
+    msg = gr.Textbox(label="Trạng thái")
 
     with gr.Row():
-        effect_input = gr.Textbox(label="Nhập tên hiệu ứng")
-        effect_dropdown = gr.Dropdown(choices=list(effects), label="Hoặc chọn hiệu ứng")
-
-    command_box = gr.Textbox(label="Lệnh FFmpeg sẽ chạy", lines=2)
-    log_box = gr.Textbox(label="Lệnh đã chạy", lines=2, interactive=False)
+        preview_img = gr.Image(label="Ảnh đã chọn", visible=False)
+        preview_vid = gr.Video(label="Video đã chọn", visible=False)
 
-    run_btn = gr.Button("Tạo hiệu ứng")
-    out_msg = gr.Textbox(label="Trạng thái")
-    out_video = gr.Video()
+    out = gr.Video(label="Kết quả")
+    btn = gr.Button("Tạo hiệu ứng")
 
-    effect_input.change(update_command, effect_input, command_box)
-    effect_dropdown.change(update_command, effect_dropdown, command_box)
-    run_btn.click(apply_effect, [f, command_box, output_type], [out_msg, out_video, log_box])
+    name_input.change(get_command, name_input, cmd_box)
+    name_select.change(get_command, name_select, cmd_box)
+    f.change(preview_media, f, [preview_img, preview_vid])
+    btn.click(apply_effect, [f, cmd_box, t], [msg, out, log_box])
 
 app.launch()