main.py

import subprocess
import os
import openai
import shlex
import keyboard
from colorama import *
from termcolor import colored, cprint

# Global variable for tracking the working directory
workingdir = os.getcwd()
global checkmanmode
# Initialize colorama
init()

def clear_console():
    os.system('cls' if os.name == 'nt' else 'clear')

def write_data(file_path, input_string):
    with open(file_path, 'a') as file:
        file.write(input_string + '\n')

def execute_command(cmd):
    try:
        global workingdir
        write_data('Big.txt', cmd)
        if cmd.startswith('clrwindow'):
            clear_console()
            return 0
        if cmd.startswith('exit') or cmd.startswith('quit'):
            exit()
        if cmd.startswith('cd '):
            path = cmd.split(' ', 1)[1]
            # If the path is surrounded by quotes, remove them
            if path.startswith('"') and path.endswith('"'):
                path = path[1:-1]
            if path == '..':
                workingdir = os.path.dirname(workingdir)
            elif '..' in path:
                levels_up = path.count('..')
                workingdir = os.path.normpath(workingdir + '/' + '/'.join(['..'] * levels_up))
            else:
                potential_path = os.path.join(workingdir, path)
                if os.path.isabs(path):
                    potential_path = path
                if os.path.exists(potential_path):
                    workingdir = potential_path
                else:
                    raise Exception("The system cannot find the path specified: '"+potential_path+"'")
            return 0
        else:
            cmd = shlex.split(cmd)
            p = subprocess.Popen(cmd, cwd=workingdir, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
            while True:
                output = p.stdout.readline().decode('utf-8')
                if output == '' and p.poll() is not None:
                    break
                if output:
                    print(colored(output.strip(), 'light_grey'))
            rc = p.poll()
            return rc
    except Exception as e:
        print('An error occurred in running the interpreter: ' + str(e))
        return 1

try:
    openai.api_key = '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'

    messages = [
        {"role": "system", "content": f'You are Alpilaz Compute-Line (version 1.2.0 made by Kevin Dalli), a helpful Command line interpreter tool. you have direct access to the users command line, when the user inputs a command in standard english, for example "create a new file" then you will execute the command in standard command line notation depending on the users system type, this data will be given to you together with the first command request.\nfor example:\nuser: "create a text file"\nassistant: "echo. > text.txt"\nyou can also use context from previous executed commands to understand what the user is implying.\nif you want to output something in the channel for the user to see (like a reminder or a note, or a response to a users question) you can type printcmd"whatever you want to be seen in channel" USE THIS FOR ALL NORMAL CONVERSATIONAL TASKS. everything you type in chat will be interpreted as a command, so if you want to ask for permission or confirmation for something use printcmd"input" as described above, if you want to print multiple lines use the standard backslash n newline character.\nremember NEVER say anything normally, if you want to tell the user something use the command specified. If you want to clear the users command line window, then use clrwindow command. You cannot run multiple commands together, like clrwindow and printcmd and run system commands, only one at a time. you must follow these rules to the point. reply with "Understood" if you understand these instructions.\nThe users current system is running:\n{os.name}'}
    ]
except Exception as e:
    print('An error occured in initializing OpenAI: ' + str(e))
    checkmanmode = 'manmode'

def gen():
    while True:
        try:
            response = openai.ChatCompletion.create(
                model="gpt-4",
                messages=messages
            )
            assistant_message = response['choices'][0]['message']['content']
            if assistant_message != 'Understood' and not assistant_message.startswith('printcmd'):
                print('\n' + Fore.LIGHTBLACK_EX + "<<INT>> : ", assistant_message)
            print(colored('\-----------------------------------------------------------/', 'red'))
            if assistant_message.startswith('printcmd'):
                assistant_message = assistant_message[8:].strip('\"').replace(r'\\n', '\n')
                lines = assistant_message.split(r'\n')
                for line in lines:
                    print(line)
            elif assistant_message.startswith('Understood'):
                print('Initiating Command Line Interpreter...')
            else:
                rc = execute_command(assistant_message)    
                if rc != 0:
                    print(f'Error: Command "{assistant_message}" exited with return code {rc}')
            user_message = input(Fore.WHITE + "|" + Fore.LIGHTYELLOW_EX + "--" + Fore.YELLOW + "\\" + Fore.LIGHTRED_EX + "[" + Fore.RED + "|ALPILAZ COMPUTE-LINE|" + Fore.LIGHTRED_EX + "]" + Fore.YELLOW + "\\" + Fore.LIGHTYELLOW_EX + "-" + Fore.WHITE + "@~ " + workingdir + Fore.LIGHTBLACK_EX + " > >  ")
            if user_message == 'manmode':
                return 'manmode'
            messages.append({"role": "user", "content": user_message})
        except Exception as e:
            print('An error occured in backend startup... ' + str(e))
            return True

def man():
    print(colored("Initiating Command Line Interpreter in MANUAL MODE...\n", 'black', 'on_red'))
    while True:
        print(Fore.RED + r'\-----------------------------------------------------------/')
        user_message = input(Fore.WHITE + "|" + Fore.LIGHTYELLOW_EX + "--" + Fore.YELLOW + "\\" + Fore.LIGHTRED_EX + "[" + Fore.RED + "|ALPILAZ COMPUTE-LINE|" + Fore.LIGHTRED_EX + "]" + Fore.YELLOW + "\\" + Fore.LIGHTYELLOW_EX + "-" + Fore.WHITE + "@~ " + workingdir + " " + colored(r"-[DM]-", 'black', 'on_red') + Fore.LIGHTBLACK_EX + " > >  ")

        # Check if the command is 'manmode'
        if user_message == 'manmode':
            print("You are already in MANUAL MODE.")
            continue

        rc = execute_command(user_message)
        if rc != 0:
            print(f'Error: Command "{user_message}" exited with return code {rc}')
        
        if user_message == 'automode':
            return 'automode'
        
        if user_message == 'exit':
            return 'exit'

def main():
    mode = 'gen'  # Start with the automated mode by default
    while True:
        if mode == 'gen':
            mode = gen()  # It will return either 'manmode' or 'exit'
        elif mode == 'manmode':
            mode = man()  # It will return either 'automode' or 'exit'
        elif mode == 'automode':
            mode = 'gen'  # Switch back to the automated mode
        elif mode == 'exit':
            break
        
if __name__ == '__main__':
    main()