# imports
from bardapi import Bard
import streamlit as st
import os

# Streamlit application
def main():
    # retrieve token
    token = 'ZAhLevytbo8gW2AD1Aln_9nZIUEHdcR1VUlhGreF81ycpNLsitgZcOWt80S5QFrVJjCA6Q.'
    
    # UI input
    question = st.text_input('Enter your question here')
    if question:
        # get answer content from Bard and display
        answer = Bard(token=token).get_answer(question)['content']
        st.write(f"Question: {question}\nAnswer: {answer}")
        # Draw a separating line
        st.markdown("---")


# main driver
if __name__ == '__main__':
    main()