stdgpt.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# Builtins
import json
import os
import sys
import time

# Local
from Classes import auth as Auth
from Classes import chat as Chat
from Classes import spinner as Spinner

# Fancy stuff
import colorama
from colorama import Fore

colorama.init(autoreset=True)

# Check if config.json exists
if not os.path.exists("config.json"):
    print(">> config.json is missing. Please create it.")
    print(f"{Fore.RED}>> Exiting...")
    exit(1)

# Read config.json
with open("config.json", "r") as f:
    config = json.load(f)
    # Check if email & password are in config.json
    if "email" not in config or "password" not in config:
        print(">> config.json is missing email or password. Please add them.")
        print(f"{Fore.RED}>> Exiting...")
        exit(1)

    # Get email & password
    email = config["email"]
    password = config["password"]


def start_chat():
    expired_creds = Auth.expired_creds()
    # print(f"{Fore.GREEN}>> Checking if credentials are expired...")
    if expired_creds:
        # print(f"{Fore.RED}>> Your credentials are expired." + f" {Fore.GREEN}Attempting to refresh them...")
        open_ai_auth = Auth.OpenAIAuth(email_address=email, password=password)

        # print(f"{Fore.GREEN}>> Credentials have been refreshed.")
        open_ai_auth.begin()
        time.sleep(3)
        is_still_expired = Auth.expired_creds()
        if is_still_expired:
            print(f"{Fore.RED}>> Failed to refresh credentials. Please try again.")
            exit(1)
        else:
            # print(f"{Fore.GREEN}>> Successfully refreshed credentials.")
            pass
    # else:
    #     print(f"{Fore.GREEN}>> Your credentials are valid.")

    # print(f"{Fore.GREEN}>> Starting chat..." + Fore.RESET)
    previous_convo_id = None
    conversation_id = None
    access_token = Auth.get_access_token()
    while True:
        try:
            if access_token == "":
                print(f"{Fore.RED}>> Access token is missing in /Classes/auth.json.")
                exit(1)

            user_input = input()

            # spinner = Spinner.Spinner()
            # spinner.start(Fore.YELLOW + "Chat GPT is typing...")

            answer, previous_convo, convo_id = Chat.ask(auth_token=access_token,
                                                        prompt=user_input,
                                                        conversation_id=conversation_id,
                                                        previous_convo_id=previous_convo_id)

            if answer == "400" or answer == "401":
                # print(f"{Fore.RED}>> Your token is invalid. Attempting to refresh..")
                open_ai_auth = Auth.OpenAIAuth(email_address=email, password=password)
                open_ai_auth.begin()
                time.sleep(3)
                access_token = Auth.get_access_token()
            else:
                if previous_convo is not None:
                    previous_convo_id = previous_convo

                if convo_id is not None:
                    conversation_id = convo_id

                # spinner.stop()
                # print(f"Chat GPT: {answer}")
                print(answer)
            if "--once" in sys.argv:
                exit(0)
        except KeyboardInterrupt:
#            print(f"{Fore.RED}>> Exiting...")
            exit(1)


if __name__ == "__main__":
    start_chat()