Create app.py

app.py

@@ -0,0 +1,358 @@
+import os
+import uuid
+import threading
+import asyncio
+import requests
+import json
+import time
+from flask import Flask, request, jsonify, render_template_string, send_from_directory
+import moviepy.editor as mp
+import whisper
+import edge_tts
+
+app = Flask(__name__)
+app.config['UPLOAD_FOLDER'] = 'uploads'
+os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+
+tasks = {}
+
+# Konfigurasi Suara Edge-TTS
+VOICE_MAP = {
+    'id-ID': 'id-ID-ArdiNeural', 
+    'en-US': 'en-US-ChristopherNeural', 
+    'ja-JP': 'ja-JP-KeitaNeural'
+}
+
+print("Loading Whisper Model...")
+whisper_model = whisper.load_model("base")
+print("Whisper Model Loaded!")
+
+# ==========================================
+# FUNGSI TRANSLATOR LLM (DENGAN BACKOFF)
+# ==========================================
+def translate_with_llm(text, custom_prompt, max_retries=3):
+    # Gabungkan instruksi user dengan teks asli
+    instruction = custom_prompt if custom_prompt else "Terjemahkan teks berikut. Hanya berikan hasil terjemahannya saja tanpa penjelasan tambahan, tanpa tanda kutip."
+    full_prompt = f"{instruction}\n\nTeks asli:\n{text}"
+
+    url = "https://www.puruboy.kozow.com/api/ai/notegpt"
+    payload = {
+        "prompt": full_prompt,
+        "model": "gemini-3-flash-preview",
+        "chat_mode": "standard"
+    }
+    headers = {"Content-Type": "application/json"}
+
+    for attempt in range(max_retries):
+        try:
+            # Menggunakan stream=True karena response berupa SSE
+            response = requests.post(url, json=payload, headers=headers, stream=True)
+            response.raise_for_status()
+
+            translated_text = ""
+            for line in response.iter_lines():
+                if line:
+                    decoded_line = line.decode('utf-8')
+                    if decoded_line.startswith("data: "):
+                        json_str = decoded_line[len("data: "):]
+                        try:
+                            data = json.loads(json_str)
+                            # Ambil teks jika ada
+                            if "text" in data:
+                                translated_text += data["text"]
+                        except json.JSONDecodeError:
+                            continue # Abaikan jika bukan JSON valid (misal: finish)
+
+            translated_text = translated_text.strip()
+            
+            # Jika respon tidak kosong, kembalikan hasilnya
+            if translated_text:
+                return translated_text
+            
+            print(f"Respon AI kosong. Mencoba lagi... (Percobaan ke-{attempt+1})")
+
+        except Exception as e:
+            print(f"Error koneksi ke LLM: {str(e)}")
+
+        # Exponential backoff: tunggu 1s, 2s, 4s...
+        time.sleep(2 ** attempt)
+
+    raise Exception("Gagal mendapatkan terjemahan dari AI setelah beberapa kali percobaan (Respon selalu kosong).")
+
+
+# ==========================================
+# FUNGSI PEMROSESAN UTAMA (BACKGROUND)
+# ==========================================
+def process_dubbing(task_id, video_path, target_voice, custom_prompt):
+    try:
+        tasks[task_id]['status'] = 'Mengekstrak audio dari video...'
+        
+        # 1. Pengecekan Durasi & Ekstrak Audio
+        video = mp.VideoFileClip(video_path)
+        if video.duration > 120.0:
+            video.close()
+            os.remove(video_path)
+            raise Exception("Durasi video melebihi batas maksimal 2 menit (120 detik).")
+
+        audio_path = os.path.join(app.config['UPLOAD_FOLDER'], f"{task_id}.wav")
+        if video.audio is None:
+            raise Exception("Video tidak memiliki suara.")
+        video.audio.write_audiofile(audio_path, logger=None)
+        
+        # 2. Transkripsi Audio ke Teks (Speech-to-Text)
+        tasks[task_id]['status'] = 'Mentranskripsi suara asli...'
+        result = whisper_model.transcribe(audio_path)
+        original_text = result['text']
+        
+        if not original_text.strip():
+            raise Exception("Tidak ada kata-kata yang terdeteksi di dalam video.")
+
+        # 3. Terjemahkan Teks via Custom LLM
+        tasks[task_id]['status'] = 'Menerjemahkan teks (Menggunakan AI)...'
+        translated_text = translate_with_llm(original_text, custom_prompt)
+
+        # 4. Text-to-Speech (AI Voice generation)
+        tasks[task_id]['status'] = 'Membuat Suara AI (Dubbing)...'
+        ai_audio_path = os.path.join(app.config['UPLOAD_FOLDER'], f"{task_id}_ai.mp3")
+        voice = VOICE_MAP.get(target_voice, 'id-ID-ArdiNeural')
+        
+        async def generate_tts():
+            communicate = edge_tts.Communicate(translated_text, voice)
+            await communicate.save(ai_audio_path)
+        asyncio.run(generate_tts())
+
+        # 5. Gabungkan Audio AI ke Video Asli
+        tasks[task_id]['status'] = 'Menggabungkan Video dan Suara AI...'
+        output_video_path = os.path.join(app.config['UPLOAD_FOLDER'], f"{task_id}_output.mp4")
+        
+        new_audio = mp.AudioFileClip(ai_audio_path)
+        final_video = video.set_audio(new_audio)
+        final_video.write_videofile(output_video_path, codec='libx264', audio_codec='aac', logger=None)
+        
+        # Bersihkan memori dan file temp
+        video.close()
+        new_audio.close()
+        os.remove(audio_path)
+        os.remove(ai_audio_path)
+
+        tasks[task_id]['status'] = 'Selesai'
+        tasks[task_id]['result_video'] = f"/download/{task_id}_output.mp4"
+
+    except Exception as e:
+        tasks[task_id]['status'] = 'Error'
+        tasks[task_id]['error_message'] = str(e)
+
+
+# ==========================================
+# FRONTEND HTML (Bootstrap 5)
+# ==========================================
+HTML_TEMPLATE = """
+<!DOCTYPE html>
+<html lang="id">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI Video Dubbing</title>
+    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/css/bootstrap.min.css" rel="stylesheet">
+    <style>
+        body { background-color: #f8f9fa; }
+        .card { border-radius: 15px; box-shadow: 0 4px 6px rgba(0,0,0,0.1); }
+    </style>
+</head>
+<body>
+    <div class="container mt-4 mb-5">
+        <div class="row justify-content-center">
+            <div class="col-md-8 col-lg-6">
+                <div class="card p-4">
+                    <h3 class="text-center mb-3">🎙️ AI Video Dubbing</h3>
+                    <div class="alert alert-info text-center small">Maximal durasi video: 2 Menit</div>
+                    
+                    <form id="uploadForm">
+                        <div class="mb-3">
+                            <label class="form-label fw-bold">Unggah Video (.mp4)</label>
+                            <input class="form-control" type="file" id="videoFile" accept="video/mp4,video/x-m4v,video/*" required>
+                        </div>
+                        
+                        <div class="mb-3">
+                            <label class="form-label fw-bold">Karakter Suara AI (Output)</label>
+                            <select class="form-select" id="targetVoice" required>
+                                <option value="id-ID">Pria Indonesia (id-ID)</option>
+                                <option value="en-US">Pria Inggris (en-US)</option>
+                                <option value="ja-JP">Pria Jepang (ja-JP)</option>
+                            </select>
+                        </div>
+
+                        <div class="mb-3">
+                            <label class="form-label fw-bold">Prompt Terjemahan (Opsional)</label>
+                            <textarea class="form-control" id="customPrompt" rows="2" placeholder="Contoh: Terjemahkan ke bahasa Indonesia gaul dan santai"></textarea>
+                            <div class="form-text">Biarkan kosong untuk terjemahan baku.</div>
+                        </div>
+
+                        <button type="submit" class="btn btn-primary w-100 fw-bold" id="submitBtn">Generate Dubbing</button>
+                    </form>
+
+                    <div id="statusArea" class="mt-4 d-none text-center">
+                        <h5 id="statusText" class="text-primary fw-bold">Memproses...</h5>
+                        <div class="spinner-border text-primary my-2" role="status" id="loadingSpinner"></div>
+                    </div>
+
+                    <div id="resultArea" class="mt-4 d-none">
+                        <h5 class="text-success text-center fw-bold">Berhasil!</h5>
+                        <video id="resultVideo" controls class="w-100 rounded mt-2 border"></video>
+                        <a id="downloadBtn" href="#" class="btn btn-success w-100 mt-3 fw-bold" download>⬇️ Unduh Video</a>
+                    </div>
+                </div>
+            </div>
+        </div>
+    </div>
+
+    <script>
+        document.getElementById('uploadForm').addEventListener('submit', async (e) => {
+            e.preventDefault();
+            
+            const fileInput = document.getElementById('videoFile');
+            const voiceInput = document.getElementById('targetVoice');
+            const promptInput = document.getElementById('customPrompt');
+            const submitBtn = document.getElementById('submitBtn');
+            const statusArea = document.getElementById('statusArea');
+            const resultArea = document.getElementById('resultArea');
+            const statusText = document.getElementById('statusText');
+            const loadingSpinner = document.getElementById('loadingSpinner');
+
+            if (fileInput.files.length === 0) return;
+
+            const file = fileInput.files[0];
+
+            // Validasi Durasi di Frontend (Max 120 Detik)
+            const videoElement = document.createElement('video');
+            videoElement.preload = 'metadata';
+            videoElement.onloadedmetadata = async function() {
+                window.URL.revokeObjectURL(videoElement.src);
+                if (videoElement.duration > 120) {
+                    alert('Error: Durasi video tidak boleh lebih dari 2 menit.');
+                    return;
+                }
+                
+                // Lanjut Upload jika durasi aman
+                mulaiProsesUpload();
+            }
+            videoElement.src = URL.createObjectURL(file);
+
+            async function mulaiProsesUpload() {
+                const formData = new FormData();
+                formData.append('video', file);
+                formData.append('voice', voiceInput.value);
+                formData.append('prompt', promptInput.value);
+
+                submitBtn.disabled = true;
+                statusArea.classList.remove('d-none');
+                resultArea.classList.add('d-none');
+                loadingSpinner.classList.remove('d-none');
+                statusText.innerText = 'Mengunggah video...';
+                statusText.className = 'text-primary fw-bold';
+
+                try {
+                    const response = await fetch('/generate', { method: 'POST', body: formData });
+                    const data = await response.json();
+                    
+                    if (response.ok && data.task_id) {
+                        pollStatus(data.task_id);
+                    } else {
+                        alert(data.error || 'Gagal memulai proses.');
+                        resetUI();
+                    }
+                } catch (err) {
+                    alert('Terjadi kesalahan jaringan saat mengunggah.');
+                    resetUI();
+                }
+            }
+
+            function resetUI() {
+                submitBtn.disabled = false;
+                statusArea.classList.add('d-none');
+            }
+        });
+
+        async function pollStatus(taskId) {
+            const statusText = document.getElementById('statusText');
+            const interval = setInterval(async () => {
+                try {
+                    const res = await fetch(`/status?task_id=${taskId}`);
+                    const data = await res.json();
+
+                    statusText.innerText = data.status;
+
+                    if (data.status === 'Selesai') {
+                        clearInterval(interval);
+                        document.getElementById('loadingSpinner').classList.add('d-none');
+                        document.getElementById('resultArea').classList.remove('d-none');
+                        document.getElementById('resultVideo').src = data.result_video;
+                        document.getElementById('downloadBtn').href = data.result_video;
+                        document.getElementById('submitBtn').disabled = false;
+                    } else if (data.status === 'Error') {
+                        clearInterval(interval);
+                        statusText.innerText = 'Error: ' + data.error_message;
+                        statusText.className = 'text-danger fw-bold';
+                        document.getElementById('loadingSpinner').classList.add('d-none');
+                        document.getElementById('submitBtn').disabled = false;
+                    }
+                } catch (e) {
+                    console.error("Polling error", e);
+                }
+            }, 2000); // Cek status tiap 2 detik
+        }
+    </script>
+</body>
+</html>
+"""
+
+# ==========================================
+# ROUTING FLASK
+# ==========================================
+
+@app.route('/')
+def index():
+    return render_template_string(HTML_TEMPLATE)
+
+@app.route('/generate', methods=['POST'])
+def generate():
+    if 'video' not in request.files:
+        return jsonify({'error': 'Tidak ada video yang diunggah'}), 400
+    
+    file = request.files['video']
+    target_voice = request.form.get('voice', 'id-ID')
+    custom_prompt = request.form.get('prompt', '').strip()
+
+    if file.filename == '':
+        return jsonify({'error': 'File kosong'}), 400
+
+    task_id = str(uuid.uuid4())
+    video_path = os.path.join(app.config['UPLOAD_FOLDER'], f"{task_id}.mp4")
+    file.save(video_path)
+
+    tasks[task_id] = {
+        'status': 'Menunggu antrean...',
+        'result_video': None,
+        'error_message': None
+    }
+
+    # Jalankan proses background
+    thread = threading.Thread(target=process_dubbing, args=(task_id, video_path, target_voice, custom_prompt))
+    thread.start()
+
+    return jsonify({'task_id': task_id}), 200
+
+@app.route('/status', methods=['GET'])
+def status():
+    task_id = request.args.get('task_id')
+    if task_id in tasks:
+        return jsonify(tasks[task_id])
+    return jsonify({'status': 'Error', 'error_message': 'Task ID tidak ditemukan'}), 404
+
+@app.route('/download/<filename>', methods=['GET'])
+def download(filename):
+    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
+
+if __name__ == '__main__':
+    # Jalankan di Port 7860
+    app.run(host='0.0.0.0', port=7860, debug=False)