File size: 1,326 Bytes
cdcdde4
59e85c1
146f324
e6bd1c2
7d8d573
 
 
 
 
 
 
 
6e0d033
3d3330b
e6bd1c2
cdcdde4
 
 
 
 
 
 
 
 
 
 
 
 
cbcd5da
 
 
 
cdcdde4
ab84625
10dd5a8
 
 
 
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
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()