Update app.py

app.py

@@ -5,6 +5,8 @@ import time
 import subprocess
 import random
 import string
+import tempfile
+import shutil
 import mimetypes
 
 BASE_FOLDER = "uploaded_files"
@@ -46,43 +48,8 @@ else:
 st.title("📂 File Manager por Categoria")
 
 # --- Funções auxiliares ---
-def random_string(n=12):
-    return ''.join(random.choices(string.ascii_letters + string.digits, k=n))
-
-def randomize_video_metadata(file_path):
-    """FULL HARD REAL: remove LvMetaInfo/Aigc e reinsere metadados randômicos"""
-    mime, _ = mimetypes.guess_type(file_path)
-    if not mime or not mime.startswith("video"):
-        return
-
-    temp_path = file_path + ".tmp"
-
-    cmd = [
-        "ffmpeg", "-y", "-i", file_path,
-        "-map", "0:v", "-map", "0:a",   # ⚡ só vídeo e áudio
-        "-map_metadata", "-1",          # limpa tudo
-        "-c:v", "libx264", "-preset", "fast", "-crf", "23",
-        "-c:a", "aac", "-b:a", "128k",
-        "-movflags", "use_metadata_tags+faststart",
-        "-brand", random.choice(["mp42", "isom", "iso2", "avc1"]),
-        "-metadata", f"encoder={random_string(10)}",
-        "-metadata", f"title={random_string(12)}",
-        "-metadata", f"comment={random_string(12)}",
-        "-metadata", f"description={random_string(12)}",
-        "-metadata", f"artist={random_string(10)}",
-        "-metadata", f"album={random_string(10)}",
-        temp_path
-    ]
-
-    try:
-        subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-        os.replace(temp_path, file_path)
-        print(f"✅ FULL HARD REAL aplicado em {file_path}")
-    except Exception as e:
-        print(f"⚠ Erro ao randomizar {file_path}: {e}")
-
 def remove_expired_files():
-    """Remove arquivos vencidos automaticamente"""
+    """Remove arquivos expirados automaticamente"""
     changed = False
     now = time.time()
     expired_files = []
@@ -103,6 +70,48 @@ def remove_expired_files():
         with open(EXPIRATION_FILE, "w") as f:
             json.dump(expirations, f)
 
+def random_string(length=12):
+    return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
+
+def process_video(file_path, hard_mode=False):
+    """Processa vídeo removendo metadados. Se hard_mode=True, usa FFmpeg + MP4Box"""
+    temp_fd, temp_out = tempfile.mkstemp(suffix=".mp4")
+    os.close(temp_fd)
+
+    if not hard_mode:
+        # Upload normal sem mexer
+        shutil.copy(file_path, temp_out)
+        return temp_out
+
+    # 1. Reencode básico com ffmpeg
+    temp_ffmpeg = temp_out + ".ffmpeg.mp4"
+    cmd_ffmpeg = [
+        "ffmpeg", "-y", "-i", file_path,
+        "-map", "0:v", "-map", "0:a?",
+        "-map_metadata", "-1",
+        "-c:v", "libx264", "-preset", "fast", "-crf", "23",
+        "-c:a", "aac", "-b:a", "128k",
+        "-movflags", "use_metadata_tags+faststart",
+        "-brand", random.choice(["mp42", "isom", "iso2", "avc1"]),
+        "-metadata", f"encoder={random_string(10)}",
+        "-metadata", f"title={random_string(12)}",
+        "-metadata", f"comment={random_string(12)}",
+        "-metadata", f"description={random_string(12)}",
+        "-metadata", f"artist={random_string(10)}",
+        "-metadata", f"album={random_string(10)}",
+        temp_ffmpeg
+    ]
+    subprocess.run(cmd_ffmpeg, check=True)
+
+    # 2. Remove UUID atoms (LvMetaInfo, AigcInfo) com MP4Box
+    cmd_mp4box = ["MP4Box", "-rem-uuid", temp_ffmpeg, "-out", temp_out]
+    subprocess.run(cmd_mp4box, check=True)
+
+    # 3. Apaga temporário
+    os.remove(temp_ffmpeg)
+
+    return temp_out
+
 # --- Apagar arquivos vencidos ao iniciar ---
 remove_expired_files()
 
