Refining UI

app.py

@@ -100,6 +100,11 @@ def generate_text_llama(key, prompt, temperature, top_p):
 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: 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: []"
         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)