updating UI

app.py

@@ -102,7 +102,7 @@ def sanitize_outputs(outputs):
     for output in outputs:
         output = output.replace("\n", " ")
         output = re.sub(r"(Diagnose:|Answer:)", "", output, flags=re.IGNORECASE).strip()
-        diagnoses = ["Psoriasis", "Arthritis", "Bronchial asthma", "Cervical spondylosis"]
+        diagnoses = ["Psoriasis", "Arthritis", "Bronchial Asthma", "Cervical spondylosis"]
         found_diagnoses = [disease for disease in diagnoses if disease in output]
 
         if found_diagnoses:
@@ -117,8 +117,8 @@ def diagnose(gpt_key, llama_key, gemini_key, top_p, temperature, symptoms):
 
     if symptoms:
         prompt = "Given the next set of symptoms, classify the diagnosis as one of the following: "
-        prompt += "Psoriasis, Arthritis, Bronchial asthma, Cervical spondylosis. Please only output the classified diagnosis and nothing after that."
-        prompt += "Choose only one among the words Psoriasis, Arthritis, Bronchial asthma or Cervical spondylosis"
+        prompt += "Psoriasis, Arthritis, Bronchial Asthma, Cervical spondylosis. Please only output the classified diagnosis and nothing after that."
+        prompt += "Choose only one among the words Psoriasis, Arthritis, Bronchial Asthma or Cervical spondylosis"
         prompt += "Do not list the symptoms again in the response. Do not add any additional text. Do not attempt to explain your answer."
         prompt += symptoms
         prompt += "Your Diagnosis: []"
@@ -132,33 +132,10 @@ 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 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: " + str(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: " + str(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: " + str(confidence) + "%"
-        else:
-            message = "Confidence output is not a valid value, please check what is wrong."
+        return gpt_message, llama_message, gemini_message, majority_output, confidence
     else:
-        message = "Please add the symptoms data to start the ranking process."
-    
-    return message
+        return {"is_valid": False, "message": f'<p style="color: red;">Please add the symptoms data to start the ranking process.</p>'}
+
 
 def update_model_components(selected_model):
     model_map = {
@@ -211,8 +188,15 @@ with gr.Blocks() as ui:
             symptoms = gr.Textbox(label="Add the symptom data in the input to receive diagnosis")
             llm_btn = gr.Button(value="Diagnose Disease", variant="primary", elem_id="diagnose", interactive=False)            
             symptoms.input(toggle_button, inputs=[symptoms, gpt_key, llama_key, gemini_key], outputs=llm_btn)
-            output = gr.Textbox(label="LLM output with majority vote and confidence", interactive=False, placeholder="Output will appear here...")
+            with gr.Row(equal_height=200):
+                with gr.Column(scale=1, min_width=150):
+                    gpt_message = gr.Textbox(label="GPT Output", interactive=False, placeholder="GPT Output")
+                    llama_message = gr.Textbox(label="LLaMA Output", interactive=False, placeholder="LLaMA Output")
+                    gemini_message = gr.Textbox(label="Gemini Output", interactive=False, placeholder="Gemini Output")
+                    majority_output = gr.Textbox(label="Majority Output", interactive=False, placeholder="Majority Output")
+                    confidence = gr.Textbox(label="Confidence Score (%)", interactive=False, placeholder="Confidence Score")
+
             llm_btn.click(fn=diagnose, inputs=[gpt_key, llama_key, gemini_key, top_p, temperature, symptoms], 
-                          outputs=output, api_name="LLM_Comparator")
+                          outputs=[gpt_message, llama_message, gemini_message, majority_output, confidence], api_name="LLM_Comparator")
 
 ui.launch(share=True)