Frontend/corretor_gemini/gemini_corretor.py [24:53]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
class Correcao():
    """
    Classe que representa a resposta do corretor, possui as informações da redação, a nota e os comentários.
    """
    def __init__(self, redacao_data, correcao_response, sumarizacao=None):
        self.redacao_data = redacao_data
        self.correcao_data = json.loads(correcao_response)
        self.correcao_df = pd.DataFrame(self.correcao_data)
        self.sumarizacao = sumarizacao

    def get_nota_comptencias(self):
        """
        Calcula a nota de cada competência da redação.
        
        Returns:
            pd.Series: Série com as notas de cada competência.
        """
        nota_competencias = self.correcao_df.apply(calcular_nota_competencia, axis=0)
        nota_competencias = nota_competencias.T
        nota_competencias.columns = ['Competência 2', 'Competência 3', 'Competência 4', 'Competência 5']
        return nota_competencias

    def get_nota_enem(self):
        """
        Calcula a nota total da redação.
        
        Returns:
            int: Nota total da redação.
        """
        return self.get_nota_comptencias().sum()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



IAC/temp/function-source/src/corretor_gemini/gemini_corretor.py [30:59]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
class Correcao():
    """
    Classe que representa a resposta do corretor, possui as informações da redação, a nota e os comentários.
    """
    def __init__(self, redacao_data, correcao_response, sumarizacao=None):
        self.redacao_data = redacao_data
        self.correcao_data = json.loads(correcao_response)
        self.correcao_df = pd.DataFrame(self.correcao_data)
        self.sumarizacao = sumarizacao

    def get_nota_comptencias(self):
        """
        Calcula a nota de cada competência da redação.
        
        Returns:
            pd.Series: Série com as notas de cada competência.
        """
        nota_competencias = self.correcao_df.apply(calcular_nota_competencia, axis=0)
        nota_competencias = nota_competencias.T
        nota_competencias.columns = ['Competência 2', 'Competência 3', 'Competência 4', 'Competência 5']
        return nota_competencias

    def get_nota_enem(self):
        """
        Calcula a nota total da redação.
        
        Returns:
            int: Nota total da redação.
        """
        return self.get_nota_comptencias().sum()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -