Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 import os
 import hashlib
+import base64
 
 # Constants and Settings
 UPLOAD_FOLDER = "uploads"
 PASSWORD_FILE = "password_file.txt"
-ADMIN_USERNAME = "admin"  # Default admin username
-ADMIN_PASSWORD_HASH = hashlib.sha256("admin_password".encode()).hexdigest()  # Hash for the default admin password
+ADMIN_USERNAME = "admin"
+ADMIN_PASSWORD_HASH = hashlib.sha256("admin_password".encode()).hexdigest()
 app_settings = {
     "file_uploads_enabled": True
 }
@@ -14,26 +15,39 @@ app_settings = {
 # Ensure the upload directory exists
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
+# Function to encode the password in a "Very Hard Encoded Language"
+def encode_password(password):
+    encoded_bytes = base64.b64encode(password.encode('utf-8'))
+    return encoded_bytes.decode('utf-8')
+
+# Function to decode the password
+def decode_password(encoded_password):
+    decoded_bytes = base64.b64decode(encoded_password.encode('utf-8'))
+    return decoded_bytes.decode('utf-8')
+
 # Function to register a user
 def register(username, password):
     hashed_password = hashlib.sha256(password.encode()).hexdigest()
+    encoded_password = encode_password(hashed_password)
     
     with open(PASSWORD_FILE, "a") as f:
-        f.write(f"{username}:{hashed_password}\n")
+        f.write(f"{username}:{encoded_password}\n")
     
     return f"User '{username}' registered successfully!"
 
 # Function to log in a user
 def login(username, password):
     hashed_password = hashlib.sha256(password.encode()).hexdigest()
+    encoded_password = encode_password(hashed_password)
     
-    with open(PASSWORD_FILE, "r") as f:
-        users = f.readlines()
-    
-    for user in users:
-        stored_username, stored_password = user.strip().split(":")
-        if stored_username == username and stored_password == hashed_password:
-            return "Login successful!"
+    if os.path.exists(PASSWORD_FILE):
+        with open(PASSWORD_FILE, "r") as f:
+            users = f.readlines()
+        
+        for user in users:
+            stored_username, stored_encoded_password = user.strip().split(":")
+            if stored_username == username and stored_encoded_password == encoded_password:
+                return "Login successful!"
     
     return "Invalid username or password."
 
@@ -58,8 +72,8 @@ def get_users():
     if os.path.exists(PASSWORD_FILE):
         with open(PASSWORD_FILE, "r") as f:
             for line in f:
-                username, password = line.strip().split(":")
-                users[username] = password
+                username, encoded_password = line.strip().split(":")
+                users[username] = decode_password(encoded_password)  # Decoding for verification
     return users
 
 # Gradio Interface