from flask import Flask, request, send_file, render_template_string, jsonify import subprocess, os, uuid, json, re, threading, queue, time from datetime import datetime app = Flask(__name__) # ================= CONFIGURAÇÕES ================= VIDEO_DIR = "/var/www/files/drige" TMP_THUMBS = "/tmp/thumbs" CUT_DIR = "/var/www/files/cortes" SNAPSHOT_DIR = "/var/www/files/snapshots" QUEUE_FILE = "/tmp/video_queue.json" os.makedirs(TMP_THUMBS, exist_ok=True) os.makedirs(CUT_DIR, exist_ok=True) os.makedirs(SNAPSHOT_DIR, exist_ok=True) # ================= SISTEMA DE FILA ================= class ConversionQueue: def __init__(self, max_workers=3): self.queue = queue.Queue() self.active_jobs = {} self.completed_jobs = {} self.failed_jobs = {} self.max_workers = max_workers self.lock = threading.Lock() self.load_state() for _ in range(max_workers): threading.Thread(target=self.worker, daemon=True).start() def load_state(self): try: if os.path.exists(QUEUE_FILE): with open(QUEUE_FILE, 'r') as f: data = json.load(f) self.completed_jobs = data.get('completed', {}) self.failed_jobs = data.get('failed', {}) except: pass def save_state(self): try: with open(QUEUE_FILE, 'w') as f: json.dump({'completed': self.completed_jobs, 'failed': self.failed_jobs}, f) except: pass def clear_history(self): with self.lock: self.completed_jobs = {} self.failed_jobs = {} self.save_state() def get_all_status(self): with self.lock: return { 'queued': list(self.active_jobs.values()), 'processing': [j for j in self.active_jobs.values() if j['status'] == 'processing'], 'completed': list(self.completed_jobs.values())[::-1], 'failed': list(self.failed_jobs.values())[::-1] } def add_job(self, job_id, job_data): with self.lock: job_data['id'] = job_id job_data['status'] = 'queued' job_data['progress'] = 0 job_data['created_at'] = datetime.now().strftime("%H:%M:%S") self.active_jobs[job_id] = job_data self.queue.put(job_data) return job_id def worker(self): while True: job = self.queue.get() if job is None: break try: with self.lock: self.active_jobs[job['id']]['status'] = 'processing' self.process_conversion(job) with self.lock: job['status'] = 'completed' self.completed_jobs[job['id']] = job if job['id'] in self.active_jobs: del self.active_jobs[job['id']] self.save_state() except Exception as e: with self.lock: job['status'] = 'failed' job['error'] = str(e) self.failed_jobs[job['id']] = job if job['id'] in self.active_jobs: del self.active_jobs[job['id']] self.save_state() finally: self.queue.task_done() def process_conversion(self, job): src_file = os.path.join(VIDEO_DIR, job['pasta'], job['file']) dest_dir = os.path.join(CUT_DIR, job['pasta_corte']) if job['pasta_corte'] else CUT_DIR os.makedirs(dest_dir, exist_ok=True) if job.get('custom_name'): safe_name = "".join([c for c in job['custom_name'] if c.isalnum() or c in "._- "]).strip() if not safe_name: safe_name = f"cut_{job['id'][:8]}" else: safe_name = f"cut_{job['id'][:8]}" output_path = os.path.join(dest_dir, f"{safe_name}.mp4") # -ss antes de -i: seek rápido (progresso correto) # -avoid_negative_ts make_zero: corrige artefato de primeiro frame para CapCut cmd = [ "ffmpeg", "-y", "-ss", str(job['start']), "-i", src_file, "-t", str(job['duration']), "-avoid_negative_ts", "make_zero", ] vf_filters = [] if job.get('mode') == 'tiktok': vf_filters.append("crop=ih*(9/16):ih") if vf_filters: cmd.extend(["-vf", ",".join(vf_filters)]) crf = "28" if job.get('mode') == 'whatsapp' else "22" cmd.extend([ "-c:v", "libx264", "-preset", "ultrafast", "-crf", crf, "-g", "30", "-keyint_min", "1", "-x264-params", "open-gop=0", "-pix_fmt", "yuv420p", ]) if job.get('mute'): cmd.append("-an") else: cmd.extend(["-c:a", "aac", "-b:a", "128k", "-ar", "44100"]) # -progress pipe:2 envia progresso pro stderr junto com logs cmd.extend(["-progress", "pipe:2", "-movflags", "+faststart", output_path]) process = subprocess.Popen(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.PIPE, universal_newlines=True) stderr_lines = [] while True: line = process.stderr.readline() if not line: break stderr_lines.append(line) if 'out_time_ms' in line: match = re.search(r'out_time_ms=(\d+)', line) if match: time_s = int(match.group(1)) / 1000000.0 progress = min(int((time_s / job['duration']) * 100), 100) with self.lock: if job['id'] in self.active_jobs: self.active_jobs[job['id']]['progress'] = progress process.wait() if process.returncode != 0: raise Exception(f"Erro FFmpeg: {''.join(stderr_lines)[-300:]}") job['output_path'] = output_path conversion_queue = ConversionQueue() # ================= TEMPLATE HTML ================= TEMPLATE = """ ✂️ Studio Mobile V3.3
✂️ STUDIO MOBILE V3.3
Ação realizada!
""" # ================= ROTAS FLASK ================= @app.route("/") def index(): pastas = sorted([d for d in os.listdir(VIDEO_DIR) if os.path.isdir(os.path.join(VIDEO_DIR, d))]) pastas_corte = sorted([d for d in os.listdir(CUT_DIR) if os.path.isdir(os.path.join(CUT_DIR, d))]) return render_template_string(TEMPLATE, pastas=pastas, pastas_corte=pastas_corte) @app.route("/videos_list") def videos_list(): pasta = request.args.get("pasta", "") dir_path = os.path.join(VIDEO_DIR, pasta) videos = [f for f in os.listdir(dir_path) if f.lower().endswith((".mp4",".mkv",".avi"))] if os.path.exists(dir_path) else [] return jsonify(sorted(videos)) @app.route("/video") def video_stream(): return send_file(os.path.join(VIDEO_DIR, request.args.get("pasta"), request.args.get("file"))) @app.route("/take_snapshot", methods=["POST"]) def take_snapshot(): data = request.json src = os.path.join(VIDEO_DIR, data['pasta'], data['file']) name = f"snap_{int(time.time())}.jpg" out = os.path.join(SNAPSHOT_DIR, name) subprocess.run([ "ffmpeg", "-ss", str(data['time']), "-i", src, "-frames:v", "1", "-q:v", "2", "-y", out ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) return jsonify({'success': True, 'name': name}) @app.route("/thumbs") def generate_thumbs(): file, pasta = request.args["file"], request.args.get("pasta", "") src = os.path.join(VIDEO_DIR, pasta, file) vid_id = uuid.uuid5(uuid.NAMESPACE_URL, os.path.join(pasta, file)).hex out_dir = os.path.join(TMP_THUMBS, vid_id) os.makedirs(out_dir, exist_ok=True) def worker(): try: dur = float(subprocess.check_output(["ffprobe","-v","0","-show_entries","format=duration","-of","default=nw=1:nk=1",src])) for t in range(0, int(dur), 60): out = os.path.join(out_dir, f"f_{t:05d}.jpg") if not os.path.exists(out): subprocess.run(["ffmpeg", "-ss", str(t), "-i", src, "-frames:v", "1", "-q:v", "8", "-vf", "scale=320:-1", out, "-y", "-loglevel", "quiet"]) except: pass threading.Thread(target=worker, daemon=True).start() return "ok" @app.route("/thumbs_list") def thumbs_list(): vid_id = uuid.uuid5(uuid.NAMESPACE_URL, os.path.join(request.args.get("pasta"), request.args.get("file"))).hex out_dir = os.path.join(TMP_THUMBS, vid_id) if not os.path.exists(out_dir): return jsonify([]) files = sorted(os.listdir(out_dir)) return jsonify([{"sec": int(f[2:7]), "src": f"/thumb/{vid_id}/{f}"} for f in files]) @app.route("/thumb//") def serve_thumb(vid, n): return send_file(os.path.join(TMP_THUMBS, vid, n)) @app.route("/add_to_queue", methods=["POST"]) def add_to_queue(): job_id = uuid.uuid4().hex conversion_queue.add_job(job_id, request.json) return jsonify({'id': job_id}) @app.route("/queue_status") def queue_status(): return jsonify(conversion_queue.get_all_status()) @app.route("/clear_queue", methods=["POST"]) def clear_queue(): conversion_queue.clear_history() return jsonify({'success': True}) @app.route("/download_job/") def download_job(job_id): job = conversion_queue.completed_jobs.get(job_id) return send_file(job['output_path'], as_attachment=True) if job else ("Erro", 404) if __name__ == "__main__": app.run("0.0.0.0", 8080)