from classess.virustotal import *
from time import sleep

## main interface that will be introduced in the beginning
def interface():
    print("""
      _______                     ______        _ 
     |__   __|                   |___  /       | |
        | | ___  __ _ _ __ ___      / / ___  __| |
        | |/ _ \/ _' | '_ ' _ \    / / / _ \/ _' |
        | |  __/ (_| | | | | | |  / /_|  __/ (_| |
        |_|\___|\__,_|_| |_| |_| /_____\___|\__,_|
                                
                                Malware Detection Tool

        1) Scan specific file
        2) Scan Entire Disk
        3) Exit
    """)
    
    user_input = input("What do you want? ")
    if(user_input == "1"):
        route = os.path.abspath(".")
        print("These files are in the current folder")
        print(os.listdir("."))
        target_file = input("whats the target file? ")
        query_virusTotal(target_file)
    elif(user_input == "2"):
        query_virusTotal_Folder()
    else:
        print("Exiting.....")
        sleep(2.5)
        exit(0)
    input("#===========================Finished============================#")



while True:
    interface()

from classess.virustotal import *
from time import sleep

## main interface that will be introduced in the beginning
def interface():
    print("""
      _______                     ______        _ 
     |__   __|                   |___  /       | |
        | | ___  __ _ _ __ ___      / / ___  __| |
        | |/ _ \/ _' | '_ ' _ \    / / / _ \/ _' |
        | |  __/ (_| | | | | | |  / /_|  __/ (_| |
        |_|\___|\__,_|_| |_| |_| /_____\___|\__,_|
                                
                                Malware Detection Tool

        1) Scan specific file
        2) Scan Entire Disk
        3) Exit
    """)
    
    user_input = input("What do you want? ")
    if(user_input == "1"):
        route = os.path.abspath(".")
        print("These files are in the current folder")
        print(os.listdir("."))
        target_file = input("whats the target file? ")
        query_virusTotal(target_file)
    elif(user_input == "2"):
        query_virusTotal_Folder()
    else:
        print("Exiting.....")
        sleep(2.5)
        exit(0)
    input("#===========================Finished============================#")



while True:
    interface()

## main interface that will be introduced in the beginning _______                     ______        _       |__   __|                   |___  /       | |         | | ___  __ _ _ __ ___      / / ___  __| |         | |/ _ \/ _' | '_ ' _ \    / / / _ \/ _' |         | |  __/ (_| | | | | | |  / /_|  __/ (_| |         |_|\___|\__,_|_| |_| |_| /_____\___|\__,_|                                                                  Malware Detection Tool          1) Scan specific file         2) Scan Entire Disk         3) Exit #")