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

chatbot = pipeline(task="conversational", model="microsoft/DialoGPT-medium")

def Convo():
    if convo := st.chat_input("Enter your message")
    
    conversation = Conversation(convo)



    # candidate_labels = ["HELP", "PROBLEM SOLVE", "GENERAL TALK"]
  
   
    ans = chatbot(conversation)
    # add_user_input = st.button("Add User Input")

    # conversation.add_user_input("{}".format(convo))
    # conversation = chatbot(conversation)
    with st.chat_message("assistant"):
        ans.generated_responses[-1]
    # st.write(ans)


# convo()