Update app.py

app.py

@@ -31,11 +31,11 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Cria pastas
+# Cria as pastas se não existirem
 for cat in CATEGORIES:
     os.makedirs(os.path.join(BASE_FOLDER, cat), exist_ok=True)
 
-# Carrega expirações
+# Carrega dados de expiração
 if os.path.exists(EXPIRATION_FILE):
     with open(EXPIRATION_FILE, "r") as f:
         expirations = json.load(f)
@@ -44,11 +44,12 @@ else:
 
 st.title("📂 File Manager por Categoria")
 
-# --- Função remover expirados ---
+# --- Função: apagar arquivos expirados ---
 def remove_expired_files():
     changed = False
     now = time.time()
     expired_files = []
+
     for file_full, expire_time in list(expirations.items()):
         cat, file = file_full.split("|||")
         file_path = os.path.join(BASE_FOLDER, cat, file)
@@ -57,16 +58,19 @@ def remove_expired_files():
                 os.remove(file_path)
                 expired_files.append(file_full)
                 changed = True
+
     for file_full in expired_files:
         expirations.pop(file_full)
+
     if changed:
         with open(EXPIRATION_FILE, "w") as f:
             json.dump(expirations, f)
 
+# --- Apagar arquivos vencidos ao iniciar ---
 remove_expired_files()
 
-# --- Função aleatorizar metadados com ffmpeg ---
-def randomize_video_metadata(input_path, output_path, progress_placeholder):
+# --- Função: randomizar metadados de vídeo ---
+def randomize_video_metadata_ffmpeg(input_path, output_path):
     major_brands = ["isom", "mp42", "iso5", "avc1"]
     compatible_sets = [
         ["isom", "iso2", "avc1", "mp41"],
@@ -77,10 +81,12 @@ def randomize_video_metadata(input_path, output_path, progress_placeholder):
     major = random.choice(major_brands)
     compatible = ",".join(random.choice(compatible_sets))
 
-    # Gera datas aleatórias
+    # Data aleatória nos últimos 6 meses
     start_date = datetime.datetime.now() - datetime.timedelta(days=180)
-    random_date = start_date + datetime.timedelta(days=random.randint(0, 180),
-                                                  seconds=random.randint(0, 86400))
+    random_date = start_date + datetime.timedelta(
+        days=random.randint(0, 180),
+        seconds=random.randint(0, 86400)
+    )
     date_str = random_date.strftime("%Y-%m-%d %H:%M:%S")
 
     def randstr(n=12): 
@@ -96,33 +102,36 @@ def randomize_video_metadata(input_path, output_path, progress_placeholder):
         "publisher": randstr(10),
         "encoded_by": randstr(10),
         "creation_time": date_str,
-        "major_brand": major,
-        "compatible_brands": compatible,
     }
 
-    # Monta comando ffmpeg
-    cmd = ["ffmpeg", "-i", input_path, "-map_metadata", "-1"]
+    cmd = [
+        "ffmpeg", "-i", input_path,
+        "-map_metadata", "-1",   # remove metadados antigos
+        "-metadata", f"major_brand={major}",
+        "-metadata", f"compatible_brands={compatible}",
+    ]
+
     for k, v in metadata.items():
         cmd += ["-metadata", f"{k}={v}"]
 
-    # Mantém qualidade alta
-    cmd += ["-c:v", "libx264", "-preset", "fast", "-crf", "18",
-            "-c:a", "aac", "-b:a", "192k", output_path]
+    cmd += [
+        "-c:v", "libx264", "-preset", "fast", "-crf", "18",
+        "-c:a", "aac", "-b:a", "192k",
+        output_path
+    ]
 
-    # Executa com barra de progresso
-    progress = 0
-    with subprocess.Popen(cmd, stderr=subprocess.PIPE, universal_newlines=True) as proc:
-        for line in proc.stderr:
-            if "time=" in line:
-                progress = min(progress + 5, 100)  # fake progress
-                progress_placeholder.progress(progress)
-        proc.wait()
-    progress_placeholder.progress(100)
+    subprocess.run(cmd, check=True)
 
 # --- Upload ---
+
 st.header("📤 Upload de Arquivos")
+
+st.subheader("Selecione uma categoria:")
+
 categoria = st.radio("Categoria:", CATEGORIES, index=None)
