correcting scores

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import time
 
 from loaddataset import ExtractRagBenchData
 from createmilvusschema import CreateMilvusDbSchema
@@ -36,7 +37,11 @@ print(rag_extracted_data.head(5))
 EmbedAllDocumentsAndInsert(QUERY_EMBEDDING_MODEL, rag_extracted_data, db_collection, window_size=WINDOW_SIZE, overlap=OVERLAP)
 """  
 
-def EvaluateRAGModel(query, evaluation_model):
+def EvaluateRAGModel(question, evaluation_model):
+    start_time = time.time()
+
+    query = question.strip()
+
     #invoke create milvus db function
     try:
         db_collection = CreateMilvusDbSchema()
@@ -70,18 +75,21 @@ def EvaluateRAGModel(query, evaluation_model):
     print(rmsecontextrel)
     print(rmsecontextutil)
     print(aucscore)
+    end_time = time.time()
+
+    execution_time = end_time - start_time
 
-    return answer, rmsecontextrel, rmsecontextutil, aucscore
+    return answer, rmsecontextrel, rmsecontextutil, aucscore, execution_time
 
 
 # Create Gradio UI
 with gr.Blocks() as iface:
-    gr.Markdown("## Capstone Project Group 10 - Model Evaluation")
+    gr.Markdown("## Capstone Project Group 10 ")
 
     with gr.Row():
         question_input = gr.Textbox(label="Enter your Question", lines=2)
         dropdown_input = gr.Dropdown(
-            ["LLaMA 3.3", "Mistral &B", "Model C"], 
+            ["LLaMA 3.3", "Mistral 7B"], 
             value="LLaMA 3.3", 
             label="Select a Model"
         )
@@ -91,7 +99,7 @@ with gr.Blocks() as iface:
     with gr.Row():
         with gr.Column():
             gr.Markdown("### Response")
-            response = gr.Textbox(interactive=False, show_label=False, lines=2)
+            response = gr.Textbox(interactive=False, show_label=False, lines=4)
 
     with gr.Row():
         with gr.Column():
@@ -106,8 +114,13 @@ with gr.Blocks() as iface:
             gr.Markdown("### AUCROC ADHERENCE")
             aucscore = gr.Textbox(interactive=False, show_label=False, lines=2)
 
+        with gr.Column():
+            gr.Markdown("### PROCESSING TIME")
+            processingTime = 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, rmsecontextrel, rmsecontextutil, aucscore])
+    submit_button.click(EvaluateRAGModel, inputs=[question_input, dropdown_input], outputs=[response, rmsecontextrel, rmsecontextutil, aucscore, processingTime])
 
 # Run the Gradio app
 iface.launch()