Spaces:

Ricky01anjay
/

AutoClip

Paused

App Files Files Community

Ricky01anjay commited on Mar 1

Commit

d26d714

verified ·

1 Parent(s): db72ff2

Create app.py

Browse files

Files changed (1) hide show

app.py +331 -0

app.py ADDED Viewed

	@@ -0,0 +1,331 @@

+import os
+import uuid
+import threading
+import requests
+import re
+import time
+from flask import Flask, request, jsonify, render_template_string, send_file
+from werkzeug.utils import secure_filename
+from faster_whisper import WhisperModel
+from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
+app = Flask(__name__)
+app.config['UPLOAD_FOLDER'] = 'downloads'
+app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024 # Max upload 500MB
+os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+# In-memory database
+JOBS = {}
+JOB_TTL = 3600 # Waktu hidup file: 3600 detik (1 Jam)
+# Model AI Whisper (Dioptimalkan untuk CPU HuggingFace)
+whisper_model = WhisperModel("base", device="cpu", compute_type="int8")
+# ==========================================
+# AUTO CLEANUP (Pembersih Sampah agar HuggingFace Tidak Penuh)
+# ==========================================
+def cleanup_old_jobs():
+    while True:
+        time.sleep(600) # Cek setiap 10 Menit
+        now = time.time()
+        keys_to_delete = []
+        for job_id, job in JOBS.items():
+            if now - job['created_at'] > JOB_TTL:
+                print(f"[CLEANUP] Menghapus Job kedaluwarsa: {job_id}")
+                # Hapus file fisik dari SSD Container
+                for file_path in job.get('files_to_delete', []):
+                    if file_path and os.path.exists(file_path):
+                        try:
+                            os.remove(file_path)
+                            print(f"[CLEANUP] File dihapus: {file_path}")
+                        except Exception as e:
+                            print(f"[CLEANUP] Gagal hapus file: {e}")
+                keys_to_delete.append(job_id)
+        # Hapus data dari RAM
+        for key in keys_to_delete:
+            del JOBS[key]
+# Mulai thread pembersih otomatis
+cleanup_thread = threading.Thread(target=cleanup_old_jobs, daemon=True)
+cleanup_thread.start()
+# ==========================================
+# UI HTML (Tailwind & AJAX Polling)
+# ==========================================
+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 Auto Clipper Video</title>
+    <script src="https://cdn.tailwindcss.com"></script>
+</head>
+<body class="bg-gray-900 text-white min-h-screen p-4 flex flex-col items-center">
+    <div class="w-full max-w-md bg-gray-800 p-6 rounded-xl shadow-lg mt-10">
+        <h1 class="text-2xl font-bold mb-4 text-center text-blue-400">AI Auto Clipper</h1>
+        <form id="clipForm" class="space-y-4">
+            <div>
+                <label class="block text-sm font-medium mb-1">Direct URL Video (MP4)</label>
+                <input type="url" id="videoUrl" class="w-full p-2 bg-gray-700 rounded border border-gray-600 focus:border-blue-500 outline-none" placeholder="https://domain.com/video.mp4">
+            </div>
+            <div class="text-center text-sm font-bold text-gray-400">ATAU</div>
+            <div>
+                <label class="block text-sm font-medium mb-1">Upload Video</label>
+                <input type="file" id="videoFile" accept="video/mp4,video/webm" class="w-full p-2 bg-gray-700 rounded border border-gray-600">
+            </div>
+            <div>
+                <label class="block text-sm font-medium mb-1">Target Durasi</label>
+                <select id="duration" class="w-full p-2 bg-gray-700 rounded border border-gray-600">
+                    <option value="30">30 Detik (Shorts/Reels)</option>
+                    <option value="60">60 Detik</option>
+                    <option value="120">120 Detik</option>
+                </select>
+            </div>
+            <button type="submit" class="w-full bg-blue-600 hover:bg-blue-700 text-white font-bold py-2 px-4 rounded transition">Buat Clip!</button>
+        </form>
+        <div id="statusArea" class="mt-6 hidden">
+            <h3 class="font-bold text-yellow-400 mb-2">Status Proses:</h3>
+            <p id="statusText" class="text-sm bg-gray-700 p-3 rounded">Memulai...</p>
+            <div class="w-full bg-gray-600 rounded-full h-2.5 mt-2">
+              <div id="progressBar" class="bg-blue-500 h-2.5 rounded-full" style="width: 0%"></div>
+            </div>
+        </div>
+        <div id="resultArea" class="mt-6 hidden text-center">
+            <h3 class="font-bold text-green-400 mb-2">Selesai!</h3>
+            <p class="text-xs text-gray-400 mb-3">Link berlaku selama 1 Jam</p>
+            <a id="downloadBtn" href="#" class="inline-block bg-green-600 hover:bg-green-700 text-white font-bold py-2 px-4 rounded">Download Video</a>
+        </div>
+    </div>
+    <script>
+        document.getElementById('clipForm').addEventListener('submit', async (e) => {
+            e.preventDefault();
+            const formData = new FormData();
+            formData.append('duration', document.getElementById('duration').value);
+            const file = document.getElementById('videoFile').files[0];
+            const url = document.getElementById('videoUrl').value;
+            if (file) formData.append('video_file', file);
+            else if (url) formData.append('video_url', url);
+            else return alert("Masukkan URL atau Upload Video!");
+            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);
+            }
+        });
+        async function pollStatus(jobId) {
+            const interval = setInterval(async () => {
+                const res = await fetch(`/status/${jobId}`);
+                const data = await res.json();
+                document.getElementById('statusText').innerText = data.message;
+                document.getElementById('progressBar').style.width = data.progress + "%";
+                if (data.status === 'completed') {
+                    clearInterval(interval);
+                    document.getElementById('statusArea').classList.add('hidden');
+                    document.getElementById('resultArea').classList.remove('hidden');
+                    document.getElementById('downloadBtn').href = `/download/${data.result}`;
+                } else if (data.status === 'error') {
+                    clearInterval(interval);
+                    document.getElementById('statusText').innerText = "Error: " + data.message;
+                    document.getElementById('statusText').classList.add('text-red-400');
+                }
+            }, 3000);
+        }
+    </script>
+</body>
+</html>
+"""
+# ==========================================
+# HELPER FUNCTIONS
+# ==========================================
+def time_to_seconds(time_str):
+    try:
+        parts = time_str.split(':')
+        return int(parts[0]) * 60 + int(parts[1])
+    except:
+        return 0
+def background_processor(job_id, video_path, duration):
+    try:
+        JOBS[job_id]['message'] = "Step 2: Melakukan Transkripsi (AI Whisper)..."
+        JOBS[job_id]['progress'] = 20
+        # Transkripsi Video
+        segments, info = whisper_model.transcribe(video_path, word_timestamps=True)
+        transcript = ""
+        words_data = []
+        for segment in segments:
+            for word in segment.words:
+                words_data.append(word)
+                transcript += f"[{int(word.start//60):02d}:{int(word.start%60):02d}] {word.word.strip()}\n"
+        JOBS[job_id]['message'] = "Step 3: Meminta AI menganalisis Hook terbaik..."
+        JOBS[job_id]['progress'] = 50
+        # Meminta LLM menganalisa clip
+        prompt = f"Cari bagian paling menarik (hook bagus) durasi ~{duration} detik dari transkrip ini. Balas HANYA format 'MM:SS - MM:SS'.\n\nTranskrip:\n{transcript[:3000]}"
+        api_url = "https://www.puruboy.kozow.com/api/ai/letmegpt"
+        response = requests.post(api_url, json={"message": prompt})
+        ai_result = response.json().get("result", {}).get("answer", "")
+        # Ekstrak waktu
+        times = re.findall(r'(\d{1,2}:\d{2})', ai_result)
+        if len(times) >= 2:
+            start_sec = time_to_seconds(times[0])
+            end_sec = time_to_seconds(times[1])
+        else:
+            start_sec = 0
+            end_sec = int(duration)
+        JOBS[job_id]['message'] = f"Step 4: Memotong & render ({times[0]} - {times[1]})..."
+        JOBS[job_id]['progress'] = 70
+        # Penyiapan output
+        output_filename = f"clip_{job_id}.mp4"
+        output_path = os.path.join(app.config['UPLOAD_FOLDER'], output_filename)
+        # Tambahkan output hasil ke daftar sampah yang akan dihapus nanti (1 Jam kemudian)
+        JOBS[job_id]['files_to_delete'].append(output_path)
+        # Potong & Crop Rasio 9:16 (Vertikal)
+        clip = VideoFileClip(video_path).subclip(start_sec, end_sec)
+        w, h = clip.size
+        target_w = int(h * 9 / 16)
+        x_center = w / 2
+        cropped_clip = clip.crop(x1=x_center - target_w/2, y1=0, x2=x_center + target_w/2, y2=h)
+        # Font Absolute Path (Sesuai dengan apt-get fonts-dejavu di Dockerfile)
+        font_path = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"
+        # Pembuatan Subtitle
+        txt_clips = []
+        for word in words_data:
+            if word.start >= start_sec and word.end <= end_sec:
+                rel_start = word.start - start_sec
+                rel_end = word.end - start_sec
+                txt = TextClip(
+                    word.word.strip(),
+                    font=font_path,
+                    fontsize=50,
+                    color='yellow',
+                    stroke_color='black',
+                    stroke_width=3
+                )
+                txt = txt.set_position(('center', 'center')).set_start(rel_start).set_end(rel_end)
+                txt_clips.append(txt)
+        final_video = CompositeVideoClip([cropped_clip] + txt_clips)
+        # CPU Render Optimized
+        final_video.write_videofile(output_path, fps=30, codec="libx264", audio_codec="aac", threads=4, preset="ultrafast", logger=None)
+        # Tutup memori MoviePy
+        clip.close()
+        cropped_clip.close()
+        final_video.close()
+        # Update Status Selesai
+        JOBS[job_id]['status'] = "completed"
+        JOBS[job_id]['progress'] = 100
+        JOBS[job_id]['message'] = "Video selesai dibuat!"
+        JOBS[job_id]['result'] = output_filename
+    except Exception as e:
+        JOBS[job_id]['status'] = "error"
+        JOBS[job_id]['message'] = str(e)
+# ==========================================
+# ROUTES
+# ==========================================
+@app.route('/')
+def index():
+    return render_template_string(HTML_TEMPLATE)
+@app.route('/generate', methods=['POST'])
+def generate():
+    job_id = str(uuid.uuid4())
+    duration = request.form.get('duration', '60')
+    video_path = ""
+    # Menangani Upload vs URL
+    if 'video_file' in request.files and request.files['video_file'].filename != '':
+        file = request.files['video_file']
+        filename = secure_filename(file.filename)
+        video_path = os.path.join(app.config['UPLOAD_FOLDER'], f"{job_id}_{filename}")
+        file.save(video_path)
+    elif 'video_url' in request.form and request.form['video_url'] != '':
+        video_url = request.form['video_url']
+        video_path = os.path.join(app.config['UPLOAD_FOLDER'], f"{job_id}_downloaded.mp4")
+        # Download dengan sistem chunk (aman untuk RAM)
+        r = requests.get(video_url, stream=True)
+        with open(video_path, 'wb') as f:
+            for chunk in r.iter_content(chunk_size=1024*1024):
+                if chunk: f.write(chunk)
+    else:
+        return jsonify({"error": "Video tidak ditemukan"}), 400
+    # Daftarkan Job (beserta tracker file mentah agar bisa dibersihkan 1 Jam ke depan)
+    JOBS[job_id] = {
+        "status": "processing",
+        "progress": 10,
+        "message": "Step 1: Video berhasil dimuat. Bersiap memproses...",
+        "result": None,
+        "created_at": time.time(),
+        "files_to_delete": [video_path]
+    }
+    # Jalankan proses secara asynchronous
+    thread = threading.Thread(target=background_processor, args=(job_id, video_path, duration))
+    thread.start()
+    return jsonify({"job_id": job_id})
+@app.route('/status/<job_id>')
+def status(job_id):
+    job = JOBS.get(job_id)
+    if not job:
+        return jsonify({"status": "error", "message": "Job sudah kedaluwarsa atau tidak ditemukan"})
+    # Sembunyikan field sensitif (seperti path/files_to_delete) saat kirim ke client UI
+    return jsonify({
+        "status": job['status'],
+        "progress": job['progress'],
+        "message": job['message'],
+        "result": job['result']
+    })
+@app.route('/download/<filename>')
+def download(filename):
+    file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+    if not os.path.exists(file_path):
+        return "File tidak ditemukan atau sudah dihapus server.", 404
+    return send_file(file_path, as_attachment=True)
+if __name__ == '__main__':
+    # Harus jalan di port 7860 untuk HuggingFace Spaces
+    app.run(host='0.0.0.0', port=7860, debug=False)