File size: 1,021 Bytes
ec38040
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
"""
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}")