Spaces:

Ricky01anjay
/

AutoClip

Paused

App Files Files Community

Ricky01anjay commited on Mar 2

Commit

ffe2671

verified ·

1 Parent(s): b208f0f

Update app.py

Browse files

Files changed (1) hide show

app.py +109 -113

app.py CHANGED Viewed

@@ -11,23 +11,21 @@ from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip, concaten
 import moviepy.video.fx.all as vfx
 from faster_whisper import WhisperModel
-# --- KONFIGURASI ---
 PURU_API_URL = "https://puruboy-api.vercel.app/api/ai/puruai"
 app = Flask(__name__)
 app.config['UPLOAD_FOLDER'] = 'downloads'
-app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024
 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
-# Database & Queue
 JOBS = {}
-QUEUE = deque()
-MAX_QUEUE_SIZE = 5
-# Model Subtitle (Whisper Base)
 whisper_model = WhisperModel("base", device="cpu", compute_type="int8")
 # ==========================================
-# UI HTML
 # ==========================================
 HTML_TEMPLATE = """
 <!DOCTYPE html>
@@ -35,56 +33,59 @@ HTML_TEMPLATE = """
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>AI Video Clipper (Free API)</title>
     <script src="https://cdn.tailwindcss.com"></script>
 </head>
-<body class="bg-slate-950 text-slate-200 min-h-screen flex flex-col items-center p-6 font-sans">
-    <div class="w-full max-w-2xl bg-slate-900 p-8 rounded-3xl border border-slate-800 shadow-2xl">
-        <h1 class="text-3xl font-black text-center mb-1 bg-gradient-to-r from-purple-400 to-pink-400 bg-clip-text text-transparent italic">AI VIRAL CLIPPER</h1>
-        <p class="text-center text-slate-500 text-sm mb-8 font-medium">Powered by PuruBoy AI & Faster Whisper</p>
-        <form id="clipForm" class="space-y-6">
-            <input type="file" id="videoFile" accept="video/mp4" class="w-full text-sm text-slate-400 file:mr-4 file:py-3 file:px-6 file:rounded-xl file:border-0 file:bg-purple-600 file:text-white hover:file:bg-purple-500 transition-all cursor-pointer">
-            <button type="submit" class="w-full bg-purple-600 hover:bg-purple-500 text-white font-black py-4 rounded-xl transition-all transform active:scale-95">PROSES VIDEO ✨</button>
         </form>
-        <div id="statusArea" class="mt-10 hidden">
-            <div class="flex justify-between mb-2">
-                <span id="statusText" class="text-sm font-bold text-purple-400">Menunggu...</span>
-                <span id="percentText" class="text-sm font-black text-white">0%</span>
             </div>
-            <div class="w-full bg-slate-800 rounded-full h-3">
-                <div id="progressBar" class="bg-purple-500 h-3 rounded-full transition-all duration-500" style="width: 0%"></div>
             </div>
         </div>
-        <div id="resultArea" class="mt-10 hidden text-center">
-            <video id="player" controls class="w-full rounded-2xl border border-slate-700 mb-6"></video>
-            <div class="flex gap-3">
-                <a id="downloadBtn" href="#" class="flex-1 bg-white text-black font-black py-3 rounded-xl">DOWNLOAD</a>
-                <button onclick="location.reload()" class="px-6 bg-slate-800 text-white rounded-xl">ULANG</button>
-            </div>
         </div>
     </div>
     <script>
