gui.py

import tkinter as tk
from tkinter import filedialog, messagebox
from crypto import encrypt_file, decrypt_file, get_file_hash, write_log

class App:
    def __init__(self,root):
        root.title("Secure File Encryption")
        root.geometry("420x320")
        root.resizable(False, False)

        self.selected_file = tk.StringVar()
        self.password = tk.StringVar()
         
        #File selection row
        tk.Label(root, text="Selected file:", anchor="w").pack(
            fill="x", padx=20, pady=(20,0))
        
        file_frame = tk.Frame(root)
        file_frame.pack(fill="x", padx=20, pady=4)

        tk.Entry(file_frame, textvariable=self.selected_file,
                 state="readonly").pack(side="left", fill="x", expand=True)
        
        tk.Button(file_frame, text="Browse",
                  command=self.browse_file).pack(side="right", padx=(6, 0))
        
        #Password row
        tk.Label(root, text="Password:", anchor="w").pack(
            fill="x", padx=20, pady=(12, 0))
        
        tk.Entry(root, textvariable=self.password,
                 show="*").pack(fill="x", padx=20, pady=4)
        

        #Buttons
        btn_frame = tk.Frame(root)
        btn_frame.pack(pady=20)

        tk.Button(btn_frame, text="Encrypt", width=14, bg="#185FA5", fg="white", command=self.encrypt).pack(side="left", padx=8)

        tk.Button(btn_frame, text="Decrypt", width=14, bg="#0F6E56", fg="white", command=self.decrypt).pack(side="left", padx=8)

        #Status label at the bottom
        self.status = tk.StringVar()
        tk.Label(root, textvariable=self.status, fg="gray", wraplength=380).pack(pady=(0,16))

    def browse_file(self):
        path = filedialog.askopenfilename(
            title="Select a file",
            filetypes=[("All files", "*.*")]
        )
        if path:
            self.selected_file.set(path)
            self.status.set("")

    def encrypt(self):
        src = self.selected_file.get()
        pwd = self.password.get()

        if not src or not pwd:
            messagebox.showwarning("Missing input", "Please select a file and enter a password.")
            return 
        output = src + ".enc"

        try:
            original_hash = get_file_hash(src)
            duration = encrypt_file(src, output, pwd)
            write_log("ENCRYPTED", src, duration)

            self.status.set(f"Encrypted in {duration}s | SHA-256: {original_hash[:20]}...")
            messagebox.showinfo("Success", f"File encrypted in {duration}s\nSHA-256: {original_hash[:20]}...")


        except Exception as e:
            messagebox.showerror("Error", str(e))

    def decrypt(self):
        src = self.selected_file.get()
        pwd = self.password.get()

        if not src or not pwd:
            messagebox.showwarning("Missing input", "Please select a file and enter a password.")
            return 
        
        if not src.endswith(".enc"):
            messagebox.showwarning("Wrong file", "Please select a .enc file to decrypt.")
            return 
        output = src.replace(".enc", ".decrypted")

        try:
            duration = decrypt_file(src, output, pwd)
            restored_hash = get_file_hash(output)
            write_log("DECRYPTED", src, duration)
            
            self.status.set(f"Decrypted in {duration}s |SHA-256: {restored_hash[:20]}...")
            messagebox.showinfo("Success", f"Decrypted in {duration}s\nIntegrity hash: {restored_hash[:20]}...")
            
        except ValueError as e:
            messagebox.showerror("Failed", str(e))