-aleatorizar = st.checkbox("🔀 Aleatorizar metadados de vídeos")
+
+# Checkbox para randomização
+apply_metadata = st.checkbox("🔀 Aleatorizar metadados de vídeos")
 
 if categoria:
     uploaded_files = st.file_uploader(
@@ -137,19 +146,23 @@ if categoria:
             folder = os.path.join(BASE_FOLDER, categoria)
             file_path = os.path.join(folder, uploaded_file.name)
 
-            with open(file_path, "wb") as f:
+            # Salva temporário
+            temp_path = file_path + ".tmp"
+            with open(temp_path, "wb") as f:
                 f.write(uploaded_file.read())
 
-            # Se for vídeo e checkbox ativo → processa metadados
-            if aleatorizar and uploaded_file.type == "video/mp4":
-                progress_placeholder = st.empty()
-                st.info(f"🔄 Processando metadados de {uploaded_file.name}...")
-                randomized_path = os.path.join(folder, f"rnd_{uploaded_file.name}")
-                randomize_video_metadata(file_path, randomized_path, progress_placeholder)
-                os.remove(file_path)  # remove original
-                os.rename(randomized_path, file_path)
-                st.success(f"✅ {uploaded_file.name} processado com metadados aleatórios!")
-
+            # Se for vídeo e checkbox marcado, randomiza
+            if apply_metadata and uploaded_file.name.lower().endswith((".mp4", ".mov", ".mkv")):
+                try:
+                    randomize_video_metadata_ffmpeg(temp_path, file_path)
+                    os.remove(temp_path)
+                except Exception as e:
+                    st.error(f"Erro ao processar metadados: {e}")
+                    os.rename(temp_path, file_path)
+            else:
+                os.rename(temp_path, file_path)
+
+            # Salvar expiração se marcada
             if auto_delete:
                 key = f"{categoria}|||{uploaded_file.name}"
                 expirations[key] = time.time() + 24 * 60 * 60
@@ -159,31 +172,42 @@ if categoria:
         st.success("Arquivos enviados com sucesso!")
         st.rerun()
 
-# --- Lista de Arquivos ---
+# --- Lista de Arquivos agrupada por pasta ---
 st.header("📄 Arquivos Disponíveis")
+
 for categoria in CATEGORIES:
     folder = os.path.join(BASE_FOLDER, categoria)
     files = os.listdir(folder)
+
     st.subheader(f"📁 {categoria}")
+
     if not files:
         st.info("Nenhum arquivo na categoria.")
     else:
         for file in files:
             st.markdown(f"<div class='file-box'>", unsafe_allow_html=True)
             st.markdown(f"<div class='file-name'>{file}</div>", unsafe_allow_html=True)
+
             key = f"{categoria}|||{file}"
+
             expira = expirations.get(key)
             if expira:
                 restante = int(expira - time.time())
                 if restante > 0:
-                    h = restante // 3600
-                    m = (restante % 3600) // 60
-                    st.markdown(f"<div class='expire-text'>⏳ Expira em {h}h {m}min</div>", unsafe_allow_html=True)
+                    restante_horas = restante // 3600
+                    restante_min = (restante % 3600) // 60
+                    st.markdown(
+                        f"<div class='expire-text'>⏳ Expira em {restante_horas}h {restante_min}min</div>",
+                        unsafe_allow_html=True)
+                else:
+                    st.markdown("<div class='expire-text'>⚠ Expiração iminente</div>", unsafe_allow_html=True)
 
             col1, col2, col3 = st.columns(3)
+
             with col1:
                 with open(os.path.join(folder, file), "rb") as f_obj:
                     st.download_button("⬇ Download", f_obj, file_name=file, key=f"down_{categoria}_{file}")
+
             with col2:
                 if st.button("🗑 Excluir", key=f"delete_{categoria}_{file}"):
                     os.remove(os.path.join(folder, file))
@@ -192,6 +216,7 @@ for categoria in CATEGORIES:
                         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:
                     if st.download_button("⬇ Baixar & Apagar", f_obj, file_name=file, key=f"download_delete_{categoria}_{file}"):
@@ -201,4 +226,5 @@ for categoria in CATEGORIES:
                             json.dump(expirations, f)
                         st.success(f"Arquivo '{file}' baixado e removido.")
                         st.rerun()
+
             st.markdown("</div>", unsafe_allow_html=True)