import streamlit as st
from bardapi import Bard 
import os

st.set_page_config(layout="wide")

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


def Chatbot():
    st.title("Chatbot")
    if advice_input := st.chat_input("Measures for better education ecosystem in India. e.g: how we can education  wiil be more accessible, affordable & promote innovation. Comments shall be specific & factful.")
        bardans = bard.get_answer(advice_input)['content']
        anslist = bard.get_answer(f"Make a list of suggestions: \n{bardans} n\for the betterment education: \n{advice_input}\n\nThe list should be in three sections, section 1 for quality of education, section 2 for relevance of job oriented education & section 3 for pro & cons of linking Artifical intelligence with education system.")["content"]

        with st.chat_message("assistant"):
            st.write(f"What to be done to achieve better education:\n{bardans\nHow we can:\n{anslist")

Chatbot()