Homework hack for Safe Computing

import random  # th random module for generating random numbers

def caesar_cipher(text, shift, mode):
    result = ""
    for char in text:
        if char.isalpha():  
            shift_amount = shift if mode == "encrypt" else -shift
            new_char = chr(((ord(char.lower()) - 97 + shift_amount) % 26) + 97)
            result += new_char.upper() if char.isupper() else new_char
        else:
            result += char  
    return result

# Get user input
mode = input("Do you want to encrypt or decrypt? ").strip().lower()
message = input("Enter your message: ")
shift_input = input("Enter shift value (number of places to shift or type 'random'): ").strip().lower()

if shift_input == "random":
    shift = random.randint(1, 25)  
    print(f"Random shift value selected: {shift}")
else:
    shift = int(shift_input)  

# Perform encryption/decryption
output = caesar_cipher(message, shift, mode)
print(f"Result: {output}")

Random shift value selected: 15
Result: st espcp