import streamlit as st
import os 
from transformers import pipeline, Conversation

convo = pipeline("conversational")



def chatbot():
    st.title("ChatBot")
    if message := st.chat_input("Enter your message"):
        umsg = Conversation(message)
        answer = convo(umsg)
        with st.chat_message("assistant"):
            st.write(answer)
            
            
            
chatbot()