@@ -112,12 +121,7 @@ st.subheader("Selecione uma categoria:")
 
 categoria = st.radio("Categoria:", CATEGORIES, index=None)
 
-# 🔀 Opções antes do upload
-randomize_meta = st.checkbox("🔀 Randomizar metadados de vídeos (e renomear)")
-auto_delete = st.checkbox("🕒 Excluir automaticamente após 24 horas")
-
-if "uploaded_once" not in st.session_state:
-    st.session_state["uploaded_once"] = False
+hard_mode = st.checkbox("Ativar randomização HARD (vídeos apenas)")
 
 if categoria:
     uploaded_files = st.file_uploader(
@@ -125,37 +129,44 @@ if categoria:
         accept_multiple_files=True,
         key=f"uploader_{categoria}"
     )
+    auto_delete = st.checkbox("Excluir automaticamente após 24 horas")
 
-    if uploaded_files and not st.session_state["uploaded_once"]:
+    if uploaded_files:
         for uploaded_file in uploaded_files:
             folder = os.path.join(BASE_FOLDER, categoria)
+            os.makedirs(folder, exist_ok=True)
+
+            mime, _ = mimetypes.guess_type(uploaded_file.name)
+            is_video = mime and mime.startswith("video")
 
-            # Renomear se randomização estiver ativa
-            if randomize_meta:
-                ext = os.path.splitext(uploaded_file.name)[1]
-                new_name = random_string(16) + ext
+            with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]) as tmp:
+                tmp.write(uploaded_file.read())
+                tmp_path = tmp.name
+
+            # Se for vídeo + hard_mode ativo → processa
+            if is_video and hard_mode:
+                final_tmp = process_video(tmp_path, hard_mode=True)
+            else:
+                final_tmp = tmp_path
+
+            # Renomeia com string aleatória se hard_mode
+            if hard_mode and is_video:
+                new_name = random_string(16) + ".mp4"
             else:
                 new_name = uploaded_file.name
 
             file_path = os.path.join(folder, new_name)
+            shutil.move(final_tmp, file_path)
 
-            # Salva arquivo
-            with open(file_path, "wb") as f:
-                f.write(uploaded_file.read())
-
-            # Se marcado → randomizar metadados (apenas vídeos)
-            if randomize_meta:
-                randomize_video_metadata(file_path)
-
-            # Salvar expiração se marcada
+            # Salvar expiração
             if auto_delete:
                 key = f"{categoria}|||{new_name}"
                 expirations[key] = time.time() + 24 * 60 * 60
                 with open(EXPIRATION_FILE, "w") as f:
                     json.dump(expirations, f)
 
-        st.session_state["uploaded_once"] = True
-        st.success("✅ Arquivos enviados com sucesso!")
+        st.success("Arquivos enviados com sucesso!")
+        st.rerun()
 
 # --- Lista de Arquivos agrupada por pasta ---
 st.header("📄 Arquivos Disponíveis")
@@ -200,6 +211,7 @@ for categoria in CATEGORIES:
                     with open(EXPIRATION_FILE, "w") as f:
                         json.dump(expirations, f)
                     st.success(f"Arquivo '{file}' excluído.")
+                    st.rerun()
 
             with col3:
                 with open(os.path.join(folder, file), "rb") as f_obj:
@@ -209,5 +221,6 @@ for categoria in CATEGORIES:
                         with open(EXPIRATION_FILE, "w") as f:
                             json.dump(expirations, f)
                         st.success(f"Arquivo '{file}' baixado e removido.")
+                        st.rerun()
 
             st.markdown("</div>", unsafe_allow_html=True)