pc.py

import os
import sys
import cores as c
import banners
from pystyle import Colorate, Colors
from rich.console import Console
from InquirerPy import prompt
from InquirerPy.base.control import Choice
from rich.style import Style
import util

console = Console()

# Estilos para InquirerPy
question_style = Style(color="red", bold=True)
choice_style = Style(color="red")
pointer_style = Style(color="red")
selected_style = Style(color="red", bold=True)

# Perguntas para o prompt
questions = [
    {
        "type": "list",
        "name": "choice",
        "message": ("                                               What do you want?"),
        "choices": [
            "                                               1. Hide message in an image",
            "                                               2. Scan image",
            "                                               3. Software Informations",
            "                                               4. Exit",
        ],
    }
]

try:
    util.loading_pc()
    while True:
        os.system('clear')
        print(Colorate.Vertical(Colors.yellow_to_red, banners.banner1))
        print(f"                                                                                                     {c.lwhite}v{util.VERSION}")
        answers = prompt(questions, style={
            "question": "fg:ansired bold",
            "pointer": "fg:ansired",
            "highlight": "fg:ansired bold",
            "answer": "fg:ansired bold",
        })
        respost = answers['choice']

        if "1" in respost:
            os.system('clear')
            print(Colorate.Vertical(Colors.yellow_to_red, banners.banner1))
            image_path = input(f"\n{c.bwhite}[{c.byellow}>{c.bwhite}]{c.white} Enter the path to the image: ")
            message_to_hide = input(f"{c.bwhite}[{c.byellow}>{c.bwhite}]{c.white} Enter the message to be hidden in the image: ")
            if util.hide_message(image_path, message_to_hide):
                print(f"\n{c.green}Message successfully hidden in {image_path} !{c.white}")
                input(f"\n{c.white}Press {c.bwhite}[ENTER]{c.white} to continue")
        elif "2" in respost:
            os.system('clear')
            print(Colorate.Vertical(Colors.yellow_to_red, banners.banner1))
            image_path = input(f"\n{c.bwhite}[{c.byellow}>{c.bwhite}]{c.white} Enter the path of the image with the hidden message: ")
            message_recovered = util.scan_image_metadata(image_path)
            if message_recovered:
                print(f'\n{message_recovered}')
                input(f"\n{c.white}Press {c.bwhite}[ENTER]{c.white} to continue")
        elif "3" in respost:
            os.system('clear')
            print(Colorate.Vertical(Colors.yellow_to_red, banners.banner2))
            print(banners.options)
            respost_informations = input(f"{c.bwhite}[{c.byellow}>{c.bwhite}]{c.white} You choice: ")
            match respost_informations:
                case "1":
                    os.system('clear')
                    print(Colorate.Vertical(Colors.red_to_yellow, banners.banner6))
                    print(f"\n{c.byellow}[{c.white}Creator 👔 {c.byellow}]{c.white}Phant0m The Great{c.white}\n\n{c.byellow}[{c.white}Discord Contact 🪪 {c.byellow}]{c.white}phant0mthegreat.")
                    input(f"\nPress {c.bwhite}[ENTER]{c.white} to continue")
                case "2":
                    os.system('clear')
                    print(Colorate.Vertical(Colors.red_to_yellow, banners.banner7))
                    print(f"\n{c.byellow}[{c.white}Version ⚙️ {c.byellow}]{c.white}{util.VERSION}\n\n{c.byellow}[{c.white}Programming L. 💾 {c.byellow}]{c.white}Python\n\n{c.byellow}[{c.white}Total lines of code 🩻 {c.byellow}]{c.white}602")
                    input(f"\nPress {c.bwhite}[ENTER]{c.white} to continue")
        elif "4" in respost:
            os.system('clear')
            print(Colorate.Vertical(Colors.yellow_to_red, banners.banner3))
            print(f"\n{c.byellow}Thank you for using Stark")
            sys.exit()
except KeyboardInterrupt:
    print('\n[#] Program interrupted')