Refining UI

app.py

@@ -105,9 +105,9 @@ def diagnose(gpt_key, llama_key, gemini_key, top_p, temperature, symptoms):
         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: []"
-        gpt_message = generate_text_chatgpt(gpt_key, symptoms, temperature, top_p)
-        llama_message = generate_text_llama(llama_key, symptoms, temperature, top_p)
-        gemini_message = generate_text_gemini(gemini_key, symptoms, temperature, top_p)
+        gpt_message = generate_text_chatgpt(gpt_key, prompt, temperature, top_p)
+        llama_message = generate_text_llama(llama_key, prompt, temperature, top_p)
+        gemini_message = generate_text_gemini(gemini_key, prompt, temperature, top_p)
 
         outputs = [gpt_message, llama_message, gemini_message]
         output_counts = Counter(outputs)
@@ -117,7 +117,7 @@ def diagnose(gpt_key, llama_key, gemini_key, top_p, temperature, symptoms):
         majority_output = "Please add the symptoms data to start the ranking process."
         confidence = 0
     
-    return majority_output, confidence
+    return outputs, majority_output, confidence
 
 def update_model_components(selected_model):
     model_map = {