Update app.py

app.py

@@ -14,6 +14,8 @@ tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/T5_MineAI_Prototype")
 
 pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
 
+greetings=["Hello! My name is MineAI, A specially trained LLM here to assist you on your Mining Related Queries.","How may I help you?"]
+
 st.set_page_config(page_title='Sample Chatbot', layout='wide')
 
 if 'messages' not in st.session_state:
@@ -27,18 +29,29 @@ for message in st.session_state.messages:
         
 ## messages element format: {'role':'user', 'content':'<user prompt>'}
 
-if prompt:=st.chat_input("What is up!"):
+for gr in greetings:
+    with st.chat_message("assistant"):
+        st.markdown(gr)
+
+    st.session_state.messages.append({
+        'role':'assistant',
+        'content': gr})
+
+if prompt:=st.chat_input("Enter your query"):
     with st.chat_message("user"):
         st.markdown(prompt)
-        
+
     st.session_state.messages.append({
         'role':'user',
         'content': prompt})
 
-    out=pipe(prompt)
-    response=out[0]['generated_text']
+    if prompt in responses:
+        response=responses[prompt]
+    else:
+        out=pipe(prompt)
+        response=out[0]['generated_text']
     
-    response = f"Analysis: {response}"
+    # response = f"Analysis: {response}"
     
     with st.chat_message("assistant"):
         st.markdown(response)