Update app.py

app.py

@@ -5,15 +5,17 @@ import hashlib
 from fpdf import FPDF
 from datetime import datetime
 
-# --- 1. SETUP THUẬT TOÁN (Giữ nguyên) ---
+# --- 1. SETUP THUẬT TOÁN (Core Logic) ---
 SECRET_CODE = os.getenv("ES")
 exec_context = {}
 encrypt_fn = decrypt_fn = None
+
 if SECRET_CODE:
     try:
         exec(SECRET_CODE, exec_context)
         encrypt_fn, decrypt_fn = exec_context.get('transform_byte'), exec_context.get('untransform_byte')
-    except Exception as e: print(f"Error: {e}")
+    except Exception as e: 
+        print(f"Error loading ES Logic: {e}")
 
 def get_file_sha256(file_path):
     h = hashlib.sha256()
@@ -21,83 +23,87 @@ def get_file_sha256(file_path):
         while chunk := f.read(8192): h.update(chunk)
     return h.hexdigest()
 
-# --- 2. HÀM THIẾT KẾ CHỨNG CHỈ SIÊU CẤP ---
+# --- 2. HÀM THIẾT KẾ CHỨNG CHỈ (BẢN FRIENDLY BY MINH ĐỨC) ---
 class T4S_PDF(FPDF):
     def header(self):
-        # Vẽ một thanh header màu xanh đậm
+        # Header phong cách hiện đại
         self.set_fill_color(44, 62, 80)
         self.rect(0, 0, 297, 30, 'F')
         self.set_text_color(255, 255, 255)
-        self.set_font("Helvetica", "B", 20)
-        self.cell(0, 10, "TITANIUM4S - QUANTUM SECURE PROTOCOL", ln=True, align="C")
-        self.set_font("Helvetica", "", 10)
-        self.cell(0, 5, "Official Security Audit & Ownership Certificate", ln=True, align="C")
+        self.set_font("Helvetica", "B", 22)
+        self.cell(0, 15, "TITANIUM4S - BIÊN BẢN 'PHONG ẤN' FILE", ln=True, align="C")
+        self.set_font("Helvetica", "I", 10)
+        self.cell(0, 5, "App này làm từ tâm huyết của Minh Đức á! <3", ln=True, align="C")
 
     def footer(self):
-        self.set_y(-20)
-        self.set_font("Helvetica", "I", 8)
-        self.set_text_color(150, 150, 150)
-        self.cell(0, 10, f"Issued by Titanium4S Zero-Trust Engine | Record ID: {hashlib.md5(str(datetime.now()).encode()).hexdigest()[:10].upper()}", align="C")
+        self.set_y(-25)
+        self.set_font("Helvetica", "I", 9)
+        self.set_text_color(100, 100, 100)
+        self.cell(0, 5, "Cần gì cứ nhắn mình: truonghoangminhduc123@gmail.com", ln=True, align="C")
+        self.set_text_color(44, 62, 80)
+        self.set_font("Helvetica", "B", 10)
+        self.cell(0, 7, "Thấy xịn thì đừng quên: dub.sh/donatemepls ;)", ln=True, align="C")
 
 def create_detailed_pdf(file_name, sha_hash, password, output_path):
     pdf = T4S_PDF(orientation="L", unit="mm", format="A4")
     pdf.add_page()
     
-    # 1. PHẦN THÔNG TIN FILE (Bố cục cột)
+    # 1. THÔNG TIN FILE
     pdf.ln(10)
     pdf.set_text_color(0, 0, 0)
     pdf.set_font("Helvetica", "B", 16)
-    pdf.cell(0, 10, "1. PROTECTED ASSET IDENTITY", ln=True)
+    pdf.cell(0, 10, "1. ĐỐI TƯỢNG ĐÃ ĐƯỢC 'XỬ LÝ'", ln=True)
     pdf.set_line_width(0.5); pdf.set_draw_color(44, 62, 80); pdf.line(10, pdf.get_y(), 287, pdf.get_y())
     
     pdf.ln(5)
     pdf.set_font("Helvetica", "", 12)
-    pdf.cell(40, 10, "Filename:", 0); pdf.set_font("Helvetica", "B", 12); pdf.cell(0, 10, file_name, ln=True)
-    pdf.set_font("Helvetica", "", 12)
-    pdf.cell(40, 10, "SHA-256 Fingerprint:", 0); pdf.set_font("Courier", "B", 10); pdf.cell(0, 10, sha_hash, ln=True)
+    pdf.cell(40, 10, "Tên file gốc:", 0); pdf.set_font("Helvetica", "B", 12); pdf.cell(0, 10, file_name, ln=True)
+    pdf.set_font("Helvetica", "I", 10); pdf.set_text_color(100, 100, 100)
+    pdf.cell(0, 5, "À umm, file của bạn giờ thành bột mì (mã hóa) hết rồi nhé :/", ln=True)
     
