dalle-gen-emoji - annna - Annna the nice friendly bot.
 (HTM) 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