Update app.py

app.py

@@ -114,12 +114,13 @@ def predict(query,data):
         list_outputs = []
         for i in range(k):
             temp = [df.iloc[n] for n in range(k)][i]
-            text = ''
-            text += 'Probabilidades: '+ temp.Probabilidades + '\n\n' 
-            text += 'Respuesta: ' +temp.Respuesta + '\n\n' 
-            text += 'Contexto: '+temp.Contexto + '\n\n' 
-            list_outputs.append(text)
-        return list_outputs
+            tupla = (temp.Respuesta, temp.Contexto, temp.Probabilidades)
+            # text = ''
+            # text += 'Probabilidades: '+ temp.Probabilidades + '\n\n' 
+            # text += 'Respuesta: ' +temp.Respuesta + '\n\n' 
+            # text += 'Contexto: '+temp.Contexto + '\n\n' 
+            list_outputs.append(tupla)
+        return list_outputs[0]
     except Exception as e:
         print(e)
         print(st)
@@ -192,13 +193,13 @@ def predict(query,data):
     
     list_outputs = []
     for i in range(k):
-        text = ''
         temp = [df.iloc[n] for n in range(k)][i]
-        text += 'Probabilidades: '+ temp.Probabilidades + '\n\n' 
-        text += 'Respuesta: ' +temp.Respuesta + '\n\n' 
-        text += 'Contexto: '+temp.Contexto + '\n\n' 
-  
-        list_outputs.append(text)
+        tupla = (temp.Respuesta, temp.Contexto, temp.Probabilidades)
+        # text = ''
+        # text += 'Probabilidades: '+ temp.Probabilidades + '\n\n' 
+        # text += 'Respuesta: ' +temp.Respuesta + '\n\n' 
+        # text += 'Contexto: '+temp.Contexto + '\n\n' 
+        list_outputs.append(tupla)
     
     return list_outputs[0]
 
@@ -237,6 +238,7 @@ with gr.Blocks() as demo:
     ask = gr.Button("Preguntar")
     gr.Markdown("## Respuesta")
     answer = gr.Markdown(label="Respuesta")
+    prob = gr.Markdown(label="Probabilidades")
     with gr.Accordion("Contexto", open=False):
         gr.Markdown(
             "### Contexto\n\nEl siguiente contexto ha sido utilizado para generar la respuesta:")
@@ -244,7 +246,9 @@ with gr.Blocks() as demo:
     # ask.click(fn=do_ask, inputs=[query, buildb,
     #                              dataset], outputs=[answer, context])
     ask.click(fn=predict, inputs=[query,
-                                 file], outputs=[answer])
+                                 file], outputs=[answer, context, prob])
+
+    examples = ["¿Cuándo suelen comenzar las adicciones?","Entrevista Miguel Ruiz.txt"]
 
 demo.queue(concurrency_count=20)
 demo.launch(show_error=True)