relu

#!/usr/bin/env python3
import sys, os
from utils_relu.moduls import *

RUTA_ARCHIVO = "/".join(os.path.abspath(__file__).split("/")[:-1])
list_params = sys.argv
JSON_CONFIG = load_json()
LANG = load_json(f"languages/{JSON_CONFIG['language']}.json")

list_lang = [i[:-len(".json")] for i in os.listdir(f"{RUTA_ARCHIVO}/utils_relu/languages")]

if JSON_CONFIG["language"] not in list_lang:
    print (ERROR_NOT.format(f"The system detected a non-existent language in the {RUTA_ARCHIVO}/utils_relu/languages directory, resetting it to English.\nAVAILABLE LANGUAGES: {list_lang}"))
    JSON_CONFIG["language"] = "en"
    save_json(JSON_CONFIG)
    LANG = load_json(f"languages/{JSON_CONFIG['language']}.json")

if len(sys.argv) == 1:
    print (NOTIFICATION.format(LANG["help_param"]))
    sys.exit()

if "--name" in sys.argv[1]:
    try:
        name = sys.argv[2]
    except IndexError:
        print (ERROR_NOT.format(LANG["name_not_found"]))
        sys.exit()

    JSON_CONFIG["name"] = name
    save_json(JSON_CONFIG)
    print (NOTIFICATION.format(LANG["success_save_name"]))
    sys.exit()

if sys.argv[1] == "--clear" or sys.argv[1] == "-c":
    try:
        save_chat(None)
    except Exception as Error:
        notifyc(LANG["error_act"], LANG["error_desc_clear"].format(Error), False)
    else:
        notifyc(LANG["success_act"], LANG["success_desc_clear"], True)
    sys.exit()

if "--key" in sys.argv[1]:
    try:
        key = sys.argv[2]
    except IndexError:
        print (ERROR_NOT.format(LANG["key_not_found"]))
        sys.exit()

    JSON_CONFIG["key_gemini"] = key
    save_json(JSON_CONFIG)
    print (NOTIFICATION.format(LANG["success_save_apikey"]))
    sys.exit()

if "--lang" in sys.argv[1]:
    try:
        lang = sys.argv[2]
    except IndexError:
        print (ERROR_NOT.format(LANG["lang_not_found"]))
        sys.exit()

    JSON_CONFIG["language"] = lang
    save_json(JSON_CONFIG)
    LANG = load_json(f"languages/{lang}.json")
    try:
        print (NOTIFICATION.format(LANG["success_save_lang"]))
    except:
        print (ERROR_NOT.format(f"The system detected a non-existent language in the {RUTA_ARCHIVO}/utils_relu/languages/{lang}.json directory, resetting it to English.\nAVAILABLE LANGUAGES: {list_lang}"))
        JSON_CONFIG["language"] = "en"
        save_json(JSON_CONFIG)
        LANG = load_json(f"languages/{JSON_CONFIG['language']}.json")

    sys.exit()

if JSON_CONFIG["key_gemini"] == "":
    print (ERROR_NOT.format(LANG["help_api_key"]))
    sys.exit()

if "--remember" in sys.argv or "-r" in sys.argv:
    conversation = load_chat()

    if conversation == None:
        print (NOTIFICATION.format(LANG["empty_chat"]))
    else:
        import google.generativeai as genai

        genai.configure(api_key=JSON_CONFIG["key_gemini"])
        modelo = genai.GenerativeModel("gemini-pro")
        chat = modelo.start_chat(history=conversation)
        MODELO = conversation[-1]
        USUARIO = conversation[-2]
        print (f"""Messages: {len(conversation)} | Tokens: {chat.model.count_tokens(contents=conversation).total_tokens}\n{USER.format(USUARIO.parts[0].text)}\n{MODEL.format(MODELO.parts[0].text)}""")
    sys.exit()

if sys.argv[1].startswith("-"):
    print (ERROR_NOT.format(LANG["param_unrecognized"].format(sys.argv[1])))
    sys.exit()

prompt = " ".join(sys.argv[1:])

RUTA = os.getcwd()

print (NOTIFICATION.format(LANG["in_process"]))

import google.generativeai as genai

genai.configure(api_key=JSON_CONFIG["key_gemini"])
modelo = genai.GenerativeModel("gemini-pro")

conversation = load_chat()
chat = modelo.start_chat(history=conversation)

if conversation == None or chat.model.count_tokens(contents=conversation).total_tokens >= 25000 or chat.model.count_tokens(contents=conversation).total_tokens <= 10:
    data_OS = get_dataOS()
    msg = ""
    msg += "\n".join([f"- {name}: {data_OS[name]}" for name in data_OS])
    prompt_f = LANG["prompt_relu"].format(msg)
    response = chat.send_message(prompt_f)
    conversation = chat.history
    if chat.model.count_tokens(contents=conversation).total_tokens >= 25000:
        conversation = chat.history[3:]

chat = modelo.start_chat(history=conversation)

print("\033[F\033[K", end="")

try:
    for evento in chat.send_message(prompt, stream=True):
        print (FONDO_NEGRO+evento.text, end='', flush=True)
except Exception as Error:
    if "400 API key not valid" in str(Error).split("\n")[0]:
        print (ERROR_NOT.format(LANG["apikey_invalid"]))
    else:
        print (ERROR_NOT.format(Error))
    sys.exit()

save_chat(chat.history[:50])

print (FONDO_NEUTRO)