File size: 2,734 Bytes
416f2ae
996185e
63b3ab3
 
416f2ae
67b19a2
 
 
 
 
 
63b3ab3
 
67b19a2
 
 
 
 
 
d363b77
90261a5
e0cffe2
 
7d67aa9
915b8f9
be6ee88
ae6d0a2
38d24c6
d363b77
 
73b8e48
d363b77
63b3ab3
d363b77
 
915b8f9
d363b77
 
 
 
 
 
 
73b8e48
d363b77
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import gradio as gr
import examples
from document_quieries import ExtractiveProposalQueries
from config import Config

query = ExtractiveProposalQueries(es_host = Config.es_host, es_index = Config.proposals_index,
                                  es_user = Config.es_user, es_password = Config.es_password,
                                  reader_name_or_path = Config.reader_model_name_or_path,
                                  use_gpu = Config.use_gpu)

def search(question):
  p1_result = query.search_by_query(query = question, retriever_top_k = 2, reader_top_k = 1, es_index = "petro")
  p2_result = query.search_by_query(query = question, retriever_top_k = 2, reader_top_k = 1, es_index = "rodolfo")

  return [p1_result[0].answer, 
          p1_result[0].context,
          p2_result[0].answer,
          p2_result[0].context]

description = """
   <p> 
    El debate ha sido un ausente en la segunda vuelta por la presidencia de Colombia del 2022. 
    para ayudar con el conocimiento de las propuestas de los candidatos construí este espacio,
    puede hacerle preguntas a la propuestas oficiales de los candidatos (Junio de 2022).
    Un pequeño aporte a un fin de semana determinante en la democracia Colombiana.
    <br></br>
    <br>Este espacio hace parte de un proyecto open source que busca ayudar con el entendimiento de temas relevantes para el país, como estas elecciones y la memoria histórica Colombiana. <a href= "https://github.com/jorge-henao/ask_to_democracy">repo en github con FastAPI</a></br>
    Por: Jorge Henao 🇨🇴 <a href="https://twitter.com/jhenaotw" target='_blank'>Twitter</a> <a href="https://www.linkedin.com/in/henaojorge" target='_blank'/>LinkedIn</a>
    </p> 
"""
 
iface = gr.Interface(fn=search,
                      title= "Pregúntale a la democracia 🇨🇴 - Elecciones precidenciales Colombia 2022",
                      description = description,
                      inputs=[
                              gr.inputs.Textbox(lines=2, label="hazle una pregunta a los candidatos [Abajo algunos ejemplos]👇", placeholder='que va a hacer con la deuda del ICETEX?')
                              ],
                      outputs=[
                                gr.outputs.Textbox(label="Respuesta de Petro"),
                                gr.outputs.Textbox(label="Contexto de la propuesta"),
                                gr.outputs.Textbox(label="Respuesta de Rodolfo Hernandez"),
                                gr.outputs.Textbox(label="Contexto de la propuesta")                    
                              ],
                      examples = examples.examples,
                      theme = 'peach'
                     )
iface.launch(enable_queue=True)