Image_Steganography.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/ritvic/Image-Steganography/blob/main/Copy_of_Network_Security_Project.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 1,
      "metadata": {
        "id": "Eke0-hLgOH4T",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "ba2d76fc-17f5-4a02-d607-6ad3dd903972"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: Crypto in /usr/local/lib/python3.8/dist-packages (1.4.1)\n",
            "Requirement already satisfied: shellescape in /usr/local/lib/python3.8/dist-packages (from Crypto) (3.8.1)\n",
            "Requirement already satisfied: Naked in /usr/local/lib/python3.8/dist-packages (from Crypto) (0.1.32)\n",
            "Requirement already satisfied: requests in /usr/local/lib/python3.8/dist-packages (from Naked->Crypto) (2.23.0)\n",
            "Requirement already satisfied: pyyaml in /usr/local/lib/python3.8/dist-packages (from Naked->Crypto) (6.0)\n",
            "Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.8/dist-packages (from requests->Naked->Crypto) (2.10)\n",
            "Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.8/dist-packages (from requests->Naked->Crypto) (3.0.4)\n",
            "Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /usr/local/lib/python3.8/dist-packages (from requests->Naked->Crypto) (1.24.3)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.8/dist-packages (from requests->Naked->Crypto) (2022.12.7)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: colorama==0.4.4 in /usr/local/lib/python3.8/dist-packages (0.4.4)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: commonmark==0.9.1 in /usr/local/lib/python3.8/dist-packages (0.9.1)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: Pillow==8.1.0 in /usr/local/lib/python3.8/dist-packages (8.1.0)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: pycryptodome==3.9.9 in /usr/local/lib/python3.8/dist-packages (3.9.9)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: pyfiglet==0.8.post1 in /usr/local/lib/python3.8/dist-packages (0.8.post1)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: Pygments==2.7.4 in /usr/local/lib/python3.8/dist-packages (2.7.4)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: rich==9.10.0 in /usr/local/lib/python3.8/dist-packages (9.10.0)\n",
            "Requirement already satisfied: colorama<0.5.0,>=0.4.0 in /usr/local/lib/python3.8/dist-packages (from rich==9.10.0) (0.4.4)\n",
            "Requirement already satisfied: commonmark<0.10.0,>=0.9.0 in /usr/local/lib/python3.8/dist-packages (from rich==9.10.0) (0.9.1)\n",
            "Requirement already satisfied: pygments<3.0.0,>=2.6.0 in /usr/local/lib/python3.8/dist-packages (from rich==9.10.0) (2.7.4)\n",
            "Requirement already satisfied: typing-extensions<4.0.0,>=3.7.4 in /usr/local/lib/python3.8/dist-packages (from rich==9.10.0) (3.7.4.3)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: termcolor==1.1.0 in /usr/local/lib/python3.8/dist-packages (1.1.0)\n",
            "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
            "Requirement already satisfied: typing-extensions==3.7.4.3 in /usr/local/lib/python3.8/dist-packages (3.7.4.3)\n"
          ]
        }
      ],
      "source": [
        "!pip install Crypto\n",
        "!pip install colorama==0.4.4\n",
        "!pip install commonmark==0.9.1\n",
        "!pip install Pillow==8.1.0\n",
        "!pip install pycryptodome==3.9.9\n",
        "!pip install pyfiglet==0.8.post1\n",
        "!pip install Pygments==2.7.4\n",
        "!pip install rich==9.10.0\n",
        "!pip install termcolor==1.1.0\n",
        "!pip install typing-extensions==3.7.4.3"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "from PIL import Image\n",
        "import os.path\n",
        "from os import path\n",
        "# import math\n",
        "from Crypto.Cipher import AES\n",
        "from Crypto.Hash import SHA256\n",
        "from Crypto import Random\n",
        "import base64\n",
        "# from colorama import init\n",
        "# from termcolor import cprint \n",
        "# from pyfiglet import figlet_format\n",
        "from rich import print\n",
        "from rich.console import Console\n",
        "from rich.table import Table\n",
        "from rich.progress import track\n",
        "# import os\n",
        "import getpass\n",
        "import sys\n",
        "\n",
        "DEBUG = False\n",
        "console = Console()\n",
        "headerText = \"M6nMjy5THr2J\""
      ],
      "metadata": {
        "id": "LcHt-asn0Q93"
      },
      "execution_count": 2,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def encrypt(key, source, encode=True):\n",
        "    key = SHA256.new(key).digest()  # use SHA-256 over our key to get a proper-sized AES key\n",
        "    IV = Random.new().read(AES.block_size)  # generate Initialisation Vector (IV)\n",
        "    encryptor = AES.new(key, AES.MODE_CBC, IV) # AES object\n",
        "    padding = AES.block_size - len(source) % AES.block_size  # calculate needed padding\n",
        "    source += bytes([padding]) * padding  # Store the padding size for padding size number of times\n",
        "    data = IV + encryptor.encrypt(source)  # store the IV at the beginning and encrypt\n",
        "    return base64.b64encode(data).decode() if encode else data"
      ],
      "metadata": {
        "id": "4SzgQklK0W13"
      },
      "execution_count": 3,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def decrypt(key, source, decode=True):\n",
        "    if decode:\n",
        "        source = base64.b64decode(source.encode())\n",
        "    key = SHA256.new(key).digest()  # use SHA-256 over our key to get a proper-sized AES key\n",
        "    IV = source[:AES.block_size]  # extract the IV from the beginning\n",
        "    decryptor = AES.new(key, AES.MODE_CBC, IV) # AES object\n",
        "    data = decryptor.decrypt(source[AES.block_size:])  # decrypt\n",
        "    padding = data[-1]  # pick the padding value from the end\n",
        "    if data[-padding:] != bytes([padding]) * padding:  # Checking if the padding is valid\n",
        "        raise ValueError(\"Invalid padding...\")\n",
        "    return data[:-padding]  # remove the padding and return"
      ],
      "metadata": {
        "id": "qRreiN_50cFq"
      },
      "execution_count": 4,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# If the image is not in RGB format, convert it to RGB format by creating a new image\n",
        "# and pasting the current image to it.\n",
        "def convertToRGB(img):\n",
        "    try:\n",
        "        rgba_image = img\n",
        "        rgba_image.load()\n",
        "        background = Image.new(\"RGB\", rgba_image.size, (255, 255, 255))\n",
        "        background.paste(rgba_image, mask = rgba_image.split()[3])\n",
        "        print(\"[yellow]Converted image to RGB [/yellow]\")\n",
        "        return background\n",
        "    except Exception as e:\n",
        "        print(\"[red]Couldn't convert image to RGB [/red]- %s\"%e)\n",
        "\n",
        "# Returns the pixel count of the image\n",
        "def getPixelCount(img):\n",
        "    width, height = Image.open(img).size\n",
        "    return width*height"
      ],
      "metadata": {
        "id": "YBzJ2-mQ0gEm"
      },
      "execution_count": 5,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def encodeImage(image,message,filename):\n",
        "    with console.status(\"[green]Encoding image..\") as status:\n",
        "        try:\n",
        "            width, height = image.size\n",
        "            pix = image.getdata()\n",
        "\n",
        "            current_pixel = 0\n",
        "            tmp=0\n",
        "            # three_pixels = []\n",
        "            x=0\n",
        "            y=0\n",
        "            for ch in message:\n",
        "                binary_value = format(ord(ch), '08b')\n",
        "                \n",
        "                # For each character, get 3 pixels at a time\n",
        "                p1 = pix[current_pixel]\n",
        "                p2 = pix[current_pixel+1]\n",
        "                p3 = pix[current_pixel+2]\n",
        "\n",
        "                three_pixels = [val for val in p1+p2+p3]\n",
        "\n",
        "                for i in range(0,8):\n",
        "                    current_bit = binary_value[i]\n",
        "\n",
        "                    # 0 - Even\n",
        "                    # 1 - Odd\n",
        "                    if current_bit == '0':\n",
        "                        if three_pixels[i]%2!=0:\n",
        "                            three_pixels[i]= three_pixels[i]-1 if three_pixels[i]==255 else three_pixels[i]+1\n",
        "                    elif current_bit == '1':\n",
        "                        if three_pixels[i]%2==0:\n",
        "                            three_pixels[i]= three_pixels[i]-1 if three_pixels[i]==255 else three_pixels[i]+1\n",
        "\n",
        "                current_pixel+=3\n",
        "                tmp+=1\n",
        "\n",
        "                #Set 9th value\n",
        "                if(tmp==len(message)):\n",
        "                    # Make as 1 (odd) - stop reading\n",
        "                    if three_pixels[-1]%2==0:\n",
        "                        three_pixels[-1]= three_pixels[-1]-1 if three_pixels[-1]==255 else three_pixels[-1]+1\n",
        "                else:\n",
        "                    # Make as 0 (even) - continue reading\n",
        "                    if three_pixels[-1]%2!=0:\n",
        "                        three_pixels[-1]= three_pixels[-1]-1 if three_pixels[-1]==255 else three_pixels[-1]+1\n",
        "\n",
        "\n",
        "                if DEBUG:\n",
        "                    print(\"Character: \",ch)\n",
        "                    print(\"Binary: \",binary_value)\n",
        "                    print(\"Three pixels before mod: \",three_pixels)\n",
        "                    print(\"Three pixels after mod: \",three_pixels)\n",
        "\n",
        "\n",
        "                three_pixels = tuple(three_pixels)\n",
        "                \n",
        "                st=0\n",
        "                end=3\n",
        "\n",
        "                for i in range(0,3):\n",
        "                    if DEBUG:\n",
        "                        print(\"Putting pixel at \",(x,y),\" to \",three_pixels[st:end])\n",
        "\n",
        "                    image.putpixel((x,y), three_pixels[st:end])\n",
        "                    st+=3\n",
        "                    end+=3\n",
        "\n",
        "                    if (x == width - 1):\n",
        "                        x = 0\n",
        "                        y += 1\n",
        "                    else:\n",
        "                        x += 1\n",
        "\n",
        "            encoded_filename = filename.split('.')[0] + \"-enc.png\"\n",
        "            image.save(encoded_filename)\n",
        "            print(\"\\n\")\n",
        "            print(\"[yellow]Original File: [u]%s[/u][/yellow]\"%filename)\n",
        "            print(\"[green]Image encoded and saved as [u][bold]%s[/green][/u][/bold]\"%encoded_filename)\n",
        "\n",
        "        except Exception as e:\n",
        "            print(\"[red]An error occured - [/red]%s\"%e)\n",
        "            sys.exit(0)"
      ],
      "metadata": {
        "id": "t78MxdHt0jwh"
      },
      "execution_count": 6,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def decodeImage(image):\n",
        "    with console.status(\"[green]Decoding image..\") as status:\n",
        "        try:\n",
        "            pix = image.getdata()\n",
        "            current_pixel = 0\n",
        "            decoded=\"\"\n",
        "            while True:\n",
        "                # Get 3 pixels each time\n",
        "                binary_value=\"\"\n",
        "                p1 = pix[current_pixel]\n",
        "                p2 = pix[current_pixel+1]\n",
        "                p3 = pix[current_pixel+2]\n",
        "                three_pixels = [val for val in p1+p2+p3]\n",
        "\n",
        "                for i in range(0,8):\n",
        "                    if three_pixels[i]%2==0:\n",
        "                        # add 0\n",
        "                        binary_value+=\"0\"\n",
        "                    elif three_pixels[i]%2!=0:\n",
        "                        # add 1\n",
        "                        binary_value+=\"1\"\n",
        "\n",
        "\n",
        "                #Convert binary value to ascii and add to string\n",
        "                binary_value.strip()\n",
        "                ascii_value = int(binary_value,2)\n",
        "                decoded+=chr(ascii_value)\n",
        "                current_pixel+=3\n",
        "\n",
        "                if DEBUG:\n",
        "                    print(\"Binary: \",binary_value)\n",
        "                    print(\"Ascii: \",ascii_value)\n",
        "                    print(\"Character: \",chr(ascii_value))\n",
        "\n",
        "                if three_pixels[-1]%2!=0:\n",
        "                    # stop reading\n",
        "                    break\n",
        "\n",
        "            # print(\"Decoded: %s\"%decoded)\n",
        "            return decoded\n",
        "        except Exception as e:\n",
        "            print(\"[red]An error occured - [/red]%s\"%e)\n",
        "            sys.exit()"
      ],
      "metadata": {
        "id": "jNoWtyhB0qim"
      },
      "execution_count": 7,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def main():\n",
        "    # insertHeaders(img)\n",
        "\n",
        "    print(\"[cyan]Choose one: [/cyan]\")\n",
        "    op = int(input(\"1. Encode\\n2. Decode\\n>>\"))\n",
        "\n",
        "    if op==1:\n",
        "        print(\"[cyan]Image path (with extension): [/cyan]\")\n",
        "        img = input(\">>\")\n",
        "        if(not(path.exists(img))):\n",
        "            raise Exception(\"Image not found!\")\n",
        "\n",
        "        \n",
        "        print(\"[cyan]Message to be hidden: [/cyan]\")\n",
        "        message = input(\">>\")\n",
        "        message = headerText + message\n",
        "        if((len(message)+len(headerText))*3 > getPixelCount(img)):\n",
        "            raise Exception(\"Given message is too long to be encoded in the image.\")\n",
        "\n",
        "\n",
        "        password=\"\"\n",
        "        while 1:\n",
        "            print(\"[cyan]Password to encrypt (leave empty if you want no password): [/cyan]\")\n",
        "            password = getpass.getpass(\">>\")\n",
        "            if password==\"\":\n",
        "                break\n",
        "            print(\"[cyan]Re-enter Password: [/cyan]\")\n",
        "            confirm_password = getpass.getpass(\">>\")\n",
        "            if(password!=confirm_password):\n",
        "                print(\"[red]Passwords don't match try again [/red]\")\n",
        "            else:\n",
        "                break\n",
        "\n",
        "        cipher=\"\"\n",
        "        if password!=\"\":\n",
        "            cipher = encrypt(key=password.encode(),source=message.encode())\n",
        "            # Add header to cipher\n",
        "            cipher = headerText + cipher\n",
        "        else:\n",
        "            cipher = message\n",
        "\n",
        "\n",
        "        if DEBUG:\n",
        "            print(\"[yellow]Encrypted : [/yellow]\",cipher)\n",
        "\n",
        "        image = Image.open(img)\n",
        "        print(\"[yellow]Image Mode: [/yellow]%s\"%image.mode)\n",
        "        if image.mode!='RGB':\n",
        "            image = convertToRGB(image)\n",
        "        newimg = image.copy()\n",
        "        encodeImage(image=newimg,message=cipher,filename=image.filename)\n",
        "\n",
        "    elif op==2:\n",
        "        print(\"[cyan]Image path (with extension): [/cyan]\")\n",
        "        img = input(\">>\")\n",
        "        if(not(path.exists(img))):\n",
        "            raise Exception(\"Image not found!\")\n",
        "\n",
        "        print(\"[cyan]Enter password (leave empty if no password): [/cyan]\")\n",
        "        password = getpass.getpass(\">>\")\n",
        "\n",
        "        image = Image.open(img)\n",
        "\n",
        "        cipher = decodeImage(image)\n",
        "\n",
        "\n",
        "        header = cipher[:len(headerText)]\n",
        "\n",
        "        if header.strip()!=headerText:\n",
        "            print(\"[red]Invalid data![/red]\")\n",
        "            sys.exit(0)\n",
        "\n",
        "\n",
        "        print()\n",
        "\n",
        "        if DEBUG:\n",
        "            print(\"[yellow]Decoded text: %s[/yellow]\"%cipher)\n",
        "\n",
        "        decrypted=\"\"\n",
        "\n",
        "        if password!=\"\":\n",
        "            cipher = cipher[len(headerText):]\n",
        "            print(\"cipher : \",cipher)\n",
        "            try:\n",
        "                decrypted = decrypt(key=password.encode(),source=cipher)\n",
        "            except Exception as e:\n",
        "                print(\"[red]Wrong password![/red]\")\n",
        "                sys.exit(0)\n",
        "\n",
        "        else:\n",
        "            decrypted=cipher\n",
        "\n",
        "\n",
        "        header = decrypted.decode()[:len(headerText)]\n",
        "\n",
        "        if header!=headerText:\n",
        "            print(\"[red]Wrong password![/red]\")\n",
        "            sys.exit(0)\n",
        "\n",
        "        decrypted = decrypted[len(headerText):].decode(\"utf-8\")\n",
        "\n",
        "\n",
        "\n",
        "        print(\"[green]Decoded Text: \\n[bold]%s[/bold][/green]\"%decrypted)\n",
        "\n",
        "\n",
        "if __name__ == \"__main__\":\n",
        "    os.system('cls' if os.name == 'nt' else 'clear')\n",
        "    # cprint(figlet_format('IMGHIDE!', font='starwars'),'yellow', attrs=['bold'])\n",
        "    # print_credits()\n",
        "    print()\n",
        "    print()\n",
        "    \n",
        "    main()"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "0J0AJJ2o08Ci",
        "outputId": "959f288b-cb9e-4f33-fd6f-c76e4984d1f6"
      },
      "execution_count": 8,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "\n",
            "Choose one: \n",
            "1. Encode\n",
            "2. Decode\n",
            ">>1\n",
            "Image path (with extension): \n",
            ">>/content/Img.jpg\n",
            "Message to be hidden: \n",
            ">>Hey folks, I am Ritvik and this is the hidden message.\n",
            "Password to encrypt (leave empty if you want no password): \n",
            ">>··········\n",
            "Re-enter Password: \n",
            ">>··········\n",
            "Image Mode: RGB\n",
            "\n",
            "\n",
            "Original File: /content/Img.jpg\n",
            "Image encoded and saved as /content/Img-enc.png\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "def main():\n",
        "    # insertHeaders(img)\n",
        "\n",
        "    print(\"[cyan]Choose one: [/cyan]\")\n",
        "    op = int(input(\"1. Encode\\n2. Decode\\n>>\"))\n",
        "\n",
        "    if op==1:\n",
        "        print(\"[cyan]Image path (with extension): [/cyan]\")\n",
        "        img = input(\">>\")\n",
        "        if(not(path.exists(img))):\n",
        "            raise Exception(\"Image not found!\")\n",
        "\n",
        "        \n",
        "        print(\"[cyan]Message to be hidden: [/cyan]\")\n",
        "        message = input(\">>\")\n",
        "        message = headerText + message\n",
        "        if((len(message)+len(headerText))*3 > getPixelCount(img)):\n",
        "            raise Exception(\"Given message is too long to be encoded in the image.\")\n",
        "\n",
        "\n",
        "        password=\"\"\n",
        "        while 1:\n",
        "            print(\"[cyan]Password to encrypt (leave empty if you want no password): [/cyan]\")\n",
        "            password = getpass.getpass(\">>\")\n",
        "            if password==\"\":\n",
        "                break\n",
        "            print(\"[cyan]Re-enter Password: [/cyan]\")\n",
        "            confirm_password = getpass.getpass(\">>\")\n",
        "            if(password!=confirm_password):\n",
        "                print(\"[red]Passwords don't match try again [/red]\")\n",
        "            else:\n",
        "                break\n",
        "\n",
        "        cipher=\"\"\n",
        "        if password!=\"\":\n",
        "            cipher = encrypt(key=password.encode(),source=message.encode())\n",
        "            # Add header to cipher\n",
        "            cipher = headerText + cipher\n",
        "        else:\n",
        "            cipher = message\n",
        "\n",
        "\n",
        "        if DEBUG:\n",
        "            print(\"[yellow]Encrypted : [/yellow]\",cipher)\n",
        "\n",
        "        image = Image.open(img)\n",
        "        print(\"[yellow]Image Mode: [/yellow]%s\"%image.mode)\n",
        "        if image.mode!='RGB':\n",
        "            image = convertToRGB(image)\n",
        "        newimg = image.copy()\n",
        "        encodeImage(image=newimg,message=cipher,filename=image.filename)\n",
        "\n",
        "    elif op==2:\n",
        "        print(\"[cyan]Image path (with extension): [/cyan]\")\n",
        "        img = input(\">>\")\n",
        "        if(not(path.exists(img))):\n",
        "            raise Exception(\"Image not found!\")\n",
        "\n",
        "        print(\"[cyan]Enter password (leave empty if no password): [/cyan]\")\n",
        "        password = getpass.getpass(\">>\")\n",
        "\n",
        "        image = Image.open(img)\n",
        "\n",
        "        cipher = decodeImage(image)\n",
        "\n",
        "\n",
        "        header = cipher[:len(headerText)]\n",
        "\n",
        "        if header.strip()!=headerText:\n",
        "            print(\"[red]Invalid data![/red]\")\n",
        "            sys.exit(0)\n",
        "\n",
        "\n",
        "        print()\n",
        "\n",
        "        if DEBUG:\n",
        "            print(\"[yellow]Decoded text: %s[/yellow]\"%cipher)\n",
        "\n",
        "        decrypted=\"\"\n",
        "\n",
        "        if password!=\"\":\n",
        "            cipher = cipher[len(headerText):]\n",
        "            print(\"cipher : \",cipher)\n",
        "            try:\n",
        "                decrypted = decrypt(key=password.encode(),source=cipher)\n",
        "            except Exception as e:\n",
        "                print(\"[red]Wrong password![/red]\")\n",
        "                sys.exit(0)\n",
        "\n",
        "        else:\n",
        "            decrypted=cipher\n",
        "\n",
        "\n",
        "        header = decrypted.decode()[:len(headerText)]\n",
        "\n",
        "        if header!=headerText:\n",
        "            print(\"[red]Wrong password![/red]\")\n",
        "            sys.exit(0)\n",
        "\n",
        "        decrypted = decrypted[len(headerText):].decode(\"utf-8\")\n",
        "\n",
        "\n",
        "\n",
        "        print(\"[green]Decoded Text: \\n[bold]%s[/bold][/green]\"%decrypted)\n",
        "\n",
        "\n",
        "if __name__ == \"__main__\":\n",
        "    os.system('cls' if os.name == 'nt' else 'clear')\n",
        "    # cprint(figlet_format('IMGHIDE!', font='starwars'),'yellow', attrs=['bold'])\n",
        "    # print_credits()\n",
        "    print()\n",
        "    print()\n",
        "    \n",
        "    main()"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "70bc1fdc-f5fc-4a67-bed5-6fcc32886e24",
        "id": "fXcHBMkKjsFq"
      },
      "execution_count": 9,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "\n",
            "Choose one: \n",
            "1. Encode\n",
            "2. Decode\n",
            ">>2\n",
            "Image path (with extension): \n",
            ">>/content/Img-enc.png\n",
            "Enter password (leave empty if no password): \n",
            ">>··········\n",
            "\n",
            "cipher :  dMji9XC9zTJ8Lvftg2ZT2nckLOV9mhlWrM1MJkfrPyndZ6u6Q/QBnM+pf1vMEUL+lxfG2X\n",
            "OYo+3rfGWHamcN3HyHEkJAqJCksQZiWOeMgg/TK+uz+YiGtaxXz7pEJsyS\n",
            "Decoded Text: \n",
            "Hey folks, I am Ritvik and this is the hidden message.\n"
          ]
        }
      ]
    }
  ],
  "metadata": {
    "colab": {
      "provenance": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}