app.py

import streamlit as st
import google.generativeai as palm

palm.configure(api_key='AIzaSyCdkkVThvSiqOqk9QeuM9ug5Co9dEsENcE') 

models = [m for m in palm.list_models() if 'generateText' in m.supported_generation_methods]
model = models[0].name

st.title('Text Generation App')

prompt = st.text_input('Enter prompt:')

if st.button('Generate'):

  completion = palm.generate_text(
      model=model,
      prompt=prompt,
      temperature=0,
      max_output_tokens=800,
  )

  st.write(completion.result)