Update app.py

app.py

@@ -111,18 +111,22 @@ if name and model_name:
             contents = result["content"]
 
             label = qrels[str(qid)].get(str(result_id), 0)
+            label_text = "None"
             if label == 3:
                 style = "style=\"color:rgb(231, 95, 43);\""
+                label_text = "High"
             elif label == 2:
                 style = "style=\"color:rgb(238, 147, 49);\""
+                label_text = "Medium"
             elif label == 1:
                 style = "style=\"color:rgb(241, 177, 118);\""
+                label_text = "Low"
             else:
                 style = "style=\"color:grey;\""
 
             print(qid, result_id, label, style)
             # output = f'<div class="row"> <b>Rank</b>: {i+1} | <b>Document ID</b>: {result_id} | <b>Score</b>:{result_score:.2f}</div>'
-            output = f'<div class="row" {style}> <b>Rank</b>: {i+1} | <b>Document ID</b>: {result_id}'
+            output = f'<div class="row" {style}> <b>Rank</b>: {i+1} | <b>Document ID</b>: {result_id} | <b>True Relevance ID</b>: {label_text}'
     
             try:
                 st.write(output, unsafe_allow_html=True)