Update app.py

app.py

@@ -3,14 +3,12 @@ import uuid
 import requests
 import shutil
 import subprocess
-import traceback
-import cv2  # Tambahkan import cv2
+import cv2
 from fastapi import FastAPI, UploadFile, File, Form, HTTPException
 from fastapi.responses import HTMLResponse, FileResponse
 from vidstab import VidStab
 
 # --- FIX UNTUK ERROR HEADLESS OPENCV ---
-# Mencegah vidstab memanggil fungsi GUI yang menyebabkan crash di server (cvDestroyAllWindows)
 cv2.destroyAllWindows = lambda: None
 cv2.imshow = lambda *args, **kwargs: None
 cv2.waitKey = lambda *args, **kwargs: 1
@@ -31,7 +29,7 @@ async def index():
     <head>
         <meta charset="UTF-8">
         <meta name="viewport" content="width=device-width, initial-scale=1.0">
-        <title>AI Video Stabilizer</title>
+        <title>AI Video Stabilizer Pro</title>
         <script src="https://cdn.tailwindcss.com"></script>
         <style>
             .loader { border-top-color: #4f46e5; animation: spinner 1.5s linear infinite; }
@@ -40,8 +38,8 @@ async def index():
     </head>
     <body class="bg-slate-50 min-h-screen flex flex-col items-center p-4 font-sans">
         <div class="w-full max-w-md bg-white rounded-3xl shadow-2xl p-8 mt-10 border border-slate-100">
-            <h1 class="text-3xl font-black text-center text-indigo-600 mb-2">STABILIZER</h1>
-            <p class="text-center text-slate-400 text-sm mb-8">AI Video Anti-Shake</p>
+            <h1 class="text-3xl font-black text-center text-indigo-600 mb-2">STABILIZER <span class="text-emerald-500">PRO</span></h1>
+            <p class="text-center text-slate-400 text-sm mb-8">Anti-Shake + Audio Preserved + Border Fill</p>
             
             <form id="uploadForm" class="space-y-6">
                 <div class="group relative border-2 border-dashed border-slate-200 rounded-2xl p-8 text-center hover:border-indigo-400 transition-colors">
@@ -71,7 +69,10 @@ async def index():
             <div id="status" class="mt-8 hidden">
                 <div class="flex flex-col items-center justify-center space-y-4">
                     <div class="loader ease-linear rounded-full border-4 border-t-4 border-slate-200 h-12 w-12"></div>
-                    <p class="text-indigo-600 font-bold text-sm animate-pulse text-center">AI sedang bekerja...<br><span class="text-[10px] text-slate-400">Menganalisis guncangan & re-encoding</span></p>
+                    <p class="text-indigo-600 font-bold text-sm animate-pulse text-center">
+                        AI sedang bekerja...<br>
+                        <span class="text-[10px] text-slate-400">Menghilangkan guncangan, mengisi border & menyatukan audio</span>
+                    </p>
                 </div>
             </div>
 
@@ -137,14 +138,11 @@ async def index():
 async def stabilize_video(file: UploadFile = File(None), url: str = Form(None)):
     job_id = str(uuid.uuid4())
     input_path = os.path.join(UPLOAD_DIR, f"{job_id}_in.mp4")
-    
-    # FIX: Ubah ekstensi raw dari .mp4 ke .avi. 
-    # Karena OpenCV menggunakan MJPG secara default, format ini sangat kompatibel dengan kontainer .avi
     raw_output_path = os.path.join(OUTPUT_DIR, f"{job_id}_raw.avi")
     final_output_path = os.path.join(OUTPUT_DIR, f"{job_id}_final.mp4")
 
     try:
-        # 1. Simpan Input
+        # 1. Simpan Input Video
         if file:
             with open(input_path, "wb") as buffer:
                 shutil.copyfileobj(file.file, buffer)
@@ -158,16 +156,37 @@ async def stabilize_video(file: UploadFile = File(None), url: str = Form(None)):
 
         # 2. Proses Stabilisasi (VidStab)
         stabilizer = VidStab()
-        # playback=False ditambahkan untuk memastikan tidak memanggil window GUI
-        stabilizer.stabilize(input_path=input_path, output_path=raw_output_path, playback=False)
-
-        # 3. Konversi dari AVI ke H.264 MP4 (agar bisa diputar di Browser/Mobile)
+        
+        # PERBAIKAN KUALITAS: 
+        # - smoothing_window=30 : Membuat gerakan kamera jauh lebih mulus (seperti pakai gimbal).
+        # - border_type='replicate' : Menghilangkan kotak/border hitam dengan cara mereplikasi pixel terluar.
+        stabilizer.stabilize(
+            input_path=input_path, 
+            output_path=raw_output_path, 
+            smoothing_window=30,
+            border_type='replicate', 
+            playback=False
+        )
+
+        # 3. Konversi + GABUNGKAN AUDIO (Audio Muxing)
+        # Mengambil Video dari file .avi (hasil stabil) dan Audio dari file .mp4 (video asli)
         ffmpeg_cmd = [
-            'ffmpeg', '-y', '-i', raw_output_path,
-            '-c:v', 'libx264', '-preset', 'ultrafast', '-crf', '25',
-            '-pix_fmt', 'yuv420p', '-c:a', 'aac', '-movflags', '+faststart',
+            'ffmpeg', '-y', 
+            '-i', raw_output_path,      # Input 0: Video Stabil (Tidak bersuara)
+            '-i', input_path,           # Input 1: Video Asli (Bersuara)
+            '-map', '0:v:0',            # Ambil track Video dari Input 0
+            '-map', '1:a:0?',           # Ambil track Audio dari Input 1 (tanda '?' = abaikan jika video asli memang bisu)
+            '-c:v', 'libx264',          # Codec Video
+            '-preset', 'fast',          # Kecepatan render sedikit lebih baik
+            '-crf', '23',               # Kualitas visual ditingkatkan (semakin kecil = semakin jernih)
+            '-pix_fmt', 'yuv420p',
+            '-c:a', 'aac',              # Codec Audio
+            '-b:a', '128k',             # Bitrate Audio
+            '-movflags', '+faststart',
             final_output_path
         ]
+        
+        # Jalankan FFmpeg
         subprocess.run(ffmpeg_cmd, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         
         # 4. Cleanup file sementara