PrototipoTUR / app.py
dtrejopizzo's picture
Update app.py
146f324
import openai
import gradio as gr
openai.api_key =''
title = "Travel Buddy"
examples = [
["Armar un itinerario para viajar a Madrid, por 7 días, con familia, tener en cuenta que quiero hacer puntos turísticos y museos."],
["Armar un plan de viaje para 10 días en Italia."],
["Conseguir el vuelo más barato desde Buenos Aires a Nueva York, para un viaje de 7 días en marzo de 2023."],
]
def sentence_builder(donde,dias,quien,actividades):
return f"""Armar itineratio para viajar a {donde}, por {dias} dias, con {quien}, tener en cuenta que quiero hacer {" y ".join(actividades)}."""
def openai_chat(prompt):
completions = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt+"The following is the prompt from teacher working in canvas infrastructure",
max_tokens=1024,
n=1,
temperature=0.5,
frequency_penalty=0,
presence_penalty=0.6,
stop=[" Human:", " AI:"]
)
message = completions.choices[0].text
return message.strip()
def chatbot(input, history=[]):
output = openai_chat(input)
history.append((input, output))
return history, history
gr.Interface(fn = chatbot,
inputs = ["text",'state'],
outputs = ["chatbot",'state'],
allow_flagging="manual").launch()