# -*- coding: utf-8 -*-
'''
José Carlos Machicao
GestioDinámica
Fecha de producción: 2022_01_31
Fecha de actualización 2022_02_15
'''

st.image('gdmk.png', width=150)

import streamlit as st
import torch
from transformers import pipeline
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
tokenizer = AutoTokenizer.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es")
modelo_qa = AutoModelForQuestionAnswering.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es")
preguntas = pipeline('question-answering', model=modelo_qa, tokenizer=tokenizer)

texto_default = '''
Este martes, la fiscal de la Nación, Zoraida Ávalos, amplió la investigación preliminar sobre la presunta injerencia de representantes del Poder Ejecutivo en el proceso de ascensos de la Policía Nacional del Perú. En esta indagación se encuentran involucrado el exsecretario de Palacio, Bruno Pacheco, y el excomandante general de la PNP, Javier Gallardo.
A través de un comunicado, el Ministerio Público comunicó que el exsubcomandante general de la PNP, Javier Bueno, dio su testimonio sobre las presuntas irregularidades en el proceso de ascensos policiales ante el fiscal Luzgardo González Rodríguez, coordinador del área de Enriquecimiento Ilícito y Denuncias Constitucionales de la Fiscalía de la Nación.
En noviembre del 2021, Bueno Victoriano protagonizó un incidente con Gallardo luego de presenciar una serie de irregularidades tras las entrevistas que realizó la Junta Selectora de Coroneles para Oficiales de Servicio, Abogados y Médicos a los coronales postulantes.
En enero de este año, el ex número 2 de la PNP reveló que se habrían cobrado entre 25.000 y 40.000 dólares por cada ascenso al grado de general, declaración que reafirmó ante la Comisión de Defensa y Orden Interno del Congreso.
Javier Bueno también aseveró que quienes coordinaban el proceso fueron el exsecratario presidencial, Bruno Pacheco, y el excomandante Javier Gallardo, quien fue pasado al retiro tras las desavenencias con el exministro del Interior, Avelino Guillen.
“A los investigados se les imputa haber realizado gestiones y/o haber obtenido beneficios económicos indebidos a cambio de favorecer a coroneles que habrían visitado la sede de Palacio de Gobierno antes de lograr el ascenso al grado de general PNP”, se lee en el comunicado.
'''

texto = st.text_area('Cargue un texto de hasta 5 hojas, 1500 palabras, ó 15000 caracteres: ', height=200)

preg_1 = st.text_input('Pregunta 1', '¿Cuál es el problema?')
preg_2 = st.text_input('Pregunta 2', '¿Quienes son los responsables?')
preg_3 = st.text_input('Pregunta 3', '¿Cómo se podría resolver el problema?')

pregs = [preg_1, preg_2, preg_3]

boton_preg2 = st.button('Enviar preguntas')

st.write(texto)

if boton_preg2:
    for preg in pregs:
      respuesta = preguntas({'question': preg, 'context': texto})
      st.write(preg)
      st.write(respuesta['answer'])
      st.write('\n Confiabilidad: ', respuesta['score'])