-        const form = document.getElementById('clipForm');
-        form.addEventListener('submit', async (e) => {
             e.preventDefault();
-            const file = document.getElementById('videoFile').files[0];
             if (!file) return alert("Pilih video!");
             const formData = new FormData();
             formData.append('video_file', file);
             document.getElementById('statusArea').classList.remove('hidden');
-            form.classList.add('hidden');
             const res = await fetch('/generate', { method: 'POST', body: formData });
             const data = await res.json();
             if (data.job_id) pollStatus(data.job_id);
-            else alert(data.error);
-        });
         async function pollStatus(jobId) {
             const interval = setInterval(async () => {
@@ -102,7 +103,7 @@ HTML_TEMPLATE = """
                     document.getElementById('resultArea').classList.remove('hidden');
                 } else if (data.status === 'error') {
                     clearInterval(interval);
-                    alert("Error: " + data.message);
                 }
             }, 3000);
         }
@@ -112,108 +113,106 @@ HTML_TEMPLATE = """
 """
 # ==========================================
-# CORE PROCESSOR
 # ==========================================
-def process_video_job(job_id):
-    video_path = JOBS[job_id]['input_path']
     try:
-        # 1. Transkripsi Whisper (Cari Teks & Timestamp)
-        JOBS[job_id].update({"message": "Menganalisis audio (Whisper)...", "progress": 20})
-        segments, _ = whisper_model.transcribe(video_path, word_timestamps=True)
-        full_transcript = []
         all_words = []
         for s in segments:
-            full_transcript.append(f"[{s.start:.2f}s - {s.end:.2f}s]: {s.text}")
-            for w in s.words:
-                all_words.append(w)
-        transcript_text = "\n".join(full_transcript)
-        # 2. Kirim ke PuruBoy AI untuk Seleksi Bagian Viral
-        JOBS[job_id].update({"message": "AI memilih bagian terbaik...", "progress": 45})
-        prompt_ai = f"""
-        Identifikasi bagian paling menarik dari transkrip video ini untuk dijadikan clip pendek (max 60 detik).
-        Berikan output HANYA dalam format JSON array seperti ini:
-        [{"start_s": 10.5, "end_s": 25.0, "reason": "lucu"}]
-        TOTAL DURASI SEMUA CLIP TIDAK BOLEH LEBIH DARI 60 DETIK.
         TRANSKRIP:
-        {transcript_text}
         """
         payload = {
             "userid": job_id,
-            "prompt": prompt_ai,
             "model": "puruboy-flash",
-            "system": "Kamu adalah editor video viral. Kamu hanya merespon dengan format JSON array berisi timestamp start_s dan end_s."
         }
-        response = requests.post(PURU_API_URL, json=payload).json()
-        ai_response_text = response['result'][0]['parts'][0]['text']
-        # Ekstrak JSON dari teks AI
-        json_match = re.search(r'\[.*\]', ai_response_text, re.DOTALL)
-        if not json_match:
-            raise Exception("AI tidak memberikan format timestamp yang benar")
         selected_clips = json.loads(json_match.group(0))
-        # 3. Editing dengan MoviePy
-        JOBS[job_id].update({"message": "Memotong & Menambah Subtitle...", "progress": 70})
-        video = VideoFileClip(video_path)
-        final_clips = []
-        # Path font (sesuaikan dengan OS, atau biarkan default)
-        font_path = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"
-        if not os.path.exists(font_path): font_path = "Arial"
-        for clip_info in selected_clips:
-            start = clip_info['start_s']
-            end = min(clip_info['end_s'], start + 59) # Safety limit per clip
-            subclip = video.subclip(start, end).fx(vfx.fadein, 0.2).fx(vfx.fadeout, 0.2)
-            # Subtitle Generator
-            txt_elements = []
             for wd in all_words:
-                if wd.start >= start and wd.end <= end:
                     t = TextClip(
-                        wd.word.strip().upper(), font=font_path,
-                        fontsize=video.h * 0.07, color='yellow',
-                        method='label', stroke_color='black', stroke_width=2
-                    ).set_start(wd.start - start).set_end(wd.end - start).set_position(('center', video.h * 0.8))
-                    txt_elements.append(t)
-            combined = CompositeVideoClip([subclip] + txt_elements)
-            final_clips.append(combined)
-        if not final_clips:
-            raise Exception("Tidak ada clip yang dihasilkan")
-        # Gabungkan semua clip (Total tetap dibatasi 60s oleh prompt AI)
-        final_video = concatenate_videoclips(final_clips, method="compose")
-        if final_video.duration > 60:
-            final_video = final_video.subclip(0, 60)
-        output_name = f"result_{job_id}.mp4"
-        output_path = os.path.join(app.config['UPLOAD_FOLDER'], output_name)
-        final_video.write_videofile(output_path, fps=24, codec="libx264", audio_codec="aac", preset="ultrafast", logger=None)
-        # Cleanup
-        video.close()
-        final_video.close()
-        os.remove(video_path)
-        JOBS[job_id].update({"status": "completed", "progress": 100, "message": "Selesai!", "result": output_name})
     except Exception as e:
-        print(f"Error: {e}")
         JOBS[job_id].update({"status": "error", "message": str(e)})
 # ==========================================
@@ -226,24 +225,21 @@ def index():
 @app.route('/generate', methods=['POST'])
 def generate():
-    if 'video_file' not in request.files:
-        return jsonify({"error": "No file"}), 400
-    file = request.files['video_file']
     job_id = str(uuid.uuid4())
-    path = os.path.join(app.config['UPLOAD_FOLDER'], f"{job_id}_in.mp4")
-    file.save(path)
-    JOBS[job_id] = {"status": "processing", "progress": 5, "message": "Memulai...", "input_path": path}
-    # Jalankan background task
-    threading.Thread(target=process_video_job, args=(job_id,)).start()
     return jsonify({"job_id": job_id})
 @app.route('/status/<job_id>')
 def status(job_id):
-    return jsonify(JOBS.get(job_id, {"status": "error", "message": "Job not found"}))
 @app.route('/view/<filename>')
 def view_video(filename):
@@ -254,4 +250,4 @@ def download(filename):
     return send_file(os.path.join(app.config['UPLOAD_FOLDER'], filename), as_attachment=True)
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)

 import moviepy.video.fx.all as vfx
 from faster_whisper import WhisperModel
+# --- CONFIG ---
 PURU_API_URL = "https://puruboy-api.vercel.app/api/ai/puruai"
 app = Flask(__name__)
 app.config['UPLOAD_FOLDER'] = 'downloads'
+app.config['MAX_CONTENT_LENGTH'] = 200 * 1024 * 1024
 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+# Database Sederhana
 JOBS = {}
+# Load Model Whisper (Gunakan CPU secara default)
 whisper_model = WhisperModel("base", device="cpu", compute_type="int8")
 # ==========================================
+# UI HTML (Dark Mode Aesthetic)
 # ==========================================
 HTML_TEMPLATE = """
 <!DOCTYPE html>
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI Viral Clipper</title>
     <script src="https://cdn.tailwindcss.com"></script>
 </head>
+<body class="bg-black text-slate-200 min-h-screen flex flex-col items-center p-6">
+    <div class="w-full max-w-xl bg-slate-900 p-8 rounded-3xl border border-slate-800 shadow-2xl">
+        <h1 class="text-2xl font-black text-center mb-6 bg-gradient-to-r from-blue-400 to-purple-500 bg-clip-text text-transparent">AI VIDEO CLIPPER</h1>
+        <form id="clipForm" class="space-y-4">
+            <div class="border-2 border-dashed border-slate-700 rounded-2xl p-6 text-center hover:border-blue-500 transition-colors">
+                <input type="file" id="videoFile" accept="video/mp4" class="hidden">
+                <label for="videoFile" class="cursor-pointer block">
+                    <span class="text-sm text-slate-400" id="fileName">Klik untuk upload MP4 (Maks 6 Menit)</span>
+                </label>
+            </div>
+            <button type="submit" class="w-full bg-blue-600 hover:bg-blue-500 text-white font-bold py-4 rounded-xl transition-all shadow-lg">BUAT VIDEO VIRAL ✨</button>
         </form>
+        <div id="statusArea" class="mt-8 hidden">
+            <div class="flex justify-between mb-2 italic">
+                <span id="statusText" class="text-xs text-blue-400">Sedang memproses...</span>
+                <span id="percentText" class="text-xs text-white">0%</span>
             </div>
+            <div class="w-full bg-slate-800 rounded-full h-2">
+                <div id="progressBar" class="bg-blue-500 h-2 rounded-full transition-all duration-500" style="width: 0%"></div>
             </div>
         </div>
+        <div id="resultArea" class="mt-8 hidden">
+            <video id="player" controls class="w-full rounded-xl border border-slate-700 mb-4"></video>
+            <a id="downloadBtn" href="#" class="block w-full text-center bg-white text-black font-bold py-3 rounded-xl">DOWNLOAD CLIP</a>
         </div>
     </div>
     <script>
+        const fileInput = document.getElementById('videoFile');
+        fileInput.onchange = () => { document.getElementById('fileName').innerText = fileInput.files[0].name; };
+        document.getElementById('clipForm').onsubmit = async (e) => {
             e.preventDefault();
+            const file = fileInput.files[0];
             if (!file) return alert("Pilih video!");
             const formData = new FormData();
             formData.append('video_file', file);
             document.getElementById('statusArea').classList.remove('hidden');
+            document.getElementById('clipForm').classList.add('hidden');
             const res = await fetch('/generate', { method: 'POST', body: formData });
             const data = await res.json();
             if (data.job_id) pollStatus(data.job_id);
+            else alert("Error: " + data.error);
+        };
         async function pollStatus(jobId) {
             const interval = setInterval(async () => {
                     document.getElementById('resultArea').classList.remove('hidden');
                 } else if (data.status === 'error') {
                     clearInterval(interval);
+                    alert("Gagal: " + data.message);
                 }
             }, 3000);
         }
 """
 # ==========================================
+# LOGIC PEMROSESAN
 # ==========================================
+def process_video(job_id):
+    path_in = JOBS[job_id]['input_path']
     try:
+        # 1. WHISPER TRANSCRIPTION
+        JOBS[job_id].update({"message": "Menyalin teks dari video...", "progress": 20})
+        segments, _ = whisper_model.transcribe(path_in, word_timestamps=True)
         all_words = []
+        transcript_for_ai = []
         for s in segments:
+            transcript_for_ai.append(f"[{s.start:.1f}s - {s.end:.1f}s]: {s.text}")
+            for w in s.words: all_words.append(w)
+        transcript_str = "\n".join(transcript_for_ai)
+        # 2. AI SELECTION (FIXED F-STRING ERROR)
+        JOBS[job_id].update({"message": "AI sedang memilih bagian viral...", "progress": 45})
+        # Perbaikan: Menggunakan {{ }} agar Python f-string tidak error membaca kurung kurawal JSON
+        prompt = f"""
+        Tugas: Pilih bagian paling menarik dari transkrip video ini untuk dijadikan clip pendek.
+        Aturan:
+        1. Total durasi semua clip harus di bawah 60 detik.
+        2. Format output WAJIB JSON array seperti contoh: [{{"start_s": 1.5, "end_s": 15.0, "reason": "highlight"}}]
+        3. Jangan ada teks penjelasan lain, hanya JSON.
         TRANSKRIP:
+        {transcript_str}
         """
         payload = {
             "userid": job_id,
+            "prompt": prompt,
             "model": "puruboy-flash",
+            "system": "Kamu adalah asisten video editor profesional. Kamu hanya memberikan output dalam format JSON array."
         }
+        resp = requests.post(PURU_API_URL, json=payload).json()
+        ai_text = resp['result'][0]['parts'][0]['text']
+        # Ekstrak JSON
+        json_match = re.search(r'\[.*\]', ai_text, re.DOTALL)
+        if not json_match: raise Exception("AI tidak memberikan data timestamp yang valid.")
         selected_clips = json.loads(json_match.group(0))
+        # 3. EDITING VIDEO
+        JOBS[job_id].update({"message": "Proses rendering & subtitle...", "progress": 70})
+        full_video = VideoFileClip(path_in)
+        final_segments = []
+        # Font settings
+        font_name = "Arial-Bold" # Atau path ke .ttf jika di Linux server
+        for clip_data in selected_clips:
+            s_time = clip_data['start_s']
+            e_time = min(clip_data['end_s'], s_time + 59)
+            # Potong clip
+            sub = full_video.subclip(s_time, e_time).fx(vfx.fadein, 0.2).fx(vfx.fadeout, 0.2)
+            # Buat subtitle per kata (Auto Highlight)
+            txt_clips = []
             for wd in all_words:
+                if wd.start >= s_time and wd.end <= e_time:
+                    # Render teks per kata
                     t = TextClip(
+                        wd.word.strip().upper(),
+                        fontsize=full_video.h * 0.08,
+                        color='yellow',
+                        stroke_color='black',
+                        stroke_width=2,
+                        method='label'
+                    ).set_start(wd.start - s_time).set_end(wd.end - s_time).set_position(('center', full_video.h * 0.8))
+                    txt_clips.append(t)
+            final_segments.append(CompositeVideoClip([sub] + txt_clips))
+        # Gabungkan semua
+        if not final_segments: raise Exception("AI tidak memilih bagian apapun.")
+        merged = concatenate_videoclips(final_segments, method="compose")
+        if merged.duration > 60: merged = merged.subclip(0, 60) # Limit keras 60 detik
+        out_filename = f"out_{job_id}.mp4"
+        out_path = os.path.join(app.config['UPLOAD_FOLDER'], out_filename)
+        merged.write_videofile(out_path, fps=24, codec="libx264", audio_codec="aac", preset="ultrafast", logger=None)
+        # Tutup file agar tidak memori leak
+        full_video.close()
+        merged.close()
+        JOBS[job_id].update({"status": "completed", "progress": 100, "message": "Selesai!", "result": out_filename})
     except Exception as e:
+        print(f"ERROR: {str(e)}")
         JOBS[job_id].update({"status": "error", "message": str(e)})
 # ==========================================
 @app.route('/generate', methods=['POST'])
 def generate():
+    file = request.files.get('video_file')
+    if not file: return jsonify({"error": "File kosong"}), 400
     job_id = str(uuid.uuid4())
+    save_path = os.path.join(app.config['UPLOAD_FOLDER'], f"{job_id}_in.mp4")
+    file.save(save_path)
+    JOBS[job_id] = {"status": "processing", "progress": 5, "message": "Video diterima...", "input_path": save_path}
+    threading.Thread(target=process_video, args=(job_id,)).start()
     return jsonify({"job_id": job_id})
 @app.route('/status/<job_id>')
 def status(job_id):
+    return jsonify(JOBS.get(job_id, {"status": "error", "message": "ID tidak ditemukan"}))
 @app.route('/view/<filename>')
 def view_video(filename):
     return send_file(os.path.join(app.config['UPLOAD_FOLDER'], filename), as_attachment=True)
 if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=False)