from chatgpt_wrapper import ChatGPT
from llm_handle.parser import extract_cmd
import os

bot = ChatGPT()
response = bot.ask("Can you give me a sample command in Mac terminal for checking the user names? Please give me the code directly.")
sample_response = """
Certainly! To list all user names on a Mac using the terminal, you can use the `dscl` command with the `list` option for the `/Users` node. Here's the command:
```
dscl . list /Users | grep -v '^_'
```

This will output a list of all user accounts on the system, excluding any system accounts that start with an underscore.
"""
# print("The response is:", response)
command = extract_cmd(str(response))
print("The command is:", command)

# execute the command in the mac terminal.
# Alert!! The execution should be really careful, because it can be dangerous.
# It is recommended to use this in a sandbox environment.
output = os.popen(command).read()
print("The output is:\n", output)
# Ideally, the output should be:
"""
daemon
uname
nobody
root"""

# delete the session in the end
bot.delete_conversation()