commit 95a9bcd24d1a6d0647fa36cb19a860f1a43a4b78 Author: Creidsu Date: Sun Feb 8 23:07:50 2026 +0000 inicio diff --git a/Dockerfile b/Dockerfile new file mode 100755 index 0000000..78c99e8 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,19 @@ +FROM python:3.11-slim + +# Instala FFmpeg e dependências de sistema +RUN apt-get update && apt-get install -y \ + ffmpeg \ + git \ + && rm -rf /var/lib/apt/lists/* + +WORKDIR /app + +COPY requirements.txt . +RUN pip install --no-cache-dir -r requirements.txt + +COPY app /app/app + +# Porta do NiceGUI +EXPOSE 8080 + +CMD ["python", "app/main.py"] diff --git a/README.md b/README.md new file mode 100755 index 0000000..14966ff --- /dev/null +++ b/README.md @@ -0,0 +1,7 @@ +# Clei-Flow + +**Clei-Flow** é um gerenciador de mídia Open Source focado em automação, padronização e compatibilidade de hardware. + +## Estrutura +* **Core:** Núcleos independentes para Renomear, Converter (FFmpeg) e Notificar (Telegram). +* **UI:** Interface NiceGUI para configuração dinâmica. diff --git a/app/__pycache__/database.cpython-311.pyc b/app/__pycache__/database.cpython-311.pyc new file mode 100644 index 0000000..c084278 Binary files /dev/null and b/app/__pycache__/database.cpython-311.pyc differ diff --git a/app/core/__init__.py b/app/core/__init__.py new file mode 100755 index 0000000..e69de29 diff --git a/app/core/__pycache__/__init__.cpython-311.pyc b/app/core/__pycache__/__init__.cpython-311.pyc new file mode 100644 index 0000000..bf23309 Binary files /dev/null and b/app/core/__pycache__/__init__.cpython-311.pyc differ diff --git a/app/core/__pycache__/bot.cpython-311.pyc b/app/core/__pycache__/bot.cpython-311.pyc new file mode 100644 index 0000000..e2bc932 Binary files /dev/null and b/app/core/__pycache__/bot.cpython-311.pyc differ diff --git a/app/core/__pycache__/ffmpeg_engine.cpython-311.pyc b/app/core/__pycache__/ffmpeg_engine.cpython-311.pyc new file mode 100644 index 0000000..869343e Binary files /dev/null and b/app/core/__pycache__/ffmpeg_engine.cpython-311.pyc differ diff --git a/app/core/__pycache__/renamer.cpython-311.pyc b/app/core/__pycache__/renamer.cpython-311.pyc new file mode 100644 index 0000000..78b4429 Binary files /dev/null and b/app/core/__pycache__/renamer.cpython-311.pyc differ diff --git a/app/core/__pycache__/state.cpython-311.pyc b/app/core/__pycache__/state.cpython-311.pyc new file mode 100644 index 0000000..6263016 Binary files /dev/null and b/app/core/__pycache__/state.cpython-311.pyc differ diff --git a/app/core/__pycache__/watcher.cpython-311.pyc b/app/core/__pycache__/watcher.cpython-311.pyc new file mode 100644 index 0000000..bdbaefe Binary files /dev/null and b/app/core/__pycache__/watcher.cpython-311.pyc differ diff --git a/app/core/bot.py b/app/core/bot.py new file mode 100755 index 0000000..ee15807 --- /dev/null +++ b/app/core/bot.py @@ -0,0 +1,166 @@ +import asyncio +import httpx # Usamos para testar a rede antes +from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup +from telegram.ext import ApplicationBuilder, ContextTypes, CallbackQueryHandler, CommandHandler +from database import AppConfig +import logging + +# Configuração de Log +logging.getLogger("httpx").setLevel(logging.WARNING) +logging.getLogger("telegram").setLevel(logging.INFO) + +class TelegramManager: + def __init__(self): + # NÃO carregamos o token aqui. Carregamos na hora de iniciar. + self.app = None + self.active_requests = {} + self.is_connected = False + + async def check_internet(self): + """Testa se o container tem internet antes de tentar o Telegram""" + try: + async with httpx.AsyncClient(timeout=5.0) as client: + await client.get("https://www.google.com") + return True + except: + return False + + async def start(self): + """Inicia o Bot""" + # 1. Pega o Token Fresquinho do Banco + token = AppConfig.get_val('telegram_token') + chat_id = AppConfig.get_val('telegram_chat_id') + + if not token: + print("🟡 Bot: Token não configurado. Aguardando você salvar no Painel...") + return + + # 2. Teste de Rede Prévio + print("🤖 Bot: Verificando conectividade...") + if not await self.check_internet(): + print("❌ Bot: ERRO DE REDE! O container não consegue acessar a internet.") + print(" -> Verifique DNS ou Firewall.") + return + + print(f"🤖 Bot: Conectando com token termina em ...{token[-5:]}") + + try: + # 3. Constroi a Aplicação + self.app = ApplicationBuilder().token(token).build() + + # Handlers + self.app.add_handler(CommandHandler("start", self.cmd_start)) + self.app.add_handler(CommandHandler("id", self.cmd_id)) + self.app.add_handler(CallbackQueryHandler(self.handle_selection)) + + # Inicializa + await self.app.initialize() + await self.app.start() + + # Inicia Polling (Limpa mensagens velhas acumuladas para não travar) + await self.app.updater.start_polling(drop_pending_updates=True) + + self.is_connected = True + print("✅ Bot Online e Rodando!") + + # Tenta mandar um oi se tiver chat_id + if chat_id: + try: + await self.app.bot.send_message(chat_id=chat_id, text="🚀 Clei-Flow: Conexão restabelecida!") + except Exception as e: + print(f"⚠️ Bot online, mas falhou ao enviar msg (Chat ID errado?): {e}") + + except Exception as e: + print(f"❌ Falha Crítica no Bot: {e}") + self.is_connected = False + + async def stop(self): + if self.app: + try: + await self.app.updater.stop() + await self.app.stop() + await self.app.shutdown() + self.is_connected = False + except: pass + + # --- COMANDOS --- + async def cmd_start(self, update: Update, context: ContextTypes.DEFAULT_TYPE): + chat_id = update.effective_chat.id + await update.message.reply_text(f"Olá! Configurado.\nSeu Chat ID é: `{chat_id}`", parse_mode='Markdown') + # Opcional: Salvar o Chat ID automaticamente se o usuário mandar /start + # AppConfig.set_val('telegram_chat_id', str(chat_id)) + + async def cmd_id(self, update: Update, context: ContextTypes.DEFAULT_TYPE): + await update.message.reply_text(f"`{update.effective_chat.id}`", parse_mode='Markdown') + + # --- INTERAÇÃO (Renomeação) --- + async def ask_user_choice(self, filename, candidates): + chat_id = AppConfig.get_val('telegram_chat_id') # Pega sempre o atual + if not chat_id or not self.is_connected: + print("❌ Bot não pode perguntar (Sem Chat ID ou Desconectado)") + return None + + request_id = f"req_{filename}" + keyboard = [] + for cand in candidates: + # Texto do botão + text = f"{cand['title']} ({cand['year']})" + # Dados (ID|Tipo) + callback_data = f"{request_id}|{cand['tmdb_id']}|{cand['type']}" + keyboard.append([InlineKeyboardButton(text, callback_data=callback_data)]) + + keyboard.append([InlineKeyboardButton("🚫 Ignorar", callback_data=f"{request_id}|IGNORE|NONE")]) + reply_markup = InlineKeyboardMarkup(keyboard) + + try: + await self.app.bot.send_message( + chat_id=chat_id, + text=f"🤔 Clei-Flow Precisa de Ajuda:\nArquivo: {filename}", + reply_markup=reply_markup, + parse_mode='HTML' + ) + except Exception as e: + print(f"Erro ao enviar pergunta: {e}") + return None + + loop = asyncio.get_running_loop() + future = loop.create_future() + self.active_requests[request_id] = future + + try: + # Espera 12 horas + result = await asyncio.wait_for(future, timeout=43200) + return result + except asyncio.TimeoutError: + if request_id in self.active_requests: del self.active_requests[request_id] + return None + + async def handle_selection(self, update: Update, context: ContextTypes.DEFAULT_TYPE): + query = update.callback_query + await query.answer() + + data = query.data.split('|') + if len(data) < 3: return + + req_id = data[0] + tmdb_id = data[1] + media_type = data[2] + + if req_id in self.active_requests: + future = self.active_requests[req_id] + if tmdb_id == 'IGNORE': + await query.edit_message_text(text=f"🚫 Ignorado.") + future.set_result(None) + else: + await query.edit_message_text(text=f"✅ Processando...") + future.set_result({'tmdb_id': int(tmdb_id), 'type': media_type}) + del self.active_requests[req_id] + else: + await query.edit_message_text(text="⚠️ Solicitação expirada.") + + async def send_notification(self, message): + chat_id = AppConfig.get_val('telegram_chat_id') + if self.app and chat_id and self.is_connected: + try: + await self.app.bot.send_message(chat_id=chat_id, text=message) + except: pass \ No newline at end of file diff --git a/app/core/ffmpeg_engine.py b/app/core/ffmpeg_engine.py new file mode 100755 index 0000000..b1395b8 --- /dev/null +++ b/app/core/ffmpeg_engine.py @@ -0,0 +1,121 @@ +import subprocess +import json +from database import FFmpegProfile +from core.state import state + +class FFmpegEngine: + def __init__(self, profile_id=None): + if profile_id: + self.profile = FFmpegProfile.get_by_id(profile_id) + else: + self.profile = FFmpegProfile.get_or_none(FFmpegProfile.is_active == True) + + if not self.profile: + state.log("⚠️ Nenhum perfil FFmpeg ativo!") + + def get_file_info(self, filepath): + cmd = ['ffprobe', '-v', 'quiet', '-print_format', 'json', '-show_streams', '-show_format', filepath] + try: + output = subprocess.check_output(cmd).decode('utf-8') + return json.loads(output) + except: return None + + def get_duration(self, filepath): + try: return float(self.get_file_info(filepath)['format']['duration']) + except: return 0 + + def build_command(self, input_file, output_file): + if not self.profile: raise Exception("Perfil não selecionado") + p = self.profile + metadata = self.get_file_info(input_file) + if not metadata: raise Exception("Metadados inválidos") + + cmd = ['ffmpeg', '-y'] + + # --- HARDWARE INIT --- + # VAAPI (Intel Linux/Docker) - O Jeito Correto + if 'vaapi' in p.video_codec: + cmd.extend(['-init_hw_device', 'vaapi=va:/dev/dri/renderD128']) + cmd.extend(['-hwaccel', 'vaapi', '-hwaccel_output_format', 'vaapi', '-hwaccel_device', 'va']) + cmd.extend(['-i', input_file]) + # Filtro essencial para VAAPI: garante formato NV12 na GPU + # Mas como usamos hwaccel_output_format vaapi, o filtro pode ser simplificado ou scale_vaapi + # Vamos usar o padrão seguro que funciona em Haswell: + video_filters = 'format=nv12,hwupload' + + elif 'qsv' in p.video_codec: + cmd.extend(['-hwaccel', 'qsv', '-i', input_file]) + video_filters = None + + elif 'nvenc' in p.video_codec: + cmd.extend(['-hwaccel', 'cuda', '-i', input_file]) + video_filters = None + + else: + # CPU + cmd.extend(['-i', input_file]) + video_filters = None + + # --- VÍDEO --- + cmd.extend(['-map', '0:v:0']) + + if p.video_codec == 'copy': + cmd.extend(['-c:v', 'copy']) + else: + cmd.extend(['-c:v', p.video_codec]) + + # Se tem filtro de hardware (VAAPI precisa subir pra GPU se hwaccel falhar no decode) + if 'vaapi' in p.video_codec: + # Se usarmos -hwaccel vaapi, o stream ja esta na GPU. + # Mas as vezes precisamos garantir o filtro scale_vaapi se fosse redimensionar. + # Para manter simples e funcional no Haswell: + # cmd.extend(['-vf', 'format=nv12,hwupload']) <--- Se nao usar hwaccel + # Com hwaccel, nao precisa do hwupload, mas precisa garantir compatibilidade + pass + + # Configs de Encoder + if 'vaapi' in p.video_codec: + # VAAPI usa QP, não CRF padrão + # Se der erro, troque '-qp' por '-rc_mode CQP -global_quality' + cmd.extend(['-qp', str(p.crf)]) + + elif 'qsv' in p.video_codec: + cmd.extend(['-global_quality', str(p.crf), '-look_ahead', '1']) + cmd.extend(['-preset', p.preset]) + + elif 'nvenc' in p.video_codec: + cmd.extend(['-cq', str(p.crf), '-preset', p.preset]) + + elif 'libx264' in p.video_codec: + cmd.extend(['-crf', str(p.crf), '-preset', p.preset]) + + # --- ÁUDIO --- + allowed = p.audio_langs.split(',') if p.audio_langs else [] + audio_streams = [s for s in metadata['streams'] if s['codec_type'] == 'audio'] + acount = 0 + for s in audio_streams: + l = s.get('tags', {}).get('language', 'und') + if not allowed or l in allowed or 'und' in allowed: + cmd.extend(['-map', f'0:{s["index"]}']) + cmd.extend([f'-c:a:{acount}', 'aac', f'-b:a:{acount}', '192k']) + acount += 1 + if acount == 0: cmd.extend(['-map', '0:a:0', '-c:a', 'aac']) + + # --- LEGENDAS --- + lallowed = p.subtitle_langs.split(',') if p.subtitle_langs else [] + sub_streams = [s for s in metadata['streams'] if s['codec_type'] == 'subtitle'] + scount = 0 + for s in sub_streams: + l = s.get('tags', {}).get('language', 'und') + if not lallowed or l in lallowed or 'und' in lallowed: + cmd.extend(['-map', f'0:{s["index"]}']) + cmd.extend([f'-c:s:{scount}', 'copy']) + scount += 1 + + cmd.extend(['-metadata', 'title=', '-metadata', 'comment=CleiFlow']) + cmd.append(output_file) + + # LOG DO COMANDO PARA DEBUG + state.log(f"🛠️ CMD: {' '.join(cmd)}") + + return cmd \ No newline at end of file diff --git a/app/core/flow.py b/app/core/flow.py new file mode 100755 index 0000000..ce3de75 --- /dev/null +++ b/app/core/flow.py @@ -0,0 +1,15 @@ +from .renamer import RenamerCore +from .ffmpeg_engine import FFmpegCore +from .bot import BotCore + +class CleiFlow: + """ + Gerencia o ciclo de vida do arquivo. + Modos: Manual, Híbrido, Automático. + """ + def start_pipeline(self, file_path): + # 1. Identificar + # 2. Se ambíguo -> Chamar Bot.ask_for_decision() -> Pausar Thread + # 3. Converter (FFmpegCore) + # 4. Mover + pass diff --git a/app/core/renamer.py b/app/core/renamer.py new file mode 100755 index 0000000..f0d8c61 --- /dev/null +++ b/app/core/renamer.py @@ -0,0 +1,155 @@ +import os +import re +from guessit import guessit +from tmdbv3api import TMDb, Movie, TV, Search +from database import AppConfig +from difflib import SequenceMatcher + +class RenamerCore: + def __init__(self): + self.api_key = AppConfig.get_val('tmdb_api_key') + self.lang = AppConfig.get_val('tmdb_language', 'pt-BR') + self.min_confidence = int(AppConfig.get_val('min_confidence', '90')) / 100.0 + + self.tmdb = TMDb() + if self.api_key: + self.tmdb.api_key = self.api_key + self.tmdb.language = self.lang + + self.movie_api = Movie() + self.tv_api = TV() + self.search_api = Search() + + def identify_file(self, filepath): + filename = os.path.basename(filepath) + try: + guess = guessit(filename) + except Exception as e: + return {'status': 'ERROR', 'msg': str(e)} + + title = guess.get('title') + if not title: return {'status': 'NOT_FOUND', 'msg': 'Sem título'} + + if not self.api_key: return {'status': 'ERROR', 'msg': 'Sem API Key'} + + try: + media_type = guess.get('type', 'movie') + if media_type == 'episode': + results = self.search_api.tv_shows(term=title) + else: + results = self.search_api.movies(term=title) + if not results: results = self.search_api.tv_shows(term=title) + except: return {'status': 'NOT_FOUND', 'msg': 'Erro TMDb'} + + if not results: return {'status': 'NOT_FOUND', 'msg': 'Nenhum resultado TMDb'} + + # --- CORREÇÃO DE SEGURANÇA (O erro 'str object' estava aqui) --- + # Se o TMDb retornou um dicionário (paginado), pegamos a lista dentro dele. + if isinstance(results, dict) and 'results' in results: + results = results['results'] + # Se retornou um objeto que tem atributo 'results', usamos ele + elif hasattr(results, 'results'): + results = results.results + + # Se ainda assim for uma lista de strings (chaves), aborta + if results and isinstance(results, list) and len(results) > 0 and isinstance(results[0], str): + # Isso acontece se iterou sobre chaves de um dict sem querer + return {'status': 'NOT_FOUND', 'msg': 'Formato de resposta inválido'} + # ------------------------------------------------------------- + + candidates = [] + for res in results: + # Proteção extra: se o item for string, pula + if isinstance(res, str): continue + + # Obtém atributos de forma segura (funciona para dict ou objeto) + if isinstance(res, dict): + r_id = res.get('id') + r_title = res.get('title') or res.get('name') + r_date = res.get('release_date') or res.get('first_air_date') + r_overview = res.get('overview', '') + else: + r_id = getattr(res, 'id', None) + r_title = getattr(res, 'title', getattr(res, 'name', '')) + r_date = getattr(res, 'release_date', getattr(res, 'first_air_date', '')) + r_overview = getattr(res, 'overview', '') + + if not r_title or not r_id: continue + + r_year = int(str(r_date)[:4]) if r_date else 0 + + # Score e Comparação + t1 = str(title).lower() + t2 = str(r_title).lower() + + base_score = SequenceMatcher(None, t1, t2).ratio() + + if t1 in t2 or t2 in t1: + base_score = max(base_score, 0.85) + + g_year = guess.get('year') + if g_year and r_year: + if g_year == r_year: base_score += 0.15 + elif abs(g_year - r_year) <= 1: base_score += 0.05 + + final_score = min(base_score, 1.0) + + candidates.append({ + 'tmdb_id': r_id, + 'title': r_title, + 'year': r_year, + 'type': 'movie' if hasattr(res, 'title') or (isinstance(res, dict) and 'title' in res) else 'tv', + 'overview': str(r_overview)[:100], + 'score': final_score + }) + + if not candidates: return {'status': 'NOT_FOUND', 'msg': 'Sem candidatos válidos'} + + candidates.sort(key=lambda x: x['score'], reverse=True) + best = candidates[0] + + if len(candidates) == 1 and best['score'] > 0.6: + return {'status': 'MATCH', 'match': best, 'guessed': guess} + + is_clear_winner = False + if len(candidates) > 1: + if (best['score'] - candidates[1]['score']) > 0.15: + is_clear_winner = True + + if best['score'] >= self.min_confidence or is_clear_winner: + return {'status': 'MATCH', 'match': best, 'guessed': guess} + + return {'status': 'AMBIGUOUS', 'candidates': candidates[:5], 'guessed': guess} + + def get_details(self, tmdb_id, media_type): + if media_type == 'movie': return self.movie_api.details(tmdb_id) + return self.tv_api.details(tmdb_id) + + def build_path(self, category_obj, media_info, guessed_info): + clean_title = re.sub(r'[\\/*?:"<>|]', "", media_info['title']).strip() + year = str(media_info['year']) + + forced_type = category_obj.content_type + actual_type = media_info['type'] + + is_series = False + if forced_type == 'series': is_series = True + elif forced_type == 'movie': is_series = False + else: is_series = (actual_type == 'tv') + + if not is_series: + return f"{clean_title} ({year}).mkv" + else: + season = guessed_info.get('season') + episode = guessed_info.get('episode') + + if isinstance(season, list): season = season[0] + if isinstance(episode, list): episode = episode[0] + + if not season: season = 1 + if not episode: episode = 1 + + season_folder = f"Temporada {int(season):02d}" + file_suffix = f"S{int(season):02d}E{int(episode):02d}" + + return os.path.join(clean_title, season_folder, f"{clean_title} {file_suffix}.mkv") \ No newline at end of file diff --git a/app/core/state.py b/app/core/state.py new file mode 100644 index 0000000..0409e05 --- /dev/null +++ b/app/core/state.py @@ -0,0 +1,49 @@ +from collections import deque, OrderedDict + +class AppState: + def __init__(self): + # --- Logs do Sistema --- + self.logs = deque(maxlen=1000) + + # --- Referência ao Watcher --- + self.watcher = None + + # --- Lista de Tarefas (Visualização tipo Árvore/Lista) --- + self.tasks = OrderedDict() + + # --- Variáveis de Estado (Compatibilidade) --- + self.current_file = "" + self.progress = 0.0 + self.status_text = "Aguardando..." + + def log(self, message): + """Adiciona log e printa no console""" + print(message) + self.logs.append(message) + + def update_task(self, filename, status, progress=0, label=None): + """Atualiza o status de um arquivo na interface""" + # Se não existe, cria + if filename not in self.tasks: + self.tasks[filename] = { + 'status': 'pending', + 'progress': 0, + 'label': label or filename + } + # Limita a 20 itens para não travar a tela + if len(self.tasks) > 20: + self.tasks.popitem(last=False) # Remove o mais antigo + + # Atualiza dados + self.tasks[filename]['status'] = status + self.tasks[filename]['progress'] = progress + if label: + self.tasks[filename]['label'] = label + + def get_logs(self): + return list(self.logs) + +# --- INSTÂNCIA GLOBAL --- +# Ao ser importado, isso roda uma vez e cria o objeto. +# Todo mundo que fizer 'from core.state import state' vai pegar essa mesma instância. +state = AppState() \ No newline at end of file diff --git a/app/core/watcher.py b/app/core/watcher.py new file mode 100644 index 0000000..70cb327 --- /dev/null +++ b/app/core/watcher.py @@ -0,0 +1,243 @@ +import asyncio +import os +import shutil +import re +from pathlib import Path +from database import AppConfig, Category +from core.renamer import RenamerCore +from core.ffmpeg_engine import FFmpegEngine +from core.bot import TelegramManager +from core.state import state + +VIDEO_EXTENSIONS = {'.mkv', '.mp4', '.avi', '.mov', '.wmv'} + +class DirectoryWatcher: + def __init__(self, bot: TelegramManager): + self.bot = bot + self.renamer = RenamerCore() + + # Inicia pausado (True só quando ativado no Dashboard) + self.is_running = False + + self.temp_dir = Path('/app/temp') + self.temp_dir.mkdir(parents=True, exist_ok=True) + self.current_watch_path = None + + # Controle de Processo + self.current_process = None + self.pending_future = None + self.abort_flag = False + + state.watcher = self + + async def start(self): + """Inicia o loop do serviço""" + state.log("🟡 Watcher Service: Pronto. Aguardando ativação no Dashboard...") + + while True: + if self.is_running: + try: + config_path = AppConfig.get_val('monitor_path', '/downloads') + watch_dir = Path(config_path) + + if str(watch_dir) != str(self.current_watch_path): + state.log(f"📁 Monitorando: {watch_dir}") + self.current_watch_path = watch_dir + + if watch_dir.exists(): + await self.scan_folder(watch_dir) + except Exception as e: + state.log(f"❌ Erro Watcher Loop: {e}") + + await asyncio.sleep(5) + + def abort_current_task(self): + state.log("🛑 Solicitando Cancelamento...") + self.abort_flag = True + if self.current_process: + try: self.current_process.kill() + except: pass + if self.pending_future and not self.pending_future.done(): + self.pending_future.cancel() + + async def scan_folder(self, input_dir: Path): + for file_path in input_dir.glob('**/*'): + if self.abort_flag: + self.abort_flag = False + if state.current_file: + state.update_task(state.current_file, 'error', label=f"{state.current_file} (Cancelado)") + return + + if not self.is_running: return + + if file_path.is_file() and file_path.suffix.lower() in VIDEO_EXTENSIONS: + if file_path.name.startswith('.') or 'processing' in file_path.name: continue + + # Ignora se já terminou nesta sessão + if file_path.name in state.tasks and state.tasks[file_path.name]['status'] == 'done': + continue + + try: + s1 = file_path.stat().st_size + await asyncio.sleep(1) + s2 = file_path.stat().st_size + if s1 != s2: continue + except: continue + + await self.process_pipeline(file_path) + if self.abort_flag: return + + async def process_pipeline(self, filepath: Path): + fname = filepath.name + self.abort_flag = False + state.current_file = fname + + state.update_task(fname, 'running', 0, label=f"Identificando: {fname}...") + state.log(f"🔄 Iniciando: {fname}") + + # 1. IDENTIFICAÇÃO + result = self.renamer.identify_file(str(filepath)) + target_info = None + is_semi_auto = AppConfig.get_val('semi_auto', 'false') == 'true' + + if is_semi_auto: + result['status'] = 'AMBIGUOUS' + if 'match' in result: result['candidates'] = [result['match']] + + if result['status'] == 'MATCH': + target_info = {'tmdb_id': result['match']['tmdb_id'], 'type': result['match']['type']} + state.update_task(fname, 'running', 10, label=f"ID: {result['match']['title']}") + + elif result['status'] == 'AMBIGUOUS': + state.update_task(fname, 'warning', 10, label="Aguardando Telegram...") + self.pending_future = asyncio.ensure_future( + self.bot.ask_user_choice(fname, result['candidates']) + ) + try: + user_choice = await self.pending_future + except asyncio.CancelledError: + state.update_task(fname, 'error', 0, label="Cancelado Manualmente") + return + + self.pending_future = None + if not user_choice or self.abort_flag: + state.update_task(fname, 'skipped', 0, label="Ignorado/Cancelado") + return + + target_info = user_choice + else: + state.update_task(fname, 'error', 0, label="Não Identificado") + state.log(f"❌ Falha TMDb: {result.get('msg', 'Desconhecido')}") + return + + if self.abort_flag: return + + # 2. CATEGORIA + category = self.find_category(target_info['type']) + if not category: + state.update_task(fname, 'error', 0, label="Sem Categoria") + return + + # 3. CONVERSÃO & CAMINHO + try: + # Recupera dados completos para montar o nome + details = self.renamer.get_details(target_info['tmdb_id'], target_info['type']) + + full_details = { + 'title': getattr(details, 'title', getattr(details, 'name', 'Unknown')), + 'year': '0000', + 'type': target_info['type'] + } + d_date = getattr(details, 'release_date', getattr(details, 'first_air_date', '0000')) + if d_date: full_details['year'] = d_date[:4] + + # Gera caminho inteligente + guessed_data = result.get('guessed', {}) + relative_path = self.renamer.build_path(category, full_details, guessed_data) + + temp_filename = os.path.basename(relative_path) + temp_output = self.temp_dir / temp_filename + + state.update_task(fname, 'running', 15, label=f"Convertendo: {full_details['title']}") + + engine = FFmpegEngine() + total_duration = engine.get_duration(str(filepath)) + cmd = engine.build_command(str(filepath), str(temp_output)) + + self.current_process = await asyncio.create_subprocess_exec( + *cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE + ) + + while True: + if self.abort_flag: + self.current_process.kill() + break + line_bytes = await self.current_process.stderr.readline() + if not line_bytes: break + + line = line_bytes.decode('utf-8', errors='ignore') + time_match = re.search(r'time=(\d{2}):(\d{2}):(\d{2})', line) + if time_match and total_duration > 0: + h, m, s = map(int, time_match.groups()) + current_seconds = h*3600 + m*60 + s + pct = 15 + ((current_seconds / total_duration) * 80) + state.update_task(fname, 'running', pct) + + await self.current_process.wait() + + if self.abort_flag: + state.update_task(fname, 'error', 0, label="Abortado") + if temp_output.exists(): os.remove(str(temp_output)) + return + + if self.current_process.returncode != 0: + state.update_task(fname, 'error', 0, label="Erro FFmpeg") + return + self.current_process = None + + # 4. DEPLOY FINAL + state.update_task(fname, 'running', 98, label="Organizando...") + + final_full_path = Path(category.target_path) / relative_path + final_full_path.parent.mkdir(parents=True, exist_ok=True) + + shutil.move(str(temp_output), str(final_full_path)) + + if AppConfig.get_val('deploy_mode', 'move') == 'move': + os.remove(str(filepath)) + + await self.bot.send_notification(f"🎬 Organizado: `{full_details['title']}`") + state.update_task(fname, 'done', 100, label=f"{full_details['title']}") + state.current_file = "" + + # Limpeza pasta vazia + if AppConfig.get_val('cleanup_empty_folders', 'true') == 'true': + try: + parent = filepath.parent + monitor_root = Path(AppConfig.get_val('monitor_path', '/downloads')) + if parent != monitor_root and not any(parent.iterdir()): + parent.rmdir() + except: pass + + except Exception as e: + state.log(f"Erro Pipeline: {e}") + state.update_task(fname, 'error', 0, label=f"Erro: {e}") + + def find_category(self, media_type): + """Encontra a categoria correta baseada nas keywords""" + # Define keywords esperadas + keywords = ['movie', 'film', 'filme'] if media_type == 'movie' else ['tv', 'serie', 'série'] + + all_cats = list(Category.select()) + for cat in all_cats: + if not cat.match_keywords: continue + + # --- AQUI ESTAVA O ERRO POTENCIAL --- + # O .strip() precisa dos parênteses antes do .lower() + cat_keys = [k.strip().lower() for k in cat.match_keywords.split(',')] + + if any(k in cat_keys for k in keywords): + return cat + + # Fallback (primeira categoria que existir) + return all_cats[0] if all_cats else None \ No newline at end of file diff --git a/app/data/clei.db b/app/data/clei.db new file mode 100644 index 0000000..b5f200c Binary files /dev/null and b/app/data/clei.db differ diff --git a/app/data/cleiflow.db b/app/data/cleiflow.db new file mode 100644 index 0000000..55ded29 Binary files /dev/null and b/app/data/cleiflow.db differ diff --git a/app/database.py b/app/database.py new file mode 100755 index 0000000..e80f61c --- /dev/null +++ b/app/database.py @@ -0,0 +1,58 @@ +from peewee import * +from pathlib import Path + +# Garante pasta de dados +data_dir = Path('/app/data') +data_dir.mkdir(parents=True, exist_ok=True) + +db = SqliteDatabase(str(data_dir / 'cleiflow.db')) + +class BaseModel(Model): + class Meta: + database = db + +class AppConfig(BaseModel): + key = CharField(unique=True) + value = TextField() + + @classmethod + def get_val(cls, key, default=''): + try: + return cls.get(cls.key == key).value + except: + return default + + @classmethod + def set_val(cls, key, value): + cls.replace(key=key, value=value).execute() + +class Category(BaseModel): + name = CharField(unique=True) + target_path = CharField() + match_keywords = CharField(null=True) # Ex: movie, film + # NOVO CAMPO: Tipo de conteúdo (movie, series, mixed) + content_type = CharField(default='mixed') + +class FFmpegProfile(BaseModel): + name = CharField() + video_codec = CharField(default='h264_vaapi') + preset = CharField(default='medium') + crf = IntegerField(default=23) + audio_langs = CharField(default='por,eng,jpn') + subtitle_langs = CharField(default='por') + is_active = BooleanField(default=False) + +def init_db(): + db.connect() + db.create_tables([AppConfig, Category, FFmpegProfile], safe=True) + + # Migração segura para adicionar coluna se não existir + try: + db.execute_sql('ALTER TABLE category ADD COLUMN content_type VARCHAR DEFAULT "mixed"') + except: pass # Já existe + + # Perfil padrão se não existir + if FFmpegProfile.select().count() == 0: + FFmpegProfile.create(name="Padrão VAAPI (Intel)", video_codec="h264_vaapi", is_active=True) + + db.close() \ No newline at end of file diff --git a/app/main.py b/app/main.py new file mode 100755 index 0000000..4c14e7a --- /dev/null +++ b/app/main.py @@ -0,0 +1,59 @@ +import asyncio +from nicegui import ui, app +from ui import layout, dashboard, settings, manual_tools +from database import init_db +from core.bot import TelegramManager +from core.watcher import DirectoryWatcher +from core.state import state + +# Inicializa Banco +init_db() + +# Instâncias Globais +bot = TelegramManager() +watcher = DirectoryWatcher(bot) + +# --- LIFECYCLE --- +async def startup(): + # 1. Inicia o Watcher (Ele começa pausado, safe) + asyncio.create_task(watcher.start()) + + # 2. Inicia o Bot com atraso e em background + # Isso evita que a falha de conexão trave a UI + asyncio.create_task(delayed_bot_start()) + +async def delayed_bot_start(): + print("⏳ Aguardando rede estabilizar (5s)...") + await asyncio.sleep(5) + await bot.start() + +async def shutdown(): + await bot.stop() + +app.on_startup(startup) +app.on_shutdown(shutdown) + +# --- ROTAS --- +@ui.page('/') +def index_page(): + ui.colors(primary='#5898d4', secondary='#263238') + ui.page_title('Clei-Flow') + layout.create_interface() + dashboard.show() + +@ui.page('/settings') +def settings_page(): + ui.colors(primary='#5898d4', secondary='#263238') + ui.page_title('Configurações') + layout.create_interface() + settings.show() + +@ui.page('/explorer') +def explorer_page(): + ui.colors(primary='#5898d4', secondary='#263238') + ui.page_title('Explorador') + layout.create_interface() + manual_tools.show() + +if __name__ in {"__main__", "__mp_main__"}: + ui.run(title='Clei-Flow', port=8080, storage_secret='clei-secret', reload=True) \ No newline at end of file diff --git a/app/ui/__init__.py b/app/ui/__init__.py new file mode 100755 index 0000000..e69de29 diff --git a/app/ui/__pycache__/__init__.cpython-311.pyc b/app/ui/__pycache__/__init__.cpython-311.pyc new file mode 100644 index 0000000..ec37aaf Binary files /dev/null and b/app/ui/__pycache__/__init__.cpython-311.pyc differ diff --git a/app/ui/__pycache__/dashboard.cpython-311.pyc b/app/ui/__pycache__/dashboard.cpython-311.pyc new file mode 100644 index 0000000..5429511 Binary files /dev/null and b/app/ui/__pycache__/dashboard.cpython-311.pyc differ diff --git a/app/ui/__pycache__/layout.cpython-311.pyc b/app/ui/__pycache__/layout.cpython-311.pyc new file mode 100644 index 0000000..021fb41 Binary files /dev/null and b/app/ui/__pycache__/layout.cpython-311.pyc differ diff --git a/app/ui/__pycache__/manual_tools.cpython-311.pyc b/app/ui/__pycache__/manual_tools.cpython-311.pyc new file mode 100644 index 0000000..969cd4e Binary files /dev/null and b/app/ui/__pycache__/manual_tools.cpython-311.pyc differ diff --git a/app/ui/__pycache__/settings.cpython-311.pyc b/app/ui/__pycache__/settings.cpython-311.pyc new file mode 100644 index 0000000..e54e7c4 Binary files /dev/null and b/app/ui/__pycache__/settings.cpython-311.pyc differ diff --git a/app/ui/dashboard.py b/app/ui/dashboard.py new file mode 100755 index 0000000..427d259 --- /dev/null +++ b/app/ui/dashboard.py @@ -0,0 +1,138 @@ +from nicegui import ui +from database import AppConfig +from core.state import state + +def show(): + semi_auto_initial = AppConfig.get_val('semi_auto', 'false') == 'true' + + with ui.grid(columns=2).classes('w-full gap-4'): + + # --- COLUNA DA ESQUERDA: CONTROLES --- + with ui.column().classes('w-full gap-4'): + + # Painel de Controle + with ui.card().classes('w-full p-4 border-l-4 border-indigo-500'): + ui.label('🎛️ Painel de Controle').classes('text-xl font-bold mb-4 text-gray-700') + + # Switches + is_running = state.watcher.is_running if state.watcher else False + + def toggle_watcher(e): + if state.watcher: + state.watcher.is_running = e.value + state.log(f"Comando: {'INICIAR' if e.value else 'PAUSAR'}") + + switch_run = ui.switch('Monitoramento Ativo', value=is_running, on_change=toggle_watcher).props('color=green size=lg') + + def toggle_semi_auto(e): + AppConfig.set_val('semi_auto', str(e.value).lower()) + state.log(f"⚠️ Semi-Auto: {e.value}") + + switch_auto = ui.switch('Modo Semi-Automático', value=semi_auto_initial, on_change=toggle_semi_auto).props('color=amber size=lg') + + # Botão Cancelar Global + def cancel_task(): + if state.watcher: + state.watcher.abort_current_task() + ui.notify('Cancelando...', type='warning') + + btn_cancel = ui.button('CANCELAR ATUAL', on_click=cancel_task, icon='cancel').props('color=red').classes('w-full mt-4 hidden') + + # Terminal de Logs + with ui.card().classes('w-full h-64 bg-black text-green-400 font-mono text-xs p-2 overflow-hidden flex flex-col'): + ui.label('>_ Logs').classes('text-gray-500 mb-1 border-b border-gray-800 w-full') + log_container = ui.scroll_area().classes('flex-grow w-full') + log_content = ui.label().style('white-space: pre-wrap; font-family: monospace;') + with log_container: log_content.move(log_container) + + # --- COLUNA DA DIREITA: LISTA DE TAREFAS (Igual ao antigo) --- + with ui.card().classes('w-full h-[80vh] bg-gray-50 flex flex-col p-0'): + # Cabeçalho da Lista + with ui.row().classes('w-full p-4 bg-white border-b items-center justify-between'): + ui.label('📋 Fila de Processamento').classes('text-lg font-bold text-gray-700') + lbl_status_top = ui.label('Ocioso').classes('text-sm text-gray-400') + + # Container da Lista (Onde a mágica acontece) + tasks_container = ui.column().classes('w-full p-2 gap-2 overflow-y-auto flex-grow') + + # --- RENDERIZADOR DA LISTA --- + def render_tasks(): + tasks_container.clear() + + # Se não tiver tarefas + if not state.tasks: + with tasks_container: + ui.label('Nenhuma atividade recente.').classes('text-gray-400 italic p-4') + return + + # Itera sobre as tarefas (reversed para mais recentes no topo) + for fname, data in reversed(state.tasks.items()): + status = data['status'] + pct = data['progress'] + label = data['label'] + + # Estilo baseado no status (Igual ao seu código antigo) + icon = 'circle'; color = 'grey'; spin = False + bg_color = 'bg-white' + + if status == 'pending': + icon = 'hourglass_empty'; color = 'grey' + elif status == 'running': + icon = 'sync'; color = 'blue'; spin = True + bg_color = 'bg-blue-50 border-blue-200 border' + elif status == 'warning': + icon = 'warning'; color = 'orange' + bg_color = 'bg-orange-50 border-orange-200 border' + elif status == 'done': + icon = 'check_circle'; color = 'green' + elif status == 'error': + icon = 'error'; color = 'red' + elif status == 'skipped': + icon = 'block'; color = 'red' + + with tasks_container: + with ui.card().classes(f'w-full p-2 {bg_color} flex-row items-center gap-3'): + # Ícone + if spin: ui.spinner(size='sm').classes('text-blue-500') + else: ui.icon(icon, color=color, size='sm') + + # Conteúdo + with ui.column().classes('flex-grow gap-0'): + ui.label(label).classes('font-bold text-sm text-gray-800 truncate') + ui.label(fname).classes('text-xs text-gray-500 truncate') + + # Barra de Progresso (Só aparece se estiver rodando) + if status == 'running': + with ui.row().classes('w-full items-center gap-2 mt-1'): + ui.linear_progress(value=pct/100, show_value=False).classes('h-2 rounded flex-grow') + ui.label(f"{int(pct)}%").classes('text-xs font-bold text-blue-600') + + # --- LOOP DE ATUALIZAÇÃO --- + def update_ui(): + # 1. Logs + logs = state.get_logs() + log_content.set_text("\n".join(logs[-30:])) + log_container.scroll_to(percent=1.0) + + # 2. Re-renderiza a lista de tarefas + # Nota: O NiceGUI é eficiente, mas para listas muito grandes seria melhor atualizar in-place. + # Como limitamos a 20 itens no state, limpar e redesenhar é rápido e seguro. + render_tasks() + + # 3. Controles Globais + if state.watcher and state.watcher.is_running: + lbl_status_top.text = "Serviço Rodando" + lbl_status_top.classes(replace='text-green-500') + switch_run.value = True + else: + lbl_status_top.text = "Serviço Pausado" + lbl_status_top.classes(replace='text-red-400') + switch_run.value = False + + # 4. Botão Cancelar + if state.current_file: + btn_cancel.classes(remove='hidden') + else: + btn_cancel.classes(add='hidden') + + ui.timer(1.0, update_ui) # Atualiza a cada 1 segundo \ No newline at end of file diff --git a/app/ui/layout.py b/app/ui/layout.py new file mode 100755 index 0000000..078767e --- /dev/null +++ b/app/ui/layout.py @@ -0,0 +1,19 @@ +from nicegui import ui + +def create_interface(): + # Cabeçalho Azul + with ui.header().classes('bg-blue-900 text-white'): + ui.button(on_click=lambda: left_drawer.toggle(), icon='menu').props('flat color=white') + ui.label('Clei-Flow').classes('text-xl font-bold') + + # Menu Lateral + with ui.left_drawer().classes('bg-gray-100').props('width=200') as left_drawer: + ui.label('MENU').classes('text-gray-500 text-xs font-bold mb-2 px-4 pt-4') + + # Helper para criar links + def nav_link(text, target, icon_name): + ui.link(text, target).classes('text-gray-700 hover:text-blue-600 block px-4 py-2 font-medium').props(f'icon={icon_name}') + + nav_link('Dashboard', '/', 'dashboard') + nav_link('Explorador', '/explorer', 'folder') + nav_link('Configurações', '/settings', 'settings') \ No newline at end of file diff --git a/app/ui/manual_tools.py b/app/ui/manual_tools.py new file mode 100755 index 0000000..182bb33 --- /dev/null +++ b/app/ui/manual_tools.py @@ -0,0 +1,2 @@ +from nicegui import ui +# Abas antigas (Renomeador Manual, Encoder Manual) viram ferramentas aqui diff --git a/app/ui/settings.py b/app/ui/settings.py new file mode 100755 index 0000000..0911c05 --- /dev/null +++ b/app/ui/settings.py @@ -0,0 +1,326 @@ +from nicegui import ui +import os +from database import Category, FFmpegProfile, AppConfig + +# Lista de idiomas (ISO 639-2) +ISO_LANGS = { + 'por': 'Português (por)', 'eng': 'Inglês (eng)', 'jpn': 'Japonês (jpn)', + 'spa': 'Espanhol (spa)', 'fra': 'Francês (fra)', 'ger': 'Alemão (ger)', + 'ita': 'Italiano (ita)', 'rus': 'Russo (rus)', 'und': 'Indefinido (und)' +} + +# --- COMPONENTE: SELETOR DE PASTAS (Restrito ao /media) --- +async def pick_folder_dialog(start_path='/media'): + """Abre um modal para escolher pastas, restrito a /media""" + ALLOWED_ROOT = '/media' + + # Garante que começa dentro do permitido + if not start_path or not start_path.startswith(ALLOWED_ROOT): + start_path = ALLOWED_ROOT + + result = {'path': None} + + with ui.dialog() as dialog, ui.card().classes('w-96 h-[500px] flex flex-col'): + ui.label('Selecionar Pasta (/media)').classes('font-bold text-lg mb-2') + path_label = ui.label(start_path).classes('text-xs bg-gray-100 p-2 border rounded w-full break-all font-mono') + + scroll = ui.scroll_area().classes('flex-grow border rounded p-1 mt-2 bg-white') + + async def load_dir(path): + # Segurança extra + if not path.startswith(ALLOWED_ROOT): path = ALLOWED_ROOT + + path_label.text = path + scroll.clear() + + try: + # Botão Voltar (Só aparece se NÃO estiver na raiz permitida) + if path != ALLOWED_ROOT: + parent = os.path.dirname(path) + # Garante que o parent não suba além do permitido + if not parent.startswith(ALLOWED_ROOT): parent = ALLOWED_ROOT + + with scroll: + ui.button('.. (Voltar)', on_click=lambda: load_dir(parent)).props('flat dense icon=arrow_upward align=left w-full') + + # Lista Pastas + with scroll: + # Tenta listar. Se diretório não existir (ex: nome novo), mostra vazio + if os.path.exists(path): + for entry in sorted([e for e in os.scandir(path) if e.is_dir()], key=lambda x: x.name.lower()): + ui.button(entry.name, on_click=lambda p=entry.path: load_dir(p)).props('flat dense icon=folder align=left w-full color=amber-8') + else: + ui.label('Pasta não criada ainda.').classes('text-gray-400 italic p-2') + + except Exception as e: + with scroll: ui.label(f'Erro: {e}').classes('text-red text-xs') + + def select_this(): + result['path'] = path_label.text + dialog.close() + + with ui.row().classes('w-full justify-between mt-auto pt-2'): + ui.button('Cancelar', on_click=dialog.close).props('flat color=grey') + ui.button('Selecionar Esta', on_click=select_this).props('flat icon=check color=green') + + await load_dir(start_path) + await dialog + return result['path'] + +# --- TELA PRINCIPAL --- +def show(): + with ui.column().classes('w-full p-6'): + ui.label('Configurações').classes('text-3xl font-light text-gray-800 mb-4') + + with ui.tabs().classes('w-full') as tabs: + tab_ident = ui.tab('Identificação', icon='search') + tab_cats = ui.tab('Categorias', icon='category') + tab_deploy = ui.tab('Deploy & Caminhos', icon='move_to_inbox') + tab_ffmpeg = ui.tab('Motor (FFmpeg)', icon='movie') + tab_telegram = ui.tab('Telegram', icon='send') + + with ui.tab_panels(tabs, value=tab_ident).classes('w-full bg-gray-50 p-4 rounded border'): + + # --- ABA 1: IDENTIFICAÇÃO --- + with ui.tab_panel(tab_ident): + with ui.card().classes('w-full max-w-2xl mx-auto p-6'): + ui.label('🔍 Configuração do Identificador').classes('text-2xl font-bold mb-4 text-indigo-600') + tmdb_key = AppConfig.get_val('tmdb_api_key', '') + lang = AppConfig.get_val('tmdb_language', 'pt-BR') + confidence = AppConfig.get_val('min_confidence', '90') + + ui.label('API Key do TMDb').classes('font-bold text-sm') + key_input = ui.input(placeholder='Ex: 8a9b...', value=tmdb_key).props('password').classes('w-full mb-4') + ui.markdown('[Clique aqui para pegar sua API Key](https://www.themoviedb.org/settings/api)').classes('text-xs text-blue-500 mb-6') + + with ui.grid(columns=2).classes('w-full gap-4'): + lang_input = ui.input('Idioma', value=lang, placeholder='pt-BR') + conf_input = ui.number('Confiança Auto (%)', value=int(confidence), min=50, max=100) + + def save_ident(): + AppConfig.set_val('tmdb_api_key', key_input.value) + AppConfig.set_val('tmdb_language', lang_input.value) + AppConfig.set_val('min_confidence', str(int(conf_input.value))) + ui.notify('Salvo!', type='positive') + + ui.button('Salvar', on_click=save_ident).props('icon=save color=indigo').classes('w-full mt-6') + +# --- ABA 2: CATEGORIAS (ATUALIZADA) --- + with ui.tab_panel(tab_cats): + ui.label('Bibliotecas e Tipos').classes('text-xl text-gray-700 mb-2') + + cats_container = ui.column().classes('w-full gap-2') + + def load_cats(): + cats_container.clear() + cats = list(Category.select()) + if not cats: ui.label('Nenhuma categoria.').classes('text-gray-400') + + for cat in cats: + # Define ícone e cor baseados no tipo + icon = 'movie' if cat.content_type == 'movie' else 'tv' + if cat.content_type == 'mixed': icon = 'shuffle' + + color_cls = 'bg-blue-50 border-blue-200' + if cat.content_type == 'series': color_cls = 'bg-green-50 border-green-200' + if cat.content_type == 'mixed': color_cls = 'bg-purple-50 border-purple-200' + + with cats_container, ui.card().classes(f'w-full flex-row items-center justify-between p-3 border {color_cls}'): + with ui.row().classes('items-center gap-4'): + ui.icon(icon).classes('text-gray-600') + with ui.column().classes('gap-0'): + ui.label(cat.name).classes('font-bold text-lg') + # Mostra o tipo visualmente + type_map = {'movie': 'Só Filmes', 'series': 'Só Séries', 'mixed': 'Misto (Filmes e Séries)'} + ui.label(f"Tipo: {type_map.get(cat.content_type, 'Misto')} | Tags: {cat.match_keywords}").classes('text-xs text-gray-500') + + ui.button(icon='delete', color='red', on_click=lambda c=cat: delete_cat(c)).props('flat dense') + + def add_cat(): + if not name_input.value: return + try: + Category.create( + name=name_input.value, + target_path=f"/media/{name_input.value}", + match_keywords=keywords_input.value, + content_type=type_select.value # Salva o tipo escolhido + ) + name_input.value = ''; keywords_input.value = '' + ui.notify('Categoria criada!', type='positive') + load_cats() + except Exception as e: ui.notify(f'Erro: {e}', type='negative') + + def delete_cat(cat): + cat.delete_instance() + load_cats() + + # Formulário de Criação + with ui.card().classes('w-full mb-4 bg-gray-100 p-4'): + ui.label('Nova Biblioteca').classes('font-bold text-gray-700') + with ui.row().classes('w-full items-start gap-2'): + name_input = ui.input('Nome (ex: Animes)').classes('w-1/4') + keywords_input = ui.input('Tags (ex: anime, animation)').classes('w-1/4') + + # NOVO SELETOR DE TIPO + type_select = ui.select({ + 'mixed': 'Misto (Filmes e Séries)', + 'movie': 'Apenas Filmes', + 'series': 'Apenas Séries' + }, value='mixed', label='Tipo de Conteúdo').classes('w-1/4') + + ui.button('Adicionar', on_click=add_cat).props('icon=add color=green').classes('mt-2') + + load_cats() + # --- ABA 3: DEPLOY & CAMINHOS --- + with ui.tab_panel(tab_deploy): + + # Helper para o Picker funcionar com categorias ou solto + async def open_picker(input_element): + # Seletor de pastas (inicia onde estiver escrito no input ou na raiz) + start = input_element.value if input_element.value else '/' + selected = await pick_folder_dialog(start) + if selected: input_element.value = selected + + # --- 1. CONFIGURAÇÃO DA ORIGEM (MONITORAMENTO) --- + with ui.card().classes('w-full mb-6 border-l-4 border-amber-500 bg-amber-50'): + ui.label('📡 Origem dos Arquivos').classes('text-lg font-bold mb-2 text-amber-900') + ui.label('Qual pasta o Clei-Flow deve vigiar?').classes('text-xs text-amber-700 mb-2') + + monitor_path = AppConfig.get_val('monitor_path', '/downloads') + + with ui.row().classes('w-full items-center gap-2'): + mon_input = ui.input('Pasta Monitorada (Container)', value=monitor_path).classes('flex-grow font-mono bg-white rounded px-2') + # Botão de Pasta (reutilizando o dialog, mas permitindo sair do /media se quiser, ou ajustamos o dialog depois) + # Nota: O pick_folder_dialog atual trava em /media. Se sua pasta de downloads for fora, + # precisaremos liberar o picker. Por enquanto assumimos que o usuário digita ou está montado. + ui.button(icon='folder', on_click=lambda i=mon_input: open_picker(i)).props('flat dense color=amber-9') + + def save_monitor(): + if not mon_input.value.startswith('/'): + ui.notify('Caminho deve ser absoluto (/...)', type='warning'); return + AppConfig.set_val('monitor_path', mon_input.value) + ui.notify('Pasta de monitoramento salva! (Reinicie se necessário)', type='positive') + + ui.button('Salvar Origem', on_click=save_monitor).classes('mt-2 bg-amber-600 text-white') + + ui.separator() + + # --- 2. REGRAS GLOBAIS DE DEPLOY --- + with ui.card().classes('w-full mb-6 border-l-4 border-indigo-500'): + ui.label('⚙️ Regras de Destino (Deploy)').classes('font-bold mb-2') + deploy_mode = AppConfig.get_val('deploy_mode', 'move') + conflict_mode = AppConfig.get_val('conflict_mode', 'skip') + cleanup = AppConfig.get_val('cleanup_empty_folders', 'true') + + with ui.grid(columns=2).classes('w-full gap-4'): + mode_select = ui.select({'move': 'Mover (Recortar)', 'copy': 'Copiar'}, value=deploy_mode, label='Modo') + conflict_select = ui.select({'skip': 'Ignorar', 'overwrite': 'Sobrescrever', 'rename': 'Renomear Auto'}, value=conflict_mode, label='Conflito') + + cleanup_switch = ui.switch('Limpar pastas vazias na origem', value=(cleanup == 'true')).classes('mt-2') + + def save_global_deploy(): + AppConfig.set_val('deploy_mode', mode_select.value) + AppConfig.set_val('conflict_mode', conflict_select.value) + AppConfig.set_val('cleanup_empty_folders', str(cleanup_switch.value).lower()) + ui.notify('Regras salvas!', type='positive') + ui.button('Salvar Regras', on_click=save_global_deploy).classes('mt-2') + + ui.separator() + + # --- 3. MAPEAMENTO DE DESTINOS --- + ui.label('📂 Mapeamento de Destinos (/media)').classes('text-xl text-gray-700 mt-4') + paths_container = ui.column().classes('w-full gap-2') + + def save_cat_path(cat, new_path): + if not new_path.startswith('/media'): + ui.notify('O caminho deve começar com /media', type='warning'); return + cat.target_path = new_path + cat.save() + ui.notify(f'Caminho de "{cat.name}" salvo!') + + def load_deploy_paths(): + paths_container.clear() + cats = list(Category.select()) + if not cats: return + for cat in cats: + with paths_container, ui.card().classes('w-full p-4 flex-row items-center gap-4'): + with ui.column().classes('min-w-[150px]'): + ui.label(cat.name).classes('font-bold') + ui.icon('arrow_forward', color='gray') + with ui.row().classes('flex-grow items-center gap-2'): + path_input = ui.input(value=cat.target_path).classes('flex-grow font-mono') + ui.button(icon='folder', on_click=lambda i=path_input: open_picker(i)).props('flat dense color=amber-8') + ui.button(icon='save', on_click=lambda c=cat, p=path_input: save_cat_path(c, p.value)).props('flat round color=green') + + load_deploy_paths() + ui.button('Recarregar', on_click=load_deploy_paths, icon='refresh').props('flat dense').classes('mt-2 self-center') + + # --- ABA 4: FFMPEG (Com VAAPI) --- + with ui.tab_panel(tab_ffmpeg): + ui.label('Perfis de Conversão').classes('text-xl text-gray-700') + + # 1. Carrega dados + profiles_query = list(FFmpegProfile.select()) + profiles_dict = {p.id: p.name for p in profiles_query} + + # 2. Ativo + active_profile = next((p for p in profiles_query if p.is_active), None) + initial_val = active_profile.id if active_profile else None + + # 3. Função + def set_active_profile(e): + if not e.value: return + FFmpegProfile.update(is_active=False).execute() + FFmpegProfile.update(is_active=True).where(FFmpegProfile.id == int(e.value)).execute() + ui.notify(f'Perfil "{profiles_dict[e.value]}" ativado!', type='positive') + + # 4. Select + select_profile = ui.select(profiles_dict, value=initial_val, label='Perfil Ativo', on_change=set_active_profile).classes('w-64 mb-6') + + ui.separator() + + # Editor + for p in profiles_query: + with ui.expansion(f"{p.name}", icon='tune').classes('w-full bg-white mb-2 border rounded'): + with ui.column().classes('p-4 w-full'): + with ui.grid(columns=2).classes('w-full gap-4'): + c_name = ui.input('Nome', value=p.name) + c_codec = ui.select({ + 'h264_vaapi': 'Intel VAAPI (Linux/Docker)', + 'h264_qsv': 'Intel QuickSync', + 'h264_nvenc': 'Nvidia NVENC', + 'libx264': 'CPU', + 'copy': 'Copy' + }, value=p.video_codec, label='Codec') + c_preset = ui.select(['fast', 'medium', 'slow'], value=p.preset, label='Preset') + c_crf = ui.number('CRF/QP', value=p.crf, min=0, max=51) + + audios = p.audio_langs.split(',') if p.audio_langs else [] + subs = p.subtitle_langs.split(',') if p.subtitle_langs else [] + c_audio = ui.select(ISO_LANGS, value=audios, multiple=True, label='Áudios').props('use-chips') + c_sub = ui.select(ISO_LANGS, value=subs, multiple=True, label='Legendas').props('use-chips') + + def save_profile(prof=p, n=c_name, c=c_codec, pr=c_preset, cr=c_crf, a=c_audio, s=c_sub): + prof.name = n.value; prof.video_codec = c.value; prof.preset = pr.value + prof.crf = int(cr.value); prof.audio_langs = ",".join(a.value); prof.subtitle_langs = ",".join(s.value) + prof.save() + profiles_dict[prof.id] = prof.name + select_profile.options = profiles_dict + select_profile.update() + ui.notify('Salvo!') + + ui.button('Salvar', on_click=save_profile).classes('mt-2') + + # --- ABA 5: TELEGRAM --- + with ui.tab_panel(tab_telegram): + with ui.card().classes('w-full max-w-lg mx-auto p-6'): + ui.label('🤖 Integração Telegram').classes('text-2xl font-bold mb-4 text-blue-600') + t_token = AppConfig.get_val('telegram_token', '') + t_chat = AppConfig.get_val('telegram_chat_id', '') + token_input = ui.input('Bot Token', value=t_token).props('password').classes('w-full mb-2') + chat_input = ui.input('Chat ID', value=t_chat).classes('w-full mb-6') + def save_telegram(): + AppConfig.set_val('telegram_token', token_input.value) + AppConfig.set_val('telegram_chat_id', chat_input.value) + ui.notify('Salvo!', type='positive') + ui.button('Salvar', on_click=save_telegram).props('icon=save color=blue').classes('w-full') \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml new file mode 100755 index 0000000..a3317be --- /dev/null +++ b/docker-compose.yml @@ -0,0 +1,31 @@ +version: '3.8' + +services: + clei-flow: + build: . + container_name: clei-flow + restart: unless-stopped + ports: + - "8087:8080" + # Mapeamento do Hardware Intel (Para o FFmpeg Core) + devices: + - /dev/dri:/dev/dri + + volumes: + # --- AMBIENTE DE DESENVOLVIMENTO (HOT RELOAD) --- + # Mapeia o código do seu Ubuntu direto para o container + - /home/creidsu/clei-flow/app:/app/app + + # Persistência (Banco de Dados e Configs) + - /home/creidsu/clei-flow/app/data:/app/data + + # --- SUAS MÍDIAS (EDITE AQUI CONFORME SEU PC) --- + # Exemplo: - /home/creidsu/downloads:/downloads + - /media/qbit/download:/downloads + - /media:/media + + environment: + - PUID=1000 # ID do usuário linux (geralmente 1000) + - PGID=1000 + - TZ=America/Sao_Paulo + - PYTHONUNBUFFERED=1 # Garante que os logs apareçam na hora no terminal \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100755 index 0000000..360055d --- /dev/null +++ b/requirements.txt @@ -0,0 +1,7 @@ +nicegui +requests +guessit +tmdbv3api +peewee +python-telegram-bot +httpx \ No newline at end of file