chat.py

# from transformers import pipeline, Conversation
# # import streamlit_option_menu
# import streamlit as st

# def Chat():

#     query = st.chat_input("Enter your query")
#     convo = pipeline("conversational")
#     oracle = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
#     usrinput = Conversation(query)
#     chitchat = convo(usrinput)
#     ans = oracle(
#         query,
#         candidate_labels=["logout"])

#     if ans["scores"][0] > 0.85:
#         st.session_state["user"] = "visitor"
#         with st.chat_message("assistant"):
#             "You are now living in dream" 
#         st.experimental_rerun()
#     else:
#         with st.chat_message("assistant"):
#             chitchat