Refining UI

app.py

@@ -132,28 +132,29 @@ def diagnose(gpt_key, llama_key, gemini_key, top_p, temperature, symptoms):
         majority_output, majority_count = output_counts.most_common(1)[0]
         confidence = int((majority_count / len(outputs)) * 100)
 
-        if majority_output > 0:
-            if confidence < 50:
-                message = "LLMs do not seem to agree on one output." + "\n"
-                message += " GPT output: " + gpt_message + "\n"
-                message += " Llama output: " + llama_message + "\n"
-                message += " Gemini output: " + gemini_message + "\n"
-                message += " Confidence score: " + confidence + "%"
-            elif confidence > 50:
-                message = "LLMs seem to agree on one output." + "\n"
-                message += " GPT output: " + gpt_message + "\n"
-                message += " Llama output: " + llama_message + "\n"
-                message += " Gemini output: " + gemini_message + "\n"
-                message += " Confidence score: " + confidence + "%"
-            elif confidence >= 100:
-                message = "All LLMs agree on the same output." + "\n"
-                message += " GPT output: " + gpt_message + "\n"
-                message += " Llama output: " + llama_message + "\n"
-                message += " Gemini output: " + gemini_message + "\n"
-                message += " Confidence score: " + confidence + "%"
+        if confidence < 50:
+            message = "LLMs do not seem to agree on one output." + "\n"
+            message += " GPT output: " + gpt_message + "\n"
+            message += " Llama output: " + llama_message + "\n"
+            message += " Gemini output: " + gemini_message + "\n"
+            message += " Majority output: " + majority_output + "\n"
+            message += " Confidence score: " + confidence + "%"
+        elif confidence > 50:
+            message = "LLMs seem to agree on one output." + "\n"
+            message += " GPT output: " + gpt_message + "\n"
+            message += " Llama output: " + llama_message + "\n"
+            message += " Gemini output: " + gemini_message + "\n"
+            message += " Majority output: " + majority_output + "\n"
+            message += " Confidence score: " + confidence + "%"
+        elif confidence >= 100:
+            message = "All LLMs agree on the same output." + "\n"
+            message += " GPT output: " + gpt_message + "\n"
+            message += " Llama output: " + llama_message + "\n"
+            message += " Gemini output: " + gemini_message + "\n"
+            message += " Majority output: " + majority_output + "\n"
+            message += " Confidence score: " + confidence + "%"
         else:
-            message = "Majority output is not a valid value, please check what is wrong."
-        return message
+            message = "Confidence output is not a valid value, please check what is wrong."
     else:
         message = "Please add the symptoms data to start the ranking process."