app.py

import gradio as gr
import subprocess
import os

def enhance_video(input_file):
    """
    Hàm xử lý video đầu vào và trả về video đã tăng cường.
    """
    # Kiểm tra xem input_file có tồn tại không
    if not input_file:
        return "Không có file video nào được tải lên."

    # Sử dụng đường dẫn file do Gradio cung cấp
    input_path = input_file
    output_path = "enhanced_output.mp4"

    # Định nghĩa các bộ lọc ffmpeg
    filters = [
        "unsharp=5:5:1.0",  # Làm sắc nét video
        "hqdn3d=4:3:6:4",   # Giảm nhiễu
        "eq=contrast=1.2:brightness=0.1"  # Điều chỉnh tương phản và độ sáng
    ]
    
    # Kết hợp các bộ lọc thành chuỗi
    filter_chain = ",".join(filters)
    
    # Xây dựng lệnh ffmpeg
    command = [
        "ffmpeg",
        "-i", input_path,          # File đầu vào
        "-vf", filter_chain,       # Áp dụng các bộ lọc
        "-c:a", "copy",            # Sao chép âm thanh nguyên bản
        output_path                # File đầu ra
    ]
    
    try:
        # Thực thi lệnh ffmpeg
        subprocess.run(command, check=True)
        return output_path  # Trả về file đầu ra
    except subprocess.CalledProcessError as e:
        return f"Lỗi khi xử lý video: {e}"

# Tạo giao diện Gradio
with gr.Blocks() as demo:
    gr.Markdown("# Video Enhancement with FFmpeg")
    gr.Markdown("Upload a video and get an enhanced version with improved quality.")
    
    with gr.Row():
        input_video = gr.File(label="Upload Video")
        output_video = gr.Video(label="Enhanced Video")
    
    submit_button = gr.Button("Enhance Video")
    submit_button.click(enhance_video, inputs=input_video, outputs=output_video)

# Khởi chạy ứng dụng
demo.launch()