from cryptosteganography import CryptoSteganography

# Otwieranie pliku audio
mediafile = './steg/file_example_MP3_700KB.mp3'
message = None

with open(mediafile, "rb") as f:
   message = f.read()

print()
print('Program szuka pliku psy.jpg\n')
origfile = "./steg/psy.jpg"
print('Zdjęcie z ukrytym plikiem audio zostanie nazwane steg_audio_psy.png\n')
modfile = "./steg/steg_audio_psy.png"

key = "1111222233334444!"
crypto_steganography = CryptoSteganography(key)
crypto_steganography.hide(origfile, modfile, message)

print('Odszyfrowany plik zostanie nazwany odszyfrowana_probka_2.mp3 \n')
decrypted = './steg/odszyfrowana_probka_2.mp3'
secret_bin = crypto_steganography.retrieve(modfile)

# Zapis danych w nowym pliku
with open(decrypted, 'wb') as f:
    f.write(secret_bin)

