import google.generativeai as genai
import streamlit as st 
import os 

st.set_page_config(layout="wide")

st.markdown("""
<style>
#MainMenu {visibility: hidden;}
  footer {visibility: hidden;}
  header {visibility: hidden;}
</style>
""", unsafe_allow_html=True)



# Set your API key
genai.configure(api_key = os.environ['PALM_KEY'])

# Select the PaLM 2 model
# model = 'models/text-bison-001'
model = genai.GenerativeModel('gemini-pro')


# Generate text
if prompt := st.chat_input("Ask your query..."):
    enprom = f"""Answer the below provided input in context to Bhagwad Geeta. Use the verses and chapters sentences as references to your answer with suggestions
    coming from Bhagwad Geeta. Your answer to below input should only be in context to Bhagwad geeta only.\nInput= {prompt}"""
    completion = model.generate_content(enprom)

# response = palm.chat(messages=["Hello."])
# print(response.last) #  'Hello! What can I help you with?'
# response.reply("Can you tell me a joke?")

# Print the generated text
    with st.chat_message("Assistant"):
        st.write(completion.text)