Update app.py

app.py

@@ -110,8 +110,6 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
                         line_breaks=True,
                         likeable=False,
                         layout='panel')
-    
-    #chatbot = gr.Chatbot(label="Aira").style(height=500)
                          
     msg = gr.Textbox(label="Escreva uma pergunta ou instrução para Aira ...", placeholder="Olá Aira, como vai você?")
 
@@ -207,12 +205,14 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
                 toxicity = toxicityModel(**toxicity_tokens)[0].item()
                 toxicities.append(toxicity)
                 toxicity_threshold = 5
-        
-        ordered_generations = sorted(zip(decoded_text, rewards, toxicities), key=lambda x: x[1], reverse=True)
 
         if safety == "On":
+            ordered_generations = sorted(zip(decoded_text, rewards, toxicities), key=lambda x: x[1], reverse=True)
             ordered_generations = [(x, y, z) for (x, y, z) in ordered_generations if z >= toxicity_threshold]
 
+        else:
+            ordered_generations = sorted(zip(decoded_text, rewards), key=lambda x: x[1], reverse=True)
+
         if len(ordered_generations) == 0:
           bot_message = """Peço desculpa pelo incómodo, mas parece que não foi possível identificar respostas adequadas que cumpram as nossas normas de segurança. Infelizmente, isto indica que o conteúdo gerado pode conter elementos de toxicidade ou pode não ajudar a responder à sua mensagem. A sua opinião é valiosa para nós e esforçamo-nos por garantir uma conversa segura e construtiva. Não hesite em fornecer mais pormenores ou colocar quaisquer outras questões, e farei o meu melhor para o ajudar."""