correcting scores

app.py

@@ -71,6 +71,8 @@ def EvaluateRAGModel(query, evaluation_model):
     print(rmsecontextutil)
     print(aucscore)
 
+    return answer, rmsecontextrel, rmsecontextutil, aucscore
+
 
 # Create Gradio UI
 with gr.Blocks() as iface:
@@ -88,24 +90,24 @@ with gr.Blocks() as iface:
 
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### Output 1")
+            gr.Markdown("### Response")
             response = gr.Textbox(interactive=False, show_label=False, lines=2)
 
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### Output 2")
-            output2 = gr.Textbox(interactive=False, show_label=False, lines=2)
+            gr.Markdown("### RMSE-CONTEXT RELEVANCE")
+            rmsecontextrel = gr.Textbox(interactive=False, show_label=False, lines=2)
     
         with gr.Column():
-            gr.Markdown("### Output 3")
-            output3 = gr.Textbox(interactive=False, show_label=False, lines=2)
+            gr.Markdown("### RMSE-CONTEXT UTILIZATION")
+            rmsecontextutil = gr.Textbox(interactive=False, show_label=False, lines=2)
 
         with gr.Column():
-            gr.Markdown("### Output 4")
-            output4 = gr.Textbox(interactive=False, show_label=False, lines=2)
+            gr.Markdown("### AUCROC ADHERENCE")
+            aucscore = gr.Textbox(interactive=False, show_label=False, lines=2)
 
     # Connect submit button to evaluation function
-    submit_button.click(EvaluateRAGModel, inputs=[question_input, dropdown_input], outputs=[response, output2, output3, output4])
+    submit_button.click(EvaluateRAGModel, inputs=[question_input, dropdown_input], outputs=[response, rmsecontextrel, rmsecontextutil, aucscore])
 
 # Run the Gradio app
 iface.launch()