dalle-gen-emoji - annna - Annna the nice friendly bot.
HTML git clone git://bitreich.org/annna/ git://enlrupgkhuxnvlhsf6lc3fziv5h2hhfrinws65d7roiv6bfj7d652fid.onion/annna/
DIR Log
DIR Files
DIR Refs
DIR Tags
DIR README
---
dalle-gen-emoji (1344B)
---
1 #!/usr/bin/env python
2 # coding=utf-8
3 #
4 # Kiss me when you can.
5 # by annna
6 #
7
8 import os
9 import os.path
10 import sys
11 import getopt
12 import requests
13 import base64
14
15 def usage(app):
16 app = os.path.basename(app)
17 print("usage: %s [-h] [::]some-emoji[:key=value...][::]" % (app),
18 file=sys.stderr)
19 sys.exit(1)
20
21 def main(args):
22 try:
23 opts, largs = getopt.getopt(args[1:], "h")
24 except getopt.GetoptError as err:
25 print(str(err))
26 usage(args[0])
27
28 for o, a in opts:
29 if o == "-h":
30 usage(args[0])
31 else:
32 assert False, "unhandled option"
33
34 if len(largs) < 1:
35 usage(args[0])
36
37 emoji = " ".join(largs)
38 emoji = emoji.strip("::")
39 emoji = emoji.replace(":", " ").replace(",", " ").replace("=", " ").\
40 replace(".", " ").replace("/", " ")
41 emoji = emoji.lower()
42
43 emojifile = "%s.jpg" % (emoji.replace(" ", "-"))
44
45 filebase = "/br/ai/dalle/dalle-results/"
46 emojiuribase = "gophers://bitreich.org/I/emoji/"
47 imguri = "http://127.0.0.1:32553/dalle"
48 jsonreq = {"text": emoji, "num_images": 1}
49
50 filepath = "%s%s" % (filebase, emojifile)
51 if not os.path.exists(filepath):
52 try:
53 answer = requests.post(imguri, json=jsonreq)
54 except:
55 return 1
56
57 fd = open(filepath, "wb")
58 fd.write(base64.b64decode(answer.json()[0]))
59 fd.close()
60
61 print("%s%s" % (emojiuribase, emojifile))
62
63 return 0
64
65 if __name__ == "__main__":
66 sys.exit(main(sys.argv))
67