app.py

import gradio as gr
import os
import zipfile
import tempfile
import shutil
from pathlib import Path
import cv2
from tqdm import tqdm

def process_videos(zip_file):
    # 一時ディレクトリを作成
    with tempfile.TemporaryDirectory() as temp_dir:
        # ZIPファイルを一時ディレクトリに解凍
        with zipfile.ZipFile(zip_file.name, 'r') as zip_ref:
            zip_ref.extractall(temp_dir)
        
        # 出力用の一時ディレクトリを作成
        output_temp_dir = tempfile.mkdtemp()
        
        # 処理したファイルの数を追跡
        processed_count = 0
        
        # 全てのmp4ファイルを処理
        for root, dirs, files in os.walk(temp_dir):
            for file in files:
                if file.lower().endswith('.mp4'):
                    input_path = os.path.join(root, file)
                    
                    # 出力パスの作成（元のディレクトリ構造を維持）
                    relative_path = os.path.relpath(os.path.join(root, file), temp_dir)
                    output_path = os.path.join(output_temp_dir, relative_path)
                    
                    # 出力ディレクトリの作成
                    os.makedirs(os.path.dirname(output_path), exist_ok=True)
                    
                    # ビデオのリサイズ処理
                    cap = cv2.VideoCapture(input_path)
                    
                    # 入力ビデオのプロパティを取得
                    fps = int(cap.get(cv2.CAP_PROP_FPS))
                    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
                    
                    # 出力ビデオライターの設定
                    out = cv2.VideoWriter(output_path, fourcc, fps, (1080, 1920))
                    
                    while cap.isOpened():
                        ret, frame = cap.read()
                        if not ret:
                            break
                        
                        # フレームをリサイズ
                        resized_frame = cv2.resize(frame, (1080, 1920))
                        out.write(resized_frame)
                    
                    cap.release()
                    out.release()
                    processed_count += 1
        
        if processed_count == 0:
            return None
        
        # 処理済みファイルをZIPにまとめる
        output_zip = tempfile.NamedTemporaryFile(delete=False, suffix='.zip')
        with zipfile.ZipFile(output_zip.name, 'w', zipfile.ZIP_DEFLATED) as zf:
            for root, dirs, files in os.walk(output_temp_dir):
                for file in files:
                    file_path = os.path.join(root, file)
                    arcname = os.path.relpath(file_path, output_temp_dir)
                    zf.write(file_path, arcname)
        
        # 一時ディレクトリの削除
        shutil.rmtree(output_temp_dir)
        
        return output_zip.name

# Gradioインターフェースの作成
with gr.Blocks() as app:
    gr.Markdown("## 動画リサイズツール")
    gr.Markdown("1. フォルダ内の動画ファイルをZIPにまとめてアップロードしてください")
    gr.Markdown("2. 全ての動画が768x1280から1080x1920にリサイズされます")
    
    with gr.Row():
        input_file = gr.File(label="ZIPファイルをアップロード", file_types=[".zip"])
        output_file = gr.File(label="処理済みZIPファイル")
    
    upload_button = gr.Button("処理開始")
    upload_button.click(fn=process_videos, inputs=[input_file], outputs=[output_file])
    
    gr.Markdown("### 注意事項")
    gr.Markdown("- アップロードするZIPファイルには.mp4ファイルが含まれている必要があります")
    gr.Markdown("- 元のフォルダ構造は維持されます")
    gr.Markdown("- 処理には時間がかかる場合があります")

if __name__ == "__main__":
    app.launch()