rar2zip / app.py
ThorAILabs's picture
Fix FileNotFound error
26297ba verified
import gradio as gr
import rarfile
import pyzipper
from io import BytesIO
def convert_rar_to_zip(rar_file, password=None, progress=gr.Progress()):
"""Convert password-protected RAR to encrypted ZIP with progress tracking"""
try:
progress(0, desc="πŸ”“ Initializing conversion...")
# Corrected password parameter
with rarfile.RarFile(rar_file.name,
password=password.encode() if password else None) as rar:
total_files = len(rar.infolist())
zip_buffer = BytesIO()
with pyzipper.AESZipFile(zip_buffer, 'w', compression=pyzipper.ZIP_LZMA) as zipf:
if password:
zipf.setpassword(password.encode())
for i, file_info in enumerate(progress.tqdm(rar.infolist(), desc="Processing")):
if file_info.is_dir():
zipf.writestr(file_info.filename + '/', '')
else:
with rar.open(file_info) as file:
content = file.read()
zipf.writestr(file_info.filename, content,
compress_type=pyzipper.ZIP_LZMA,
encrypt=bool(password))
progress((i+1)/total_files, desc=f"πŸ”’ Processed {i+1}/{total_files} files")
zip_buffer.seek(0)
return ("converted.zip", zip_buffer.read())
except Exception as e:
return f"❌ Error: {str(e)}"
app = gr.Interface(
fn=convert_rar_to_zip,
inputs=[
gr.File(label="Upload RAR", file_types=[".rar"]),
gr.Textbox(label="Password (optional)", type="password")
],
outputs=gr.File(label="Download Encrypted ZIP"),
title="Secure RAR to ZIP Converter",
description="Convert password-protected RAR archives to AES-256 encrypted ZIP files",
allow_flagging="never"
)
app.launch()