Update app.py

app.py

@@ -26,7 +26,6 @@ def get_file_sha256(file_path):
 # --- 2. HÀM THIẾT KẾ CHỨNG CHỈ (BẢN FRIENDLY BY MINH ĐỨC) ---
 class T4S_PDF(FPDF):
     def header(self):
-        # 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)
@@ -47,37 +46,25 @@ class T4S_PDF(FPDF):
 def create_detailed_pdf(file_name, sha_hash, password, output_path):
     pdf = T4S_PDF(orientation="L", unit="mm", format="A4")
     pdf.add_page()
-    
-    # 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. ĐỐ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, "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. CÔNG NGHỆ
     pdf.ln(10)
     pdf.set_text_color(0, 0, 0)
     pdf.set_font("Helvetica", "B", 16)
     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 = (
-        "- 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!"
-    )
+    tech_info = "- Bam! SHA-512 chạy 10,000 vòng.\n- Vectorized Non-linear Logic.\n- Zero-Knowledge: Không lưu khóa."
     pdf.multi_cell(0, 7, tech_info)
-
-    # 3. MẬT KHẨU
     pdf.ln(10)
     pdf.set_fill_color(240, 240, 240)
     pdf.rect(10, pdf.get_y(), 277, 30, 'F')
@@ -86,26 +73,6 @@ def create_detailed_pdf(file_name, sha_hash, password, output_path):
     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 NHỞ
-    pdf.ln(15)
-    pdf.set_text_color(44, 62, 80)
-    pdf.set_font("Helvetica", "B", 12)
-    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. 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."
-    ))
-
-    # 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)
     return output_path
 
@@ -116,14 +83,13 @@ def stretch_key(password):
     return key[:32]
 
 def process_file(file, password, mode, progress=gr.Progress()):
-    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!"
+    if not encrypt_fn: return None, "❌ Lỗi Secret ES!"
+    if not password: return None, "⚠️ Nhập mật khẩu đi bạn ơi!"
     
     master_key = np.frombuffer(stretch_key(password), dtype=np.uint8)
     input_path = file.name
     original_name = os.path.basename(input_path)
     is_encrypt = (mode == "Mã hóa")
-    
     output_name = original_name + ".impossible" if is_encrypt else (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)
@@ -147,30 +113,30 @@ def process_file(file, password, mode, progress=gr.Progress()):
         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], "✅ Xong! File đã 'hóa vàng' và có luôn hồ sơ bảo mật nhé!"
+            return [output_path, pdf_path], "✅ Đã xong! Check vùng kết quả nhé."
         
-        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)}"
+        return [output_path], "✅ Giải mã thành công!"
+    except Exception as e: return None, f"❌ Lỗi: {str(e)}"
 
-# --- 4. GIAO DIỆN GRADIO ---
+# --- 4. GIAO DIỆN GRADIO (FIXED) ---
 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.
-    """)
+    gr.Markdown("# 🛡️ Titanium4S - Ultimate Audit Edition")
+    
     with gr.Row():
         with gr.Column():
-            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")
+            # ĐÂY LÀ VÙNG UPLOAD DUY NHẤT
+            inp_file = gr.File(label="Kéo file vào đây", file_count="single")
+            inp_pass = gr.Textbox(label="Mật khẩu", type="password", placeholder="Nhập pass...")
+            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_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)
-            
+            # ĐÂY LÀ VÙNG KẾT QUẢ (Đã tắt tính năng tương tác để không bị nhầm là ô upload)
+            out_files = gr.File(label="File nhận lại", interactive=False)
+            status = gr.Textbox(label="Thông báo", interactive=False)
+
     gr.Markdown("---")
-    gr.Markdown("© 2026 Titanium4S Project | Liên hệ: truonghoangminhduc123@gmail.com | Donate: dub.sh/donatemepls")
+    gr.Markdown("© 2026 Titanium4S Project | Liên hệ: truonghoangminhduc123@gmail.com")
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.launch()