import os import json import subprocess from flask import Flask, request, jsonify, send_file, Response app = Flask(__name__) BASE_DIR = os.path.dirname(os.path.abspath(__file__)) HTML = r""" MKV Sub Viewer
— extrator de legendas local
▶ Selecione um arquivo
para ver as legendas disponíveis
""" def find_mkv_files(): return sorted(f for f in os.listdir(BASE_DIR) if f.lower().endswith('.mkv')) def get_subtitle_tracks(filepath): cmd = ['ffprobe', '-v', 'quiet', '-print_format', 'json', '-show_streams', '-select_streams', 's', filepath] result = subprocess.run(cmd, capture_output=True, text=True) if result.returncode != 0: return [] tracks = [] for s in json.loads(result.stdout).get('streams', []): tags = s.get('tags', {}) lang = tags.get('language', 'und') title = tags.get('title', '') codec = s.get('codec_name', '') idx = s.get('index') tracks.append({ 'index': idx, 'language': lang, 'title': title, 'codec': codec, 'label': f"[{lang.upper()}] {title or codec} (stream {idx})" }) return tracks @app.route('/') def index(): return HTML @app.route('/files') def files(): return jsonify(find_mkv_files()) @app.route('/tracks') def tracks(): filepath = os.path.join(BASE_DIR, request.args.get('file', '')) if not os.path.isfile(filepath): return jsonify([]) return jsonify(get_subtitle_tracks(filepath)) @app.route('/subtitle') def subtitle(): filepath = os.path.join(BASE_DIR, request.args.get('file', '')) stream_index = request.args.get('stream', type=int) if not os.path.isfile(filepath) or stream_index is None: return 'Invalido', 400 cmd = ['ffmpeg', '-y', '-i', filepath, '-map', f'0:{stream_index}', '-c:s', 'webvtt', '-f', 'webvtt', 'pipe:1'] print(f"\n=== ffmpeg cmd: {' '.join(cmd)}") result = subprocess.run(cmd, capture_output=True) stderr = result.stderr.decode('utf-8', errors='replace') stdout = result.stdout.decode('utf-8', errors='replace') print(f"=== returncode: {result.returncode}") print(f"=== stderr:\n{stderr}") print(f"=== stdout preview:\n{stdout[:300]}") if result.returncode != 0: return stderr, 500 return Response(stdout, mimetype='text/vtt') @app.route('/video') def video(): filepath = os.path.join(BASE_DIR, request.args.get('file', '')) if not os.path.isfile(filepath): return 'Nao encontrado', 404 size = os.path.getsize(filepath) range_header = request.headers.get('Range') if not range_header: return send_file(filepath, mimetype='video/x-matroska') m = range_header.replace('bytes=', '').split('-') byte1 = int(m[0]) byte2 = int(m[1]) if m[1] else size - 1 length = byte2 - byte1 + 1 with open(filepath, 'rb') as f: f.seek(byte1) data = f.read(length) rv = Response(data, 206, mimetype='video/x-matroska', direct_passthrough=True) rv.headers['Content-Range'] = f'bytes {byte1}-{byte2}/{size}' rv.headers['Accept-Ranges'] = 'bytes' rv.headers['Content-Length'] = str(length) return rv if __name__ == '__main__': app.run(debug=True, host='0.0.0.0', port=5000)