Update app.py

app.py

@@ -40,13 +40,31 @@ def lexica(prompt, limit_size = 128, ratio_size = 256 + 128):
     lex = list(map(lambda x: min_dim_to_size(x, ratio_size)[1], lex))
     return lex
 
+def enterpix(prompt, limit_size = 100, ratio_size = 256 + 128, use_key = "bigThumbnailUrl"):
+    resp = requests.post(
+    url = "https://www.enterpix.app/enterpix/v1/image/prompt-search",
+    data= {
+        "length": limit_size,
+        "platform": "stable-diffusion,midjourney",
+        "prompt": prompt,
+        "start": 0
+        }
+    )
+    resp = resp.json()
+    resp = list(map(lambda x: x[use_key], resp["images"]))
+    resp = list(map(lambda ele: Image.open(
+                    requests.get(ele, stream = True).raw
+                    ), resp))
+    resp = list(map(lambda x: min_dim_to_size(x, ratio_size)[1], resp))
+    return resp
+
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             with gr.Row():
                 #inputs = gr.Textbox(label = 'Enter prompt to search Lexica.art')
                 inputs = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=20, min_width = 256,
-                placeholder="Enter prompt to search Lexica.art", elem_id="prompt")
+                placeholder="Enter prompt to search", elem_id="prompt")
                 #gr.Slider(label='Number of images ', minimum = 4, maximum = 20, step = 1, value = 4)]
                 text_button = gr.Button("Retrieve Images", elem_id="run_button")
             i = gr.Image(elem_id="result-image")
@@ -72,6 +90,9 @@ with gr.Blocks() as demo:
                 )
         #### gr.Textbox().submit().success()
 
-        text_button.click(lexica, inputs=inputs, outputs=outputs)
+        ### lexica
+        #text_button.click(lexica, inputs=inputs, outputs=outputs)
+        ### enterpix
+        text_button.click(enterpix, inputs=inputs, outputs=outputs)
 
 demo.launch("0.0.0.0")