Refining UI

app.py

@@ -97,12 +97,12 @@ def generate_text_llama(key, prompt, temperature, top_p):
     return response
 
 
-def diagnose(key, model, top_k, temperature, symptom_prompt):
+def diagnose(gpt_key, llama_key, gemini_key, top_p, temperature, symptoms):
 
-    if symptom_prompt:
-        gpt_message = generate_text_chatgpt(key, symptom_prompt, temperature, top_k)
-        llama_message = generate_text_llama(key, symptom_prompt, temperature, top_k)
-        gemini_message = generate_text_gemini(key, symptom_prompt, temperature, top_k)
+    if symptoms:
+        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)
 
         outputs = [gpt_message, llama_message, gemini_message]
         output_counts = Counter(outputs)