AI chatbot with Google Gemini Pro Model in Python Notebook

!pip install -q -U google-generativeai

import pathlib
import textwrap

import google.generativeai as genai

from IPython.display import display
from IPython.display import Markdown


def to_markdown(text):
  text = text.replace('•', '  *')
  return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))

import os

gemini_api_key = os.environ["GEMINI_API_KEY"]
print(gemini_api_key)
genai.configure(api_key=gemini_api_key)

#list of available models

for m in genai.list_models():
  if 'generateContent' in m.supported_generation_methods:
    print(m.name)

model = genai.GenerativeModel('gemini-pro')

%%time
prompt = input("Ask something ")
response = model.generate_content(prompt)

to_markdown(response.text)