sentiment_analysis.py

"""
Data Scientist.: Dr.Eddy Giusepe Chirinos Isidro

Objetivo: Neste script vamos analisar o sentimento que contém 
          uma frase digitada pelo usuário 🤗.

"""
import os
import openai
from dotenv import find_dotenv, load_dotenv
_ = load_dotenv(find_dotenv()) # read local .env file
openai.api_key  = os.getenv('OPENAI_API_KEY')


def get_sentiment(text, model="gpt-3.5-turbo"):
    """Esta função usa a API da OpenAI para analisar o SENTIMENTO da frase."""
    prompt = f"""Forneça, apenas em uma palavra, o sentimento do seguinte texto: ```{text}```"""
    messages = [{"role": "user", "content": prompt}]
    
    response = openai.ChatCompletion.create(
        model=model,
        messages=messages,
        temperature=0,
    )
    
    sentiment = response.choices[0].message["content"]
    return sentiment



if __name__ == "__main__":
    text = input("Digite o texto a ser analisado: ")
    sentiment = get_sentiment(text)
    
    print(f"\033[033mO sentimento do texto é\033[m: {sentiment}")