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

import streamlit as st
import torch
from transformers import pipeline
from transformers import AutoTokenizer, AutoModelForQuestionAnswering

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

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.'''

st.write('Texto ejemplo que puedes copiar y pegar en el casillero de abajo: ', texto_default)

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

st.write('También puedes modificar las preguntas.')
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', '¿Cuándo se identificaron los problemas?')

pregs = [preg_1, preg_2, preg_3]

boton_preg = st.button('Enviar preguntas')

if boton_preg:

  st.write('Procesando texto ', texto[0:100], '...')

  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)

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