Update app.py

app.py

@@ -93,6 +93,80 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
     return "", chat_history
   '''
   def respond(message, chat_history, base_model = "microsoft/phi-2", device=device): # "meta-llama/Meta-Llama-3-70B"
+   if base_model != "microsoft/phi-2":
+    # Random greetings in list format
+    greetings = [
+        "hello!",'hello', 'hii !', 'hi', "hi there!",  "hi there!", "heyy", 'good morning', 'good afternoon', 'good evening'
+        "hey", "how are you", "how are you?", "how is it going", "how is it going?",
+        "what's up?", "how are you?",
+        "hey, how are you?", "what is popping"
+        "good to see you!", "howdy!",
+        "hi, nice to meet you.", "hiya!",
+        "hi", "hi, what's new?",
+        "hey, how's your day?", "hi, how have you been?", "greetings",
+        ]
+    # Random Greetings responses
+    responses = [
+        "Thank you for using our medical chatbot. Please provide the symptoms you're experiencing, and I'll do my best to predict the possible disease.",
+        "Hello! I'm here to help you with medical predictions based on your symptoms. Please describe your symptoms in as much detail as possible.",
+        "Greetings! I am a specialized medical chatbot trained to predict potential diseases based on the symptoms you provide. Kindly list your symptoms explicitly.",
+        "Welcome to the medical chatbot. To assist you accurately, please share your symptoms in explicit detail.",
+        "Hi there! I'm a medical chatbot specialized in analyzing symptoms to suggest possible diseases. Please provide your symptoms explicitly.",
+        "Hey! I'm your medical chatbot. Describe your symptoms with as much detail as you can, and I'll generate potential disease predictions.",
+        "How can I assist you today? I'm a medical chatbot trained to predict diseases based on symptoms. Please be explicit while describing your symptoms.",
+        "Hello! I'm a medical chatbot capable of predicting diseases based on the symptoms you provide. Your explicit symptom description will help me assist you better.",
+        "Greetings! I'm here to help with medical predictions. Describe your symptoms explicitly, and I'll offer insights into potential diseases.",
+        "Hi, I'm the medical chatbot. I've been trained to predict diseases from symptoms. The more explicit you are about your symptoms, the better I can assist you.",
+        "Hi, I specialize in medical predictions based on symptoms. Kindly provide detailed symptoms for accurate disease predictions.",
+        "Hello! I'm a medical chatbot with expertise in predicting diseases from symptoms. Please describe your symptoms explicitly to receive accurate insights.",
+        ]
+    # Random goodbyes
+    goodbyes = [
+        "farewell!",'bye', 'goodbye','good-bye', 'good bye', 'bye', 'thank you', 'later', "take care!",
+        "see you later!", 'see you', 'see ya', 'see-you', 'thanks', 'thank', 'bye bye', 'byebye'
+        "catch you on the flip side!", "adios!",
+        "goodbye for now!", "till we meet again!",
+        "so long!", "hasta la vista!",
+        "bye-bye!", "keep in touch!",
+        "toodles!", "ciao!",
+        "later, gator!", "stay safe and goodbye!",
+        "peace out!", "until next time!", "off I go!",
+        ]
+    # Random Goodbyes responses
+    goodbye_replies = [
+        "Take care of yourself! If you have more questions, don't hesitate to reach out.",
+        "Stay well! Remember, I'm here if you need further medical advice.",
+        "Goodbye for now! Don't hesitate to return if you need more information in the future.",
+        "Wishing you good health ahead! Feel free to come back if you have more concerns.",
+        "Farewell! If you have more symptoms or questions, don't hesitate to consult again.",
+        "Take care and stay informed about your health. Feel free to chat anytime.",
+        "Bye for now! Remember, your well-being is a priority. Don't hesitate to ask if needed.",
+        "Have a great day ahead! If you need medical guidance later on, I'll be here.",
+        "Stay well and take it easy! Reach out if you need more medical insights.",
+        "Until next time! Prioritize your health and reach out if you need assistance.",
+        "Goodbye! Your health matters. Feel free to return if you have more health-related queries.",
+        "Stay healthy and stay curious about your health! If you need more info, just ask.",
+        "Wishing you wellness on your journey! If you have more questions, I'm here to help.",
+        "Take care and remember, your health is important. Don't hesitate to reach out if needed.",
+        "Goodbye for now! Stay informed and feel free to consult if you require medical advice.",
+        "Stay well and stay proactive about your health! If you have more queries, feel free to ask.",
+        "Farewell! Remember, I'm here whenever you need reliable medical information.",
+        "Bye for now! Stay vigilant about your health and don't hesitate to return if necessary.",
+        "Take care and keep your well-being a priority! Reach out if you have more health questions.",
+        "Wishing you good health ahead! Don't hesitate to chat if you need medical insights.",
+        "Goodbye! Stay well and remember, I'm here to assist you with medical queries.",
+    ]
+
+    # Create couple of if-else statements to capture/mimick peoples's Interaction
+  
+    if message.lower() in greetings:
+      bot_message= random.choice(responses)
+    elif message.lower() in goodbyes:
+      bot_message= random.choice(goodbye_replies)
+    else: 
+        inputs = preprocess(message)
+        bot_message = f"Based on your symptoms, I believe you may have {get_prediction(inputs)}."
+   else:
         # define the model and tokenizer.
         # model = PhiForCausalLM.from_pretrained(base_model)
         model = AutoModelForCausalLM.from_pretrained(base_model)