config.py

import os
import gspread
from oauth2client.service_account import ServiceAccountCredentials

# Read the authentication token from the environment variable
hugging_face_token = os.getenv("HUGGING_FACE_TOKEN")
replicate_token = os.getenv("REPLICATE_TOKEN")

# Google Sheets configuration
def init_google_sheets_client():
    scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
    creds = ServiceAccountCredentials.from_json_keyfile_name('tokyo-portal-326513-90aee094bab9.json', scope)
    return gspread.authorize(creds)

# Google Sheets name
google_sheets_name = "Chatbot Test"

# Define available models
huggingface_tokenizer = {
    "Meta-Llama-3-8B-Instruct": "meta-llama/Meta-Llama-3-8B-Instruct",
    "Llama-2-7B-Chat": "meta-llama/Llama-2-7b-chat-hf",
    #"Yi-6B-Chat": "01-ai/Yi-6B-Chat",
    #"Qwen2-7B-Instruct": "Qwen/Qwen2-7B-Instruct"
}

#Avaiable models for replicate
replicate_model= {
    "Meta-Llama-3-8B-Instruct": "meta/Meta-Llama-3-8B-Instruct",
    "Llama-2-7B-Chat": "meta/Llama-2-7b-chat-hf",
    #"Yi-6B-Chat": "01-ai/yi-34b-chat",
    #"Qwen2-7B-Instruct": "Qwen/Qwen2-7B-Instruct"
}

# List of models fine-tuned in 4-bit or 8-bit
#quantized_models = {
#    "Llama-3-8B-Finetuning-Stories": "rodrisouza/Llama-3-8B-Finetuning-Stories",
#}

# Default model name
default_model_name = "Meta-Llama-3-8B-Instruct"

# Define available user names
user_names = ["Laura Musto", "Brian Carpenter", "Germán Capdehourat", "Isabel Amigo", "Aiala Rosá", "Luis Chiruzzo", "Igancio Sastre", "Santiago Góngora", "Ignacio Remersaro", "Rodrigo Souza"]

MAX_INTERACTIONS = 5