Spaces:
Sleeping
Sleeping
File size: 3,363 Bytes
172982e 8b369c5 172982e 3825673 172982e 8b369c5 172982e 8b369c5 172982e 8b369c5 172982e 8b369c5 172982e 8b369c5 172982e 8b369c5 172982e ce40fe3 172982e 8b369c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import gradio as gr
import os
import time
import google.generativeai as palm
palm.configure(api_key=os.environ.get("palm_key"))
defaults = {
'model': 'models/chat-bison-001',
'temperature': 0.3,
'candidate_count': 1,
'top_k': 40,
'top_p': 0.95,
}
context = "You are an order bot with only 3 types of pizzas: Margherita, Cheesy-Bacon and Vegetarian. Prices are small (9$), medium (11$) and large (13$). You can offer 2 types of drinks coke and mountain dew."
examples = [
[
"Hi, I want to order pizzas.",
"Hello, we have 3 types of pizzas: Margherita, Cheesy-Bacon and Vegetarian. Prices are small (9$), medium (11$) and large (13$)."
],
[
"I want a large margherita and a medium cheesy-bacon.",
"Do you want any drinks? We have coke (3$) or Mountain dew (5$)."
],
[
"Yes 2 cokes.",
"Total will be 19$. Please enter your address and phone number"
],
[
"Paris eiffel tower, 0653145863",
"Perfect ! Do you need anything else?"
],
[
"No thanks.",
"Have a nice day!\nHere is the sum up: \n{\npizza_type_1: \"margherita\",\npizza_type_2: \"cheesy-bacon\",\npizza_size_1: \"L\",\npizza_size_2: \"M\",\ndrink_1: \"coke\",\ndrink_2: \"coke\",\ntotal_price: \"19\",\naddress: \"Paris, Eiffel Tower\",\nphone_number: \"0653145863\"\n}\nDo you want to change anything?"
],
[
"Yes, i'd like to change margherita size to M.",
"Have a nice day!\nHere is the sum up: \n{\npizza_type_1: \"margherita\",\npizza_type_2: \"cheesy-bacon\",\npizza_size_1: \"M\",\npizza_size_2: \"M\",\ndrink_1: \"coke\",\ndrink_2: \"coke\",\ntotal_price: \"19\",\naddress: \"Paris, Eiffel Tower\",\nphone_number: \"0653145863\"\n}\nDo you want to change anything?"
],
[
"No thanks",
"Have a nice day!"
]
]
history = [['']]
with gr.Blocks(theme=gr.themes.Soft()) as demo:
chatbot = gr.Chatbot()
msg = gr.Textbox()
btn = gr.Button("Submit", variant="primary")
clear = gr.Button("Clear")
def user(user_message, history):
history.append([user_message, None])
return gr.update(value=""), history
def bot(history):
try:
bot_message = palm.chat(
context=context,
examples=examples,
messages=[h[0] for h in history]
)
history[-1][1] = ""
for character in bot_message.last:
history[-1][1] += character
time.sleep(0.005)
except Exception as e:
# Handle the exception here
print("Error occurred:", str(e))
# You can customize the error handling as per your requirements
# For example, return an error message to the user
history[-1][1] = "Incorrect input please retry with a longer sentence in english"
return history
response = msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
bot, chatbot, chatbot
)
response = btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
bot, chatbot, chatbot
)
response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)
clear.click(lambda: None, None, chatbot, queue=False)
demo.queue()
demo.launch()
|