Update app.py

app.py

@@ -69,9 +69,8 @@ def get_label(newpred):
 
 def load_model():
     loaded_model = joblib.load('models/model.pkl') 
-
     return loaded_model
-    
+   
 def main(audio):
     newdf = transform_data(audio)
     loaded_model = load_model()
@@ -89,13 +88,13 @@ def on_click(audio):
 
         if label == 'Affirmative' and st.session_state.count == 0:
             st.session_state.pregunta_text = 'STC-110: Muchas gracias, soy un agente virtual de (MARCA) te recordamos que tienes pendiente el pago de (PRODUCTO) por un monto de (ADEUDO) pesos el cual se encuentra vencido, ¿Podrás pagar el día de hoy?\n'
-            st.session_state.affirmative_text = "**Afirmar:** 'Si', 'Está bien', 'Simón'"
-            st.session_state.negative_text = "**Negar:** 'No', 'Nel', 'Imposible',"
+            st.session_state.affirmative_text = "**Afirmar:** 'Si', 'Sí puedo', 'Creo que sí', 'Podría ser', 'Sí podré', 'Claro que sí', 'Está bien', 'Simón'"
+            st.session_state.negative_text = "**Negar:** 'No', 'Nel', 'Imposible', 'No creo', 'No tengo dinero', 'No tengo lana', 'No voy a pagar', 'No puedo'"
             st.session_state.count = 1
         elif label == 'Negative' and st.session_state.count == 0:
             st.session_state.pregunta_text = 'STC-105: Disculpe, ¿usted conoce a (NOMBRE)?'
-            st.session_state.affirmative_text = "**Afirmar:** 'Si', 'Es mi hermano', 'Es mi papá', 'Sé quién es', 'Correcto'"
-            st.session_state.negative_text = "**Negar:** 'No', 'Está equivocado', 'Número equivocado', 'Ya les dije que no'"
+            st.session_state.affirmative_text = "**Afirmar:** 'Si', 'Es mi hermano', 'Es mi papá', 'Sé quién es', 'Correcto', 'Sí lo conozco', 'Es mi hermana', 'Es mi hijo', 'Es mi mamá', 'Afirmativo', 'Así es', 'Simón', 'Sí dime'"
+            st.session_state.negative_text = "**Negar:** 'No', 'Está equivocado', 'Número equivocado', 'Ya les dije que no', 'No lo conozco', 'No sé quién es'"
             st.session_state.count = 2
         elif label == 'Affirmative' and st.session_state.count == 1:
             st.session_state.pregunta_text = 'Gracias, hemos registrado tu compromiso de pago para el día de hoy gracias por preferir (MARCA) que tengas un excelente día.\n'
@@ -104,8 +103,8 @@ def on_click(audio):
             st.session_state.count = 1
         elif label == 'Negative' and st.session_state.count == 1:
             st.session_state.pregunta_text = 'STC-130: ¿Podrás pagar antes del (FECHA_LIMITE_PAGO)?'
-            st.session_state.affirmative_text = "**Afirmar:** 'Si', 'Está bien', 'Simón'"
-            st.session_state.negative_text = "**Negar:** 'No', 'Nel', 'Imposible',"
+            st.session_state.affirmative_text = "**Afirmar:** 'Si', 'Sí puedo', 'Creo que sí', 'Podría ser', 'Sí podré', 'Claro que sí', 'Está bien', 'Simón'"
+            st.session_state.negative_text = "**Negar:** 'No', 'Nel', 'Imposible', 'No creo', 'No tengo dinero', 'No tengo lana', 'No voy a pagar', 'No puedo', 'Tampoco', 'Tampoco puedo'"
             st.session_state.count = 3
         elif label == 'Affirmative' and st.session_state.count == 2:
             st.session_state.pregunta_text = 'Gracias, por favor indíquele que lo hemos llamamos de (MARCA), y que volveremos a llamar. Gracias por preferir (MARCA) que tenga un excelente día.'
@@ -164,12 +163,12 @@ if __name__ == '__main__':
         st.session_state.pregunta_text = "STC-100: Hola, buen día. Nos comunicamos de (MARCA) me comunico con (NOMBRE)\n"
 
     if 'affirmative_text' not in st.session_state:
-        st.session_state.affirmative_text = "**Afirmar:** 'Si', 'Él habla', 'Ella habla', 'Correcto', 'Así es', 'Soy yo'"
+        st.session_state.affirmative_text = "**Afirmar:** 'Si', 'Él habla', 'Ella habla', 'Correcto', 'Así es', 'Soy yo', 'Sí, ella habla', 'Sí, diga', 'Sí, él habla', 'Sí soy', 'Sí dime'"
 
     if 'negative_text' not in st.session_state:
-        st.session_state.negative_text = "**Negar:** 'No', 'Con el hijo', 'Con el tío', 'Con el papá', 'Nel'"
+        st.session_state.negative_text = "**Negar:** 'No', 'Con el hijo', 'Con el tío', 'Nel', 'Incorrecto', 'Negativo', 'Equivocado', 'No se encuentra', 'No soy yo', 'Falleció', 'Con la esposa', 'Con el esposo', 'Con la tía'"
 
     if 'proba' not in st.session_state:
         st.session_state.proba = ""
 
-    audiorec_demo_app()
+    audiorec_demo_app()