first commit

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+import numpy as np
+from playform import playform,playform_load 
+
+
+@playform
+def sepia(input_img,url_params):
+    #print('inside fn',url_params)
+    sepia_filter = np.array([
+        [0.393, 0.769, 0.189], 
+        [0.349, 0.686, 0.168], 
+        [0.272, 0.534, 0.131]
+    ])
+    sepia_img = input_img.dot(sepia_filter.T)
+    sepia_img /= sepia_img.max()
+    return sepia_img,url_params
+
+with gr.Blocks() as demo:
+    
+    #code block needed to get url parameters
+    url_params = gr.JSON({}, visible=False, label="URL Params")
+    demo.load(lambda x: x,inputs = [url_params],outputs=[url_params],_js=playform_load)
+    #end code block needed to get url parameters
+     
+
+    with gr.Row():
+        with gr.Column():
+            img_input = gr.Image(shape=(200, 200))
+        with gr.Column():
+            img_output = gr.Image(shape=(200, 200))
+    
+    btn = gr.Button("submit")
+    btn.click(fn=sepia, inputs=[img_input, url_params], outputs=[img_output,url_params])
+    #btn.click(fn=sepia, inputs=img_input, outputs=img_output)
+
+
+
+if __name__=='__main__':
+    demo.launch(debug=True)

playform.py

@@ -0,0 +1,111 @@
+# this is to test actual playform
+
+import functools
+import requests
+
+import io
+import numpy
+from PIL import Image
+
+
+api_url_base = "http://127.0.0.1:7000/api/app/projects/"
+api_url_tail = "/generate/"
+        
+playform_load = """
+    function(url_params) {
+        console.log(url_params);
+        const params = new URLSearchParams(window.location.search);
+        url_params = Object.fromEntries(params);
+        return [url_params];
+        }
+    """
+
+
+def get_session_param(param):
+    if isinstance(param,list):
+        api_url=api_url_base+str(param[0]["id"])+api_url_tail
+        token = param[0]["token"]
+    else:
+        api_url = ""
+        token = ""
+        
+    return api_url, token
+
+def prep_image(image):
+    # takes a numpy image and prepare it to be sent
+    if image.dtype == numpy.uint8:
+        img = Image.fromarray(image)
+    else:
+        img = Image.fromarray((image*255).astype(numpy.uint8)) 
+    return img
+
+
+def send_image(img,api_url,token):
+    # takes a PIL image and send it
+
+    output=io.BytesIO()
+    img.save(output, format='JPEG')
+    f=output.getvalue()
+
+    files = [('image',('image.jpg',f,'image/jpeg'))]
+
+    # Set the headers for the request
+    headers = {'Authorization': f'Token {token}'}
+
+    payload = {}
+
+    print(api_url,token)
+
+    try:
+        response = requests.post(api_url,data=payload, headers=headers,files = files)
+        print("request sent")
+        print(response.text)
+    except:
+        print("request failed")
+    
+    return
+ 
+def playform(func):
+    @functools.wraps(func)
+    def wrapper_playform(*args, **kwargs):
+        
+        ret_value = func(*args, **kwargs)
+
+        #print(f'return type {type(ret_value)}')
+
+        value = ret_value[0]
+ 
+        # print(f'input type: {type(args)}')
+        # param= args[-1]
+        # print(f'param: , {param}')
+        
+        
+        api_url, token = get_session_param(args[-1])
+        # print(api_url)
+        # print(token)
+
+        # modified code to deal with multiple output
+        if isinstance(value,list):
+            print(f'number of outputs: {len(value)}')
+            for k in value:
+                if isinstance(k,numpy.ndarray):
+                    # print(f'output: ')
+                    # print(f'output type {type(k)}')
+                    # print(f'output dim {k.shape}')
+                    # print(f'data type {k.dtype}')
+                    # print(f'max value {k.max()} min value {k.min()}'  )
+
+                    img = prep_image(k)
+                    send_image(img,api_url,token)
+
+        elif isinstance(value,numpy.ndarray ):
+            # print('numpy array')
+            img = prep_image(value)
+            send_image(img,api_url,token)
+        else:
+            print('unpredicted return type')
+
+        
+    
+        return ret_value
+    return wrapper_playform