import streamlit as st
from bardapi import Bard
import os

uri = os.environ["BARD_API_KEY"]
bardkey = Bard(token = uri)

def Chatbot():
    st.title("Chatbot")
    if query := st.chat_input("Enter your message"):
        umsg = bard.get_answer(query)

    st.chat_message("user")
    st.write(["content"][0])

Chatbot()