CoViD19-Bot-Server/df_payload.py at master · OneFourthLabs/CoViD19-Bot-Server · GitHub

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
93
94
95
96
97
98
99
100
101
102
# Retrieve payloads for DialogFlow


def get_plot_payload(imageURL):
    reply = {
        "fulfillmentText": "",
        "fulfillmentMessages": [
            {
            }
        ],
        "payload": {
            "google":
            {
                "expectUserResponse": True,
                "richResponse":
                {
                    "items": [
                        {
                            "basicCard":
                            {
                                "title": "",
                                "formattedText": "",
                                "image":
                                {
                                    "url": imageURL,
                                },
                                "buttons": [
                                    {
                                        "title": "See large image",
                                        "openUrlAction":
                                        {
                                            "url": imageURL
                                        }
                                    },
                                ]
                            },
                        }
                    ],
                }
            }
        }
    }
    return reply


def get_card_payload(textTitle, textAnswer, imageURL="",
                     sourceURL="", sourceURLTitle="Link",
                     referenceURL="", suggestions=[]):
    reply = {
        "fulfillmentText": "",
        "fulfillmentMessages": [
            {
            }
        ],
        "source": "example.com",
        "payload": {
            "google":
            {
                "expectUserResponse": True,
                "richResponse":
                {
                    "items": [
                        {
                            "basicCard":
                            {
                                "title": textTitle,
                                "formattedText": textAnswer,
                                "image":
                                {
                                    "url": imageURL,
                                },
                                "buttons": [
                                    {
                                        "title": sourceURLTitle,
                                        "openUrlAction":
                                        {
                                            "url": sourceURL
                                        }
                                    },
                                    {
                                        "title": "Learn more",
                                        "openUrlAction":
                                        {
                                            "url": referenceURL
                                        }
                                    }
                                ],
                                "imageDisplayOptions": "CROPPED"
                            },
                        }
                    ],
                    "suggestions": suggestions
                }
            }
        }
    }
    if not referenceURL:
        # Temporary hack to remove the 2nd button
        # TODO: Handle it properly
        del reply["payload"]["google"]["richResponse"]["items"][0]["basicCard"]["buttons"][1]

    return reply