copy_of_chatbot_using_palm2.py

# -*- coding: utf-8 -*-
"""Copy of Chatbot Using Palm2.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1SiwltrRkFf5EIhLLbj9GISjb8HmDJYw3
"""

!pip install -q google-generativeai
!pip install gradio

import google.generativeai as palm
import gradio as gr

palm.configure(api_key='AIzaSyC2lkfp2czLhE0sFZ9I8Rm6bRQyBMuKexs')

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

#def get_text_response(user_message,history):
  #  response = palm.chat(messages=user_message)
  #  return response.last

#demo = gr.ChatInterface(get_text_response, examples=["How are you doing?","What are your interests?","Which places do you like to visit?"])

def get_text_response(user_message,history):
  completion = palm.generate_text(
    model=model,
    prompt=user_message,
    temperature=0.25,
    # temperature=0 >> more deterministic results // temperature=1 >> more randomness
    max_output_tokens=100
    # maximum length of response
)
  return completion.result

demo = gr.ChatInterface(get_text_response, examples=["How are you doing?","What are your interests?","Which places do you like to visit?"])

if __name__ == "__main__":
  demo.launch()