Update app.py

app.py

@@ -15,38 +15,42 @@ 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'))
+# Function to encode data in a "Very Hard Encoded Language"
+def encode_data(data):
+    encoded_bytes = base64.b64encode(data.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'))
+# Function to decode data
+def decode_data(encoded_data):
+    decoded_bytes = base64.b64decode(encoded_data.encode('utf-8'))
     return decoded_bytes.decode('utf-8')
 
 # Function to register a user
 def register(username, password):
+    # Encode username and password
+    encoded_username = encode_data(username)
     hashed_password = hashlib.sha256(password.encode()).hexdigest()
-    encoded_password = encode_password(hashed_password)
+    encoded_password = encode_data(hashed_password)
     
+    # Write the encoded username and password to the file
     with open(PASSWORD_FILE, "a") as f:
-        f.write(f"{username}:{encoded_password}\n")
+        f.write(f"{encoded_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)
+    encoded_password = encode_data(hashed_password)
+    encoded_username = encode_data(username)
     
     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:
+            stored_encoded_username, stored_encoded_password = user.strip().split(":")
+            if stored_encoded_username == encoded_username and stored_encoded_password == encoded_password:
                 return "Login successful!"
     
     return "Invalid username or password."
@@ -56,7 +60,9 @@ def upload_file(file, username):
     if not app_settings['file_uploads_enabled']:
         return "File uploads are disabled by admin."
 
-    if username not in get_users():
+    # Decode username to verify
+    decoded_username = decode_data(encode_data(username))  # Placeholder, verify appropriately
+    if decoded_username not in get_users():
         return "Please log in to upload files."
 
     # Save the uploaded file
@@ -64,7 +70,7 @@ def upload_file(file, username):
     with open(file_path, "wb") as f:
         f.write(file.read())
 
-    return f"File '{file.name}' uploaded successfully by {username}."
+    return f"File '{file.name}' uploaded successfully by {decoded_username}."
 
 # Function to retrieve the list of users
 def get_users():
@@ -72,8 +78,8 @@ def get_users():
     if os.path.exists(PASSWORD_FILE):
         with open(PASSWORD_FILE, "r") as f:
             for line in f:
-                username, encoded_password = line.strip().split(":")
-                users[username] = decode_password(encoded_password)  # Decoding for verification
+                encoded_username, encoded_password = line.strip().split(":")
+                users[decode_data(encoded_username)] = decode_data(encoded_password)  # Decoding for verification
     return users
 
 # Gradio Interface