Create app.py

app.py

@@ -0,0 +1,139 @@
+import ffmpeg
+import os
+import tempfile
+import shutil
+import re
+from PIL import Image
+from fastapi import FastAPI, File, Form, UploadFile, HTTPException
+from fastapi.responses import FileResponse
+from fastapi.middleware.cors import CORSMiddleware
+
+app = FastAPI(title="Video Conversion API")
+
+# CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Supported formats
+supported_formats = [f for f in sorted([
+    'AVI','FLV','M2TS','M4V','MKV','MOV','MP4','MPEG','MPG','MTS','TS','VOB','WEBM','WMV'
+])]
+audio_formats = [f for f in sorted([
+    'MP3','WAV','AAC','FLAC','OGG','M4A','ALAC','WMA','AIFF','OPUS','CAF','SPX','WV'
+])]
+image_formats = [f for f in sorted(Image.SAVE.keys() or ['BMP','GIF','JPEG','PNG','TIFF','WEBP'])]
+gif_formats = ['GIF']
+
+CACHE_DIR = tempfile.mkdtemp()
+
+# Audio & Video codecs
+AUDIO_CODECS = {
+    'MP3': {'acodec':'libmp3lame','audio_bitrate':'192k'},
+    'AAC': {'acodec':'aac','audio_bitrate':'192k'},
+    'WAV': {'acodec':'pcm_s16le'},
+    'FLAC': {'acodec':'flac'},
+    'OGG': {'acodec':'libvorbis','audio_bitrate':'192k'},
+    'M4A': {'acodec':'aac','audio_bitrate':'192k'},
+    'ALAC': {'acodec':'alac'},
+    'WMA': {'acodec':'wmav2','audio_bitrate':'192k'},
+    'AIFF': {'acodec':'pcm_s16be'},
+    'OPUS': {'acodec':'libopus','audio_bitrate':'128k'},
+    'CAF': {'acodec':'alac'},
+    'SPX': {'acodec':'libspeex','audio_bitrate':'32k'},
+    'WV': {'acodec':'wavpack'}
+}
+
+VIDEO_CODECS = {
+    'MP4': {'vcodec':'libx264','acodec':'aac','preset':'medium','crf':'23'},
+    'AVI': {'vcodec':'libxvid','acodec':'libmp3lame'},
+    'MOV': {'vcodec':'libx264','acodec':'aac'},
+    'MKV': {'vcodec':'libx264','acodec':'aac'},
+    'WEBM': {'vcodec':'libvpx-vp9','acodec':'libopus'},
+    'FLV': {'vcodec':'libx264','acodec':'aac'},
+    'WMV': {'vcodec':'wmv2','acodec':'wmav2'},
+    'M4V': {'vcodec':'libx264','acodec':'aac'},
+    'MPG': {'vcodec':'mpeg2video','acodec':'mp2'},
+    'MPEG': {'vcodec':'mpeg2video','acodec':'mp2'},
+    'VOB': {'vcodec':'mpeg2video','acodec':'mp2'},
+    'ASF': {'vcodec':'wmv2','acodec':'wmav2'},
+    'TS': {'vcodec':'libx264','acodec':'aac'},
+    'M2TS': {'vcodec':'libx264','acodec':'aac'},
+    'MTS': {'vcodec':'libx264','acodec':'aac'}
+}
+
+def sanitize_filename(filename):
+    return re.sub(r'[^a-zA-Z0-9_.-]', '_', filename)
+
+def convert_video(video_path, target_format, conversion_type, time_in_seconds=None):
+    base_name = os.path.splitext(os.path.basename(video_path))[0]
+    sanitized_base_name = sanitize_filename(base_name)
+    output_dir = os.path.join(CACHE_DIR, "outputs")
+    os.makedirs(output_dir, exist_ok=True)
+
+    if conversion_type == 'Video to Video':
+        output_file = os.path.join(output_dir, f"converted_{sanitized_base_name}.{target_format.lower()}")
+        codec_settings = VIDEO_CODECS.get(target_format.upper(), {})
+        input_stream = ffmpeg.input(video_path)
+        ffmpeg.output(input_stream, output_file, **codec_settings).overwrite_output().run(quiet=True)
+        return output_file
+
+    elif conversion_type == 'Video to Audio':
+        output_file = os.path.join(output_dir, f"audio_{sanitized_base_name}.{target_format.lower()}")
+        codec_settings = AUDIO_CODECS.get(target_format.upper(), {})
+        input_stream = ffmpeg.input(video_path)
+        ffmpeg.output(input_stream, output_file, vn=None, **codec_settings).overwrite_output().run(quiet=True)
+        return output_file
+
+    elif conversion_type == 'Video to GIF':
+        output_file = os.path.join(output_dir, f"gif_{sanitized_base_name}.gif")
+        ffmpeg.input(video_path).output(
+            output_file,
+            vf="fps=15,scale=480:-1:flags=lanczos,palettegen=stats_mode=diff",
+            loop=0
+        ).overwrite_output().run(quiet=True)
+        return output_file
+
+    elif conversion_type == 'Video to Image':
+        if time_in_seconds is None:
+            time_in_seconds = 0
+        temp_png_file = os.path.join(output_dir, f"temp_{sanitized_base_name}_{time_in_seconds}s.png")
+        ffmpeg.input(video_path, ss=time_in_seconds).output(temp_png_file, vframes=1, **{'qscale:v': '1'}).overwrite_output().run(quiet=True)
+        if target_format.upper() == 'PNG':
+            return temp_png_file
+        output_file = os.path.join(output_dir, f"image_{sanitized_base_name}_{time_in_seconds}s.{target_format.lower()}")
+        with Image.open(temp_png_file) as img:
+            img.convert('RGB').save(output_file, format=target_format.upper())
+        os.remove(temp_png_file)
+        return output_file
+
+@app.post("/api/convert")
+async def api_convert(file: UploadFile = File(...), conversion_type: str = Form("Video to Video"), target_format: str = Form("MP4"), time_in_seconds: int = Form(0)):
+    if not file.filename:
+        raise HTTPException(status_code=400, detail="No file provided")
+    temp_dir = tempfile.mkdtemp()
+    file_path = os.path.join(temp_dir, file.filename)
+    with open(file_path, "wb") as f:
+        f.write(await file.read())
+    output_file = convert_video(file_path, target_format, conversion_type, time_in_seconds)
+    shutil.rmtree(temp_dir, ignore_errors=True)
+    return FileResponse(output_file, filename=os.path.basename(output_file))
+
+@app.get("/api/formats")
+async def get_formats():
+    return {
+        "video_formats": supported_formats,
+        "audio_formats": audio_formats,
+        "image_formats": image_formats,
+        "gif_formats": gif_formats,
+        "supported_audio_codecs": list(AUDIO_CODECS.keys()),
+        "supported_video_codecs": list(VIDEO_CODECS.keys())
+    }
+
+@app.get("/")
+async def root():
+    return {"message": "Video Conversion API is running", "docs": "/docs"}