main.py

#!/usr/bin/python3

import openai
from dotenv import load_dotenv
import os
import subprocess
import time

try:
    from colorama import init, Fore
except ImportError:
    print(Fore.Red + "Colorama is not installed. Installing it...")
    subprocess.run(["pip", "install", "colorama"], check=True)
    from colorama import init, Fore

# Initialize colorama
init(autoreset=True)

# Install OpenAI 0.28 for error handling
subprocess.run(["pip", "install", "openai==0.28"])

# Specify the path to your .env file inside the 'src' folder
env_file_path = 'src/.env'

# Load environment variables from the specified .env file
load_dotenv(dotenv_path=env_file_path)

# Check if the script is running as root
print(Fore.BLUE + "Checking if the script is running as root!")

time.sleep(2)

if os.geteuid() != 0:
    print(Fore.YELLOW + "This script is not running as root, please run it as root!")
    exit(1)

time.sleep(1)

# Clear the terminal screen
os.system("clear")

import datetime
now = datetime.datetime.now()
print(Fore.BLUE + "Current date and time:")
print(Fore.CYAN + now.strftime("%d-%m-%Y %H:%M:%S"))

time.sleep(2)

import sys
print(Fore.GREEN + "Python version:")
print(Fore.RED + sys.version)
print(Fore.LIGHTMAGENTA_EX + "Python version:", sys.version_info)

time.sleep(2)

api_key = os.getenv("API_KEY")
if not api_key:
    print(Fore.RED + "API_KEY not found in the environment variables.")
    sys.exit(1)
openai.api_key = api_key

def chat_with_gpt3(prompt):
    response = openai.Completion.create(
        model="gpt-3.5-turbo",  # You can experiment with different models
        prompt=prompt,
        max_tokens=150  # Adjust based on your desired response length
    )

    return response.choices[0].text.strip()

def main():
    print(Fore.LIGHTMAGENTA_EX + "Welcome! You can chat with ChatGPT. Type 'exit' to end the conversation.")

    conversation_history = ""
    while True:
        user_input = input(Fore.LIGHTCYAN_EX + "You: ")
        if user_input.lower() == 'exit':
            break

        conversation_history += f"You: {user_input}\n"
        response = chat_with_gpt3(conversation_history)

        print(Fore.LIGHTBLUE_EX + f"ChatGPT: {response}")
        conversation_history += f"ChatGPT: {response}\n"

if __name__ == "__main__":
    main()