Added sources and modified layout

app.py

@@ -24,8 +24,8 @@ rag = RAG(
 
 def generate(prompt, model_parameters):
     try:
-        output, context = rag.get_response(prompt, model_parameters)
-        return output, context
+        output, context, source = rag.get_response(prompt, model_parameters)
+        return output, context, source
     except HTTPError as err:
         if err.code == 400:
             gr.Warning(
@@ -54,8 +54,13 @@ def submit_input(input_, num_chunks, max_new_tokens, repetition_penalty, top_k,
         "TEMPERATURE": temperature
     }
 
-    output, context = generate(input_, model_parameters)
-    return output, context
+    output, context, source = generate(input_, model_parameters)
+    sources_markup = ""
+
+    for url in source:
+        sources_markup += f'<a href="{url}" target="_blank">{url}</a><br>'
+        
+    return output.strip(), sources_markup, context
 
 
 def change_interactive(text):
@@ -70,6 +75,7 @@ def clear():
         None,
         None,
         None,
+        None,
         gr.Slider(value=100),
         gr.Slider(value=1.2),
         gr.Slider(value=50),
@@ -172,18 +178,21 @@ def gradio_app():
 
             with gr.Column(variant="panel"):
                 output = Textbox(
-                    lines=11, 
+                    lines=10, 
                     label="Output", 
                     interactive=False, 
                     show_copy_button=True
                 )
-                context_evaluation = Textbox(
-                    label="Context evaluation", 
-                    interactive=False, 
-                    show_copy_button=True
-                )
-
-
+                with gr.Accordion("Sources and context:", open=False):
+                    source_context = gr.Markdown(
+                    )
+                    with gr.Accordion("See full context evaluation:", open=False):
+                        context_evaluation = gr.Markdown(
+                            # show_label=False,
+                            # interactive=False, 
+                            # autoscroll=False,
+                            # show_copy_button=True
+                        )
                 
 
         input_.change(
@@ -206,7 +215,7 @@ def gradio_app():
         clear_btn.click(
             fn=clear, 
             inputs=[], 
-            outputs=[input_, output, context_evaluation] + parameters_compontents,
+            outputs=[input_, output, source_context, context_evaluation] + parameters_compontents,
               queue=False, 
               api_name=False
         )
@@ -214,7 +223,7 @@ def gradio_app():
         submit_btn.click(
             fn=submit_input, 
             inputs=[input_]+ parameters_compontents, 
-            outputs=[output,context_evaluation],
+            outputs=[output, source_context, context_evaluation],
             api_name="get-results"
         )
 
@@ -225,7 +234,7 @@ def gradio_app():
                         ["""Què és l'EADOP (Entitat Autònoma del Diari Oficial i de Publicacions)?"""],
                     ],
                     inputs=input_,
-                    outputs=[output,context_evaluation],
+                    outputs=[output, source_context, context_evaluation],
                     fn=submit_input,
                 )
                 gr.Examples(
@@ -233,7 +242,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,context_evaluation],
+                    outputs=[output, source_context, context_evaluation],
                     fn=submit_input,
                 )
                 gr.Examples(
@@ -241,7 +250,7 @@ def gradio_app():
                         ["""Com es pot inscriure una persona al Registre de catalans i catalanes residents a l'exterior?"""],
                     ],
                     inputs=input_,
-                    outputs=[output,context_evaluation],
+                    outputs=[output, source_context, context_evaluation],
                     fn=submit_input,
                 )
                 gr.Examples(
@@ -249,7 +258,7 @@ def gradio_app():
                         ["""Quina és la finalitat del Servei Meterològic de Catalunya ?"""],
                     ],
                     inputs=input_,
-                    outputs=[output,context_evaluation],
+                    outputs=[output, source_context, context_evaluation],
                     fn=submit_input,
                 )