Delete app.py

app.py

@@ -1,217 +0,0 @@
-import gradio as gr
-import os
-import time
-from functools import partial
-from helper import (
-    create_user, verify_user, get_user_files, upload_file, download_file,
-    delete_file, empty_vault, is_admin, get_all_accounts, delete_account,
-    encrypt_file, decrypt_file, is_rate_limited, is_account_locked, record_login_attempt
-)
-
-# Constants
-MAX_FILE_SIZE = 5 * 1024 * 1024 * 1024  # 5GB in bytes
-INACTIVITY_TIMEOUT = 300  # 5 minutes in seconds
-
-# Global variables
-current_user = None
-last_activity_time = time.time()
-
-def check_inactivity():
-    global current_user, last_activity_time
-    if current_user and time.time() - last_activity_time > INACTIVITY_TIMEOUT:
-        logout()
-        return True
-    return False
-
-def update_activity():
-    global last_activity_time
-    last_activity_time = time.time()
-
-def login(username, password):
-    global current_user
-    if check_inactivity():
-        return gr.update(visible=True), gr.update(visible=False), "You have been logged out due to inactivity."
-    
-    if is_rate_limited(username):
-        return gr.update(visible=True), gr.update(visible=False), "Too many login attempts. Please try again later."
-    
-    if is_account_locked(username):
-        return gr.update(visible=True), gr.update(visible=False), "Account is locked. Please try again later."
-    
-    if verify_user(username, password):
-        current_user = username
-        update_activity()
-        record_login_attempt(username, True)
-        return gr.update(visible=False), gr.update(visible=True), f"Welcome, {username}!"
-    else:
-        record_login_attempt(username, False)
-        return gr.update(visible=True), gr.update(visible=False), "Invalid username or password."
-
-def logout():
-    global current_user
-    current_user = None
-    return gr.update(visible=True), gr.update(visible=False), "You have been logged out."
-
-def register(username, password):
-    if check_inactivity():
-        return "You have been logged out due to inactivity."
-    
-    result = create_user(username, password)
-    update_activity()
-    return result
-
-def upload(files):
-    if check_inactivity():
-        return "You have been logged out due to inactivity."
-    
-    if not current_user:
-        return "Please log in to upload files."
-    
-    results = []
-    for file in files:
-        if file.size > MAX_FILE_SIZE:
-            results.append(f"File {file.name} exceeds the 5GB limit.")
-        else:
-            encrypted_file = encrypt_file(file.name, file.read())
-            result = upload_file(current_user, file.name, encrypted_file)
-            results.append(result)
-    
-    update_activity()
-    return "\n".join(results)
-
-def get_files():
-    if check_inactivity():
-        return []
-    
-    if not current_user:
-        return []
-    
-    files = get_user_files(current_user)
-    update_activity()
-    return [{"File": file[0], "Size": f"{file[1] / 1024 / 1024:.2f} MB"} for file in files]
-
-def download(filename):
-    if check_inactivity():
-        return None
-    
-    if not current_user:
-        return None
-    
-    file_content = download_file(current_user, filename)
-    if file_content:
-        decrypted_content = decrypt_file(filename, file_content)
-        update_activity()
-        return decrypted_content
-    else:
-        return None
-
-def delete(filename):
-    if check_inactivity():
-        return "You have been logged out due to inactivity."
-    
-    if not current_user:
-        return "Please log in to delete files."
-    
-    result = delete_file(current_user, filename)
-    update_activity()
-    return result
-
-def empty_user_vault(password):
-    if check_inactivity():
-        return "You have been logged out due to inactivity."
-    
-    if not current_user:
-        return "Please log in to empty your vault."
-    
-    if verify_user(current_user, password):
-        result = empty_vault(current_user)
-        update_activity()
-        return result
-    else:
-        return "Invalid password. Vault not emptied."
-
-def admin_view():
-    if not is_admin(current_user):
-        return [], "Access denied."
-    
-    accounts = get_all_accounts()
-    update_activity()
-    return accounts, "Admin view loaded."
-
-def admin_delete_account(username):
-    if not is_admin(current_user):
-        return "Access denied."
-    
-    result = delete_account(username)
-    update_activity()
-    return result
-
-# Gradio interface
-with gr.Blocks(css="styles.css") as app:
-    gr.HTML('<img src="./logo.svg" alt="Grimvault Logo" class="logo">')
-    
-    with gr.Tab("Login"):
-        login_username = gr.Textbox(label="Username")
-        login_password = gr.Textbox(label="Password", type="password")
-        login_button = gr.Button("Login")
-        register_button = gr.Button("Register")
-        login_message = gr.Textbox(label="Message", interactive=False)
-    
-    with gr.Tab("Dashboard", visible=False) as dashboard:
-        gr.HTML('<img src="file/logo.svg" alt="Grimvault Logo" class="logo-small">')
-        upload_button = gr.File(label="Upload File(s)", file_count="multiple")
-        files_table = gr.Dataframe(label="Your Files", headers=["File", "Size"])
-        refresh_button = gr.Button("Refresh Files")
-        download_button = gr.Button("Download")
-        delete_button = gr.Button("Delete")
-        selected_file = gr.Textbox(label="Selected File")
-        empty_vault_password = gr.Textbox(label="Password to Empty Vault", type="password")
-        empty_vault_button = gr.Button("Empty Vault")
-        logout_button = gr.Button("Logout")
-        dashboard_message = gr.Textbox(label="Message", interactive=False)
-    
-    with gr.Tab("Admin", visible=False) as admin_tab:
-        admin_accounts = gr.Dataframe(label="User Accounts", headers=["Username", "Created At"])
-        admin_delete_username = gr.Textbox(label="Username to Delete")
-        admin_delete_button = gr.Button("Delete User")
-        admin_message = gr.Textbox(label="Admin Message", interactive=False)
-
-    # Login events
-    login_button.click(
-        login,
-        inputs=[login_username, login_password],
-        outputs=[gr.Tabs(), dashboard, login_message]
-    )
-    register_button.click(
-        register,
-        inputs=[login_username, login_password],
-        outputs=[login_message]
-    )
-
-    # Dashboard events
-    upload_button.upload(upload, inputs=[upload_button], outputs=[dashboard_message])
-    refresh_button.click(get_files, outputs=[files_table])
-    files_table.select(lambda df, evt: evt.data['File'], inputs=[files_table], outputs=[selected_file])
-    download_button.click(
-        download,
-        inputs=[selected_file],
-        outputs=[gr.File(label="Downloaded File")]
-    )
-    delete_button.click(delete, inputs=[selected_file], outputs=[dashboard_message])
-    empty_vault_button.click(
-        empty_user_vault,
-        inputs=[empty_vault_password],
-        outputs=[dashboard_message]
-    )
-    logout_button.click(logout, outputs=[gr.Tabs(), dashboard, login_message])
-
-    # Admin events
-    admin_tab.select(admin_view, outputs=[admin_accounts, admin_message])
-    admin_delete_button.click(
-        admin_delete_account,
-        inputs=[admin_delete_username],
-        outputs=[admin_message]
-    )
-
-if __name__ == "__main__":
-    app.launch()