-    # 2. PHẦN CÔNG NGHỆ (Technology Overview)
+    # 2. CÔNG NGHỆ
     pdf.ln(10)
+    pdf.set_text_color(0, 0, 0)
     pdf.set_font("Helvetica", "B", 16)
-    pdf.cell(0, 10, "2. TECHNOLOGY SPECIFICATION", ln=True)
+    pdf.cell(0, 10, "2. 'MA THUẬT' ĐẰNG SAU", ln=True)
     pdf.set_line_width(0.5); pdf.line(10, pdf.get_y(), 287, pdf.get_y())
     
     pdf.ln(5)
     pdf.set_font("Helvetica", "", 11)
     tech_info = (
-        "- Key Stretching: SHA-512 (10,000 Iterations) - Immunity against ASIC/GPU brute-force.\n"
-        "- Cipher Core: Vectorized Non-linear Transformation (Proprietary ES Logic).\n"
-        "- Protocol: Zero-Knowledge Proof (The system never stores your secret key)."
+        "- Bam! SHA-512 chạy 10,000 vòng để hacker 'khóc thét'.\n"
+        "- Vectorized Non-linear: Chế độ xào nấu dữ liệu siêu cấp của Minh Đức.\n"
+        "- Zero-Knowledge: Mình không giữ khóa của bạn đâu, mất là mình cũng chịu thua!"
     )
     pdf.multi_cell(0, 7, tech_info)
 
-    # 3. PHẦN MẬT KHẨU & CẢNH BÁO (Làm nổi bật)
+    # 3. MẬT KHẨU
     pdf.ln(10)
-    pdf.set_fill_color(254, 249, 231) # Màu vàng nhạt cho vùng password
-    pdf.rect(10, pdf.get_y(), 277, 25, 'F')
+    pdf.set_fill_color(240, 240, 240)
+    pdf.rect(10, pdf.get_y(), 277, 30, 'F')
     pdf.set_y(pdf.get_y() + 5)
-    pdf.set_font("Helvetica", "B", 14); pdf.set_text_color(192, 57, 43)
-    pdf.cell(0, 5, "CRITICAL: YOUR SECRET KEY RECORD", ln=True, align="C")
-    pdf.set_font("Courier", "B", 18)
-    pdf.cell(0, 12, password, ln=True, align="C")
+    pdf.set_font("Helvetica", "B", 13); pdf.set_text_color(231, 76, 60)
+    pdf.cell(0, 5, "MẬT KHẨU ĐÂY NÈ (CẤM QUÊN - QUÊN LÀ NHỊN):", ln=True, align="C")
+    pdf.set_font("Courier", "B", 20)
+    pdf.cell(0, 15, f">> {password} <<", ln=True, align="C")
 
-    # 4. LỜI NHẮC CỦA TITANIUM4S
+    # 4. LỜI NHẮC NHỞ
     pdf.ln(15)
     pdf.set_text_color(44, 62, 80)
     pdf.set_font("Helvetica", "B", 12)
-    pdf.cell(0, 10, "SECURITY COMPLIANCE & WARNINGS:", ln=True)
+    pdf.cell(0, 10, "MẤY CÁI NÀY QUAN TRỌNG NÈ:", ln=True)
     pdf.set_font("Helvetica", "", 10)
     pdf.multi_cell(0, 6, (
-        "1. This PDF is the only proof of ownership. If deleted, there is no backup on any server.\n"
-        "2. The file .impossible is mathematically locked. Brute-forcing would take centuries with current hardware.\n"
-        "3. Please store this certificate in a secure vault or an offline encrypted drive."
+        "1. Cái file PDF này là tấm vé duy nhất để bạn tìm lại dữ liệu. Giữ cho kỹ nha!\n"
+        "2. File .impossible bị khóa bằng toán học rồi, siêu máy tính giải cũng mất vài thế kỷ.\n"
+        "3. Nên cất chứng chỉ này vào ổ cứng rời hoặc in ra rồi giấu dưới gối cho chắc."
     ))
 
