added server hook

1 files changed, 11 insertions, 0 deletions

diff --git a/pbserver.py b/pbserver.py
index a2fb1a7..25de07a 100755
--- a/pbserver.py
+++ b/pbserver.py
@@ -5,6 +5,7 @@ from pb.gradient import Gradient
 from pb.imgrid import Imgrid
 from pb.breaker import Breaker
 from pb.pattern import Pattern
+from pb.generate import Generate
 
 from pb.config import AWS_SECRET_ACCESS_KEY, AWS_ACCESS_KEY_ID, BUCKET_NAME, BIN_IDENTIFY
 import pb.lib.utils as utils
@@ -130,6 +131,16 @@ def imgrid():
         sys.stderr.write(str(e))
         return json.dumps({ 'error' : 'Request could not be processed' })
 
+@post('/im/api/generate')
+def generate():
+    try:
+        im = Generate(**(dict(request.forms)))
+        im.create();
+        return return_image(im)
+    except Exception as e:
+        sys.stderr.write(str(e))
+        return json.dumps({ 'error' : 'Request could not be processed' })
+
 @post('/im/api/imbreak')
 def breaker():
     try: