Add context

app.py

@@ -25,7 +25,7 @@ rag = RAG(
 def generate(prompt, model_parameters):
     try:
         output, context = rag.get_response(prompt, model_parameters)
-        return output
+        return output, context
     except HTTPError as err:
         if err.code == 400:
             gr.Warning(
@@ -54,9 +54,8 @@ def submit_input(input_, num_chunks, max_new_tokens, repetition_penalty, top_k,
         "TEMPERATURE": temperature
     }
 
-    output = generate(input_, model_parameters)
-
-    return output
+    output, context = generate(input_, model_parameters)
+    return output, context
 
 
 def change_interactive(text):
@@ -69,6 +68,8 @@ def clear():
     return (
         None, 
         None,
+        None,
+        None,
         gr.Slider(value=100),
         gr.Slider(value=1.2),
         gr.Slider(value=50),
@@ -166,7 +167,15 @@ def gradio_app():
 
             with gr.Column(variant="panel"):
                 output = Textbox(
-                    lines=11, label="Output", interactive=False, show_copy_button=True
+                    lines=11, 
+                    label="Output", 
+                    interactive=False, 
+                    show_copy_button=True
+                )
+                context_evaluation = Textbox(
+                    label="Context evaluation", 
+                    interactive=False, 
+                    show_copy_button=True
                 )
                 with gr.Row(variant="panel"):
                     clear_btn = Button(
@@ -175,6 +184,7 @@ def gradio_app():
                     submit_btn = Button("Submit", variant="primary", interactive=False)
 
 
+
                 
 
         input_.change(
@@ -197,7 +207,7 @@ def gradio_app():
         clear_btn.click(
             fn=clear, 
             inputs=[], 
-            outputs=[input_, output] + parameters_compontents,
+            outputs=[input_, output, context_evaluation] + parameters_compontents,
               queue=False, 
               api_name=False
         )
@@ -205,7 +215,7 @@ def gradio_app():
         submit_btn.click(
             fn=submit_input, 
             inputs=[input_]+ parameters_compontents, 
-            outputs=[output], 
+            outputs=[output,context_evaluation],
             api_name="get-results"
         )
 
@@ -216,7 +226,7 @@ def gradio_app():
                         ["""Què és l'EADOP (Entitat Autònoma del Diari Oficial i de Publicacions)?"""],
                     ],
                     inputs=input_,
-                    outputs=output,
+                    outputs=[output,context_evaluation],
                     fn=submit_input,
                 )
                 gr.Examples(
@@ -224,7 +234,7 @@ def gradio_app():
                         ["""Què diu el decret sobre la senyalització de les begudes alcohòliques i el tabac a Catalunya?"""],
                     ],
                     inputs=input_,
-                    outputs=output,
+                    outputs=[output,context_evaluation],
                     fn=submit_input,
                 )
                 gr.Examples(
@@ -232,7 +242,7 @@ def gradio_app():
                         ["""Com es pot inscriure una persona al Registre de catalans i catalanes residents a l'exterior?"""],
                     ],
                     inputs=input_,
-                    outputs=output,
+                    outputs=[output,context_evaluation],
                     fn=submit_input,
                 )
                 gr.Examples(
@@ -240,7 +250,7 @@ def gradio_app():
                         ["""Quina és la finalitat del Servei Meterològic de Catalunya ?"""],
                     ],
                     inputs=input_,
-                    outputs=output,
+                    outputs=[output,context_evaluation],
                     fn=submit_input,
                 )