fix: adding plot to main ui

backend/controller.py

@@ -59,15 +59,13 @@ def interference(
                 raise RuntimeError("There was an error in the selected XAI approach.")
 
         # call the explained chat function with the model instance
-        prompt_output, history_output, xai_interactive, xai_markup, xai_plot = (
-            explained_chat(
-                model=model,
-                xai=xai,
-                message=prompt,
-                history=history,
-                system_prompt=system_prompt,
-                knowledge=knowledge,
-            )
+        prompt_output, history_output, xai_interactive, xai_markup, xai_plot = explained_chat(
+            model=model,
+            xai=xai,
+            message=prompt,
+            history=history,
+            system_prompt=system_prompt,
+            knowledge=knowledge,
         )
     # if no XAI approach is selected call the vanilla chat function
     else:
@@ -86,7 +84,7 @@ def interference(
             no graphic will be displayed</h4></div>
             """,
             [("", "")],
-            None,
+            None
         )
 
     # return the outputs

main.py

@@ -272,7 +272,7 @@ with gr.Blocks(
             xai_selection,
             model_selection,
         ],
-        [user_prompt, chatbot, xai_interactive, xai_text],
+        [user_prompt, chatbot, xai_interactive, xai_text, xai_plot],
     )
     # function triggered by the enter key
     user_prompt.submit(
@@ -285,7 +285,7 @@ with gr.Blocks(
             xai_selection,
             model_selection,
         ],
-        [user_prompt, chatbot, xai_interactive, xai_text],
+        [user_prompt, chatbot, xai_interactive, xai_text, xai_plot],
     )
 
     # final row to show legal information