File size: 2,238 Bytes
4f263b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import gradio as gr
from cryptography.fernet import Fernet
# AES 加密函数（Fernet 实现）
def encrypt_file_gr(file):
# `file` 是一个 NamedTemporaryFile 对象，带有属性 `name`
file_path = file.name
# 生成密钥并初始化 cipher
key = Fernet.generate_key()
cipher = Fernet(key)
# 读取原文件并加密
with open(file_path, 'rb') as f:
data = f.read()
encrypted_data = cipher.encrypt(data)
# 写入加密文件
enc_path = file_path + '.enc'
with open(enc_path, 'wb') as f:
f.write(encrypted_data)
# 返回密钥和加密文件路径
return key.decode(), enc_path
# AES 解密函数
def decrypt_file_gr(file, key):
file_path = file.name
cipher = Fernet(key.encode())
try:
with open(file_path, 'rb') as f:
encrypted_data = f.read()
decrypted_data = cipher.decrypt(encrypted_data)
dec_path = file_path.replace('.enc', '.dec')
with open(dec_path, 'wb') as f:
f.write(decrypted_data)
return dec_path
except Exception as e:
# 如果解密失败，返回空，让 Gradio 显示错误
return None
# Gradio 界面设计
with gr.Blocks() as demo:
gr.Markdown("# 文件加密解密工具")
with gr.Tab("加密"):
encrypt_in = gr.File(label="上传文件")
encrypt_btn = gr.Button("加密文件 🔒")
encrypt_key = gr.Textbox(label="生成的密钥", interactive=False)
encrypt_out = gr.File(label="下载加密文件 (.enc)")
encrypt_btn.click(
fn=encrypt_file_gr,
inputs=encrypt_in,
outputs=[encrypt_key, encrypt_out]
)
with gr.Tab("解密"):
decrypt_in = gr.File(label="上传加密文件 (.enc)")
decrypt_key_in = gr.Textbox(label="输入密钥")
decrypt_btn = gr.Button("解密文件 🔓")
decrypt_out = gr.File(label="下载解密文件 (.dec)")
decrypt_btn.click(
fn=decrypt_file_gr,
inputs=[decrypt_in, decrypt_key_in],
outputs=decrypt_out
)
# 本地服务器启动函数，Hugging Face Spaces 会自动调用 demo.launch()
if __name__ == "__main__":
demo.launch()
|