-    # Đóng dấu "TOP SECRET" giả lập
-    pdf.set_font("Helvetica", "B", 50)
-    pdf.set_text_color(231, 76, 60)
-    pdf.rotate(25, 200, 150)
-    pdf.text(180, 180, "TOP SECRET")
+    # Dấu ấn riêng
+    pdf.set_font("Helvetica", "B", 40)
+    pdf.set_text_color(200, 200, 200)
+    pdf.rotate(20, 200, 150)
+    pdf.text(170, 185, "MINH DUC SECURED")
     pdf.rotate(0)
 
     pdf.output(output_path)
@@ -110,7 +116,9 @@ def stretch_key(password):
     return key[:32]
 
 def process_file(file, password, mode, progress=gr.Progress()):
-    if not encrypt_fn: return None, "❌ Secret ES Error!"
+    if not encrypt_fn: return None, "❌ Lỗi Secret ES! Kiểm tra biến môi trường."
+    if not password: return None, "⚠️ Nhập mật khẩu đã rồi tính tiếp bạn ơi!"
+    
     master_key = np.frombuffer(stretch_key(password), dtype=np.uint8)
     input_path = file.name
     original_name = os.path.basename(input_path)
@@ -134,28 +142,35 @@ def process_file(file, password, mode, progress=gr.Progress()):
                 key_chunk = master_key[indices % 32]
                 f_out.write(fn(data_arr, key_chunk, indices).astype(np.uint8).tobytes())
                 bytes_processed += len(chunk)
-                progress(bytes_processed / file_size, desc=f"Processing {mode}...")
+                progress(bytes_processed / file_size, desc=f"Đang {mode}...")
 
         if is_encrypt:
             pdf_path = os.path.join(os.path.dirname(input_path), f"Audit_{original_name}.pdf")
             create_detailed_pdf(original_name, file_hash, password, pdf_path)
-            return [output_path, pdf_path], "✅ Đã mã hóa & Xuất hồ sơ bảo mật!"
+            return [output_path, pdf_path], "✅ Xong! File đã 'hóa vàng' và có luôn hồ sơ bảo mật nhé!"
         
-        return [output_path], "✅ Giải mã thành công!"
-    except Exception as e: return None, f"❌ Lỗi: {str(e)}"
+        return [output_path], "✅ Giải mã thành công! Chào mừng file đã quay trở lại."
+    except Exception as e: 
+        return None, f"❌ Có biến rồi: {str(e)}"
 
-# --- 4. GIAO DIỆN ---
-with gr.Blocks(theme=gr.themes.Monochrome(), title="Titanium4S Vault") as demo:
-    gr.Markdown("# 🛡️ Titanium4S - Ultimate Audit Edition")
+# --- 4. GIAO DIỆN GRADIO ---
+with gr.Blocks(theme=gr.themes.Soft(), title="Titanium4S Vault") as demo:
+    gr.Markdown("""
+    # 🛡️ Titanium4S - Ultimate Audit Edition
+    ### Hệ thống bảo mật "Made by Minh Đức" - An toàn tuyệt đối, thân thiện tối đa.
+    """)
     with gr.Row():
         with gr.Column():
-            inp_file = gr.File(label="Target File")
-            inp_pass = gr.Textbox(label="Secret Key", type="password")
-            inp_mode = gr.Radio(["Mã hóa", "Giải mã"], label="Mode", value="Mã hóa")
-            btn = gr.Button("INITIALIZE PROTOCOL", variant="primary")
+            inp_file = gr.File(label="Kéo file vào đây nè")
+            inp_pass = gr.Textbox(label="Mật khẩu bí mật (Đừng để ai thấy nha)", type="password", placeholder="Nhập pass tại đây...")
+            inp_mode = gr.Radio(["Mã hóa", "Giải mã"], label="Bạn muốn làm gì?", value="Mã hóa")
+            btn = gr.Button("KÍCH HOẠT GIAO THỨC", variant="primary")
         with gr.Column():
-            out_files = gr.File(label="Output Assets", file_count="multiple")
-            status = gr.Textbox(label="System Status", interactive=False)
-    btn.click(process_file, [inp_file, inp_pass, inp_mode], [out_files, status])
+            out_files = gr.File(label="Kết quả thu được", file_count="multiple")
+            status = gr.Textbox(label="Trạng thái hệ thống", interactive=False)
+            
+    gr.Markdown("---")
+    gr.Markdown("© 2026 Titanium4S Project | Liên hệ: truonghoangminhduc123@gmail.com | Donate: dub.sh/donatemepls")
 
-demo.queue(1).launch(ssr_mode=False)
+if __name__ == "__main__":
+    demo.queue().launch()