Update app.py

app.py

@@ -3,7 +3,16 @@ import gradio as gr
 import numpy as np
 import hashlib
 
-# --- 1. NẠP LINH HỒN THUẬT TOÁN ---
+# --- CẤU HÌNH METADATA TITANIUM4S ---
+# Dùng để cảnh báo và định danh file của hệ thống
+T4S_MARKER = b"\n---TITANIUM4S-SECURITY-FOOTER---"
+T4S_POLICY = (
+    "\n[WARNING] Encrypted by Titanium4S Protocol."
+    "\n[POLICY] Data recovery is IMPOSSIBLE without the secret key."
+    "\n[NOTICE] This file contains encrypted binary data. DO NOT MODIFY."
+).encode('utf-8')
+
+# --- 1. NẠP THUẬT TOÁN TỪ BIẾN MÔI TRƯỜNG ES ---
 SECRET_CODE = os.getenv("ES")
 exec_context = {}
 encrypt_fn = None
@@ -19,26 +28,27 @@ if SECRET_CODE:
 
 # --- 2. CƠ CHẾ BẢO MẬT (KEY STRETCHING) ---
 def stretch_key(password, salt=b"MinhDuc_Zero_Trust_Protocol"):
-    """ Băm SHA-512 liên tục 10,000 lần để bảo vệ mật khẩu """
+    """ Băm mật khẩu 10,000 lần để chống Brute-force """
     key = password.encode() + salt
     for _ in range(10000):
         key = hashlib.sha512(key).digest()
     return key[:32]
 
-# --- 3. LOGIC XỬ LÝ FILE ---
+# --- 3. LOGIC XỬ LÝ CHÍNH ---
 def process_file(file, password, mode, progress=gr.Progress()):
     if not encrypt_fn or not decrypt_fn:
-        return None, "❌ Lỗi: Chưa nạp Secret ES trong Settings!"
+        return None, "❌ Lỗi: Chưa cấu hình Secret ES trong Settings!"
     if not file or not password:
-        return None, "⚠️ Điền đủ file và mật khẩu nhé Minh Đức!"
+        return None, "⚠️ Minh Đức ơi, hãy chọn file và nhập mật khẩu nhé!"
 
-    # Khởi tạo Master Key
+    # Khởi tạo khóa từ mật khẩu
     master_key = np.frombuffer(stretch_key(password), dtype=np.uint8)
     
     input_path = file.name
     original_name = os.path.basename(input_path)
+    file_size = os.path.getsize(input_path)
     
-    # Xử lý tên file
+    # Thiết lập chế độ và tên file đầu ra
     is_encrypt = (mode == "Mã hóa")
     if is_encrypt:
         output_name = original_name + ".impossible"
@@ -46,53 +56,74 @@ def process_file(file, password, mode, progress=gr.Progress()):
         output_name = original_name.replace(".impossible", "") if original_name.endswith(".impossible") else "rebuilt_" + original_name
     
     output_path = os.path.join(os.path.dirname(input_path), output_name)
-    file_size = os.path.getsize(input_path)
-    chunk_size = 4 * 1024 * 1024  # 4MB tối ưu
+    chunk_size = 4 * 1024 * 1024  # 4MB chunk
     bytes_processed = 0
     fn = encrypt_fn if is_encrypt else decrypt_fn
 
     try:
         with open(input_path, 'rb') as f_in, open(output_path, 'wb') as f_out:
-            while True:
-                chunk = f_in.read(chunk_size)
-                if not chunk:
-                    break
+            # --- XỬ LÝ KÍCH THƯỚC MỤC TIÊU (LOẠI BỎ METADATA KHI GIẢI MÃ) ---
+            target_data_size = file_size
+            if not is_encrypt:
+                # Kiểm tra 2KB cuối cùng để tìm Marker
+                check_range = min(file_size, 2048)
+                f_in.seek(file_size - check_range)
+                footer_data = f_in.read(check_range)
+                marker_pos = footer_data.find(T4S_MARKER)
+                if marker_pos != -1:
+                    # Tính toán lại kích thước thật (bỏ qua phần Footer)
+                    target_data_size = file_size - (check_range - marker_pos)
+                f_in.seek(0) # Quay lại đầu file để bắt đầu giải mã
+
+            # --- VÒNG LẶP XỬ LÝ DỮ LIỆU ---
+            while bytes_processed < target_data_size:
+                read_len = min(chunk_size, target_data_size - bytes_processed)
+                chunk = f_in.read(read_len)
+                if not chunk: break
                 
                 data_arr = np.frombuffer(chunk, dtype=np.uint8)
                 indices = np.arange(bytes_processed, bytes_processed + len(data_arr), dtype=np.uint64)
                 key_chunk = master_key[indices % 32]
                 
-                # Thực thi biến đổi Vectorized
                 processed = fn(data_arr, key_chunk, indices)
-                
                 f_out.write(processed.astype(np.uint8).tobytes())
+                
                 bytes_processed += len(chunk)
-                progress(bytes_processed / file_size, desc=f"Đang {mode.lower()}...")
+                progress(bytes_processed / target_data_size, desc=f"Đang {mode.lower()}...")
+
+            # --- CHÈN FOOTER KHI MÃ HÓA XONG ---
+            if is_encrypt:
+                f_out.write(T4S_MARKER)
+                f_out.write(T4S_POLICY)
 
-        return output_path, f"✅ Thành công! File lưu tại: {output_name}"
+        return output_path, f"✅ {mode} thành công! Metadata Titanium4S đã được xử lý."
     except Exception as e:
         if os.path.exists(output_path): os.remove(output_path)
         return None, f"❌ Lỗi: {str(e)}"
 
-# --- 4. GIAO DIỆN (THEME DARK/MONOCHROME) ---
-with gr.Blocks(theme=gr.themes.Monochrome(), title="Titanium4S Vault") as demo:
-    gr.Markdown("# 🛡️ Titanium4S - Professional Vault")
-    gr.Markdown("Hệ thống bảo mật dữ liệu cấp cao dành riêng cho Minh Đức.")
+# --- 4. GIAO DIỆN GRADIO (FIXED FOR 6.0+) ---
+with gr.Blocks(title="Titanium4S Vault") as demo:
+    gr.Markdown("# 🛡️ Titanium4S - Professional Vault v5.0")
+    gr.Markdown("Hệ thống bảo mật thép với cơ chế đóng dấu Metadata bảo toàn dữ liệu.")
     
     with gr.Row():
         with gr.Column():
             inp_file = gr.File(label="Tải file lên")
-            inp_pass = gr.Textbox(label="Mật khẩu bí mật", type="password")
+            inp_pass = gr.Textbox(label="Mật khẩu bí mật", type="password", placeholder="Nhập khóa để kích hoạt...")
             inp_mode = gr.Radio(["Mã hóa", "Giải mã"], label="Chế độ", value="Mã hóa")
             btn = gr.Button("KÍCH HOẠT", variant="primary")
         with gr.Column():
-            out_file = gr.File(label="Kết quả")
+            out_file = gr.File(label="Dữ liệu kết quả")
             status = gr.Textbox(label="Trạng thái", interactive=False)
 
     btn.click(process_file, [inp_file, inp_pass, inp_mode], [out_file, status])
 
-# Giới hạn 1 người dùng xử lý 1 lúc để tránh crash RAM
+# Cấu hình Queue
 demo.queue(default_concurrency_limit=1)
 
 if __name__ == "__main__":
-    demo.launch()
+    # Launch với theme Monochrome và tắt SSR để ổn định
+    demo.launch(
+        theme=gr.themes.Monochrome(),
+        ssr_mode=False
+    )