Update app.py

app.py

@@ -7,6 +7,16 @@ from PIL import Image
 import imagehash
 import requests
 
+def min_dim_to_size(img, size = 512):
+    h, w = img.size
+    ratio = size / max(h, w)
+    h, w = map(lambda x: int(x * ratio), [h, w])
+    return ( ratio ,img.resize((w, h)) )
+    #return ( ratio ,img.resize((h, w)) )
+
+#ratio_size = 512
+#ratio, img_rs = min_dim_to_size(img, ratio_size)
+
 def image_click(images, evt: gr.SelectData):
     img_selected = images[evt.index]
     return images[evt.index]['name']
@@ -14,30 +24,31 @@ def image_click(images, evt: gr.SelectData):
 def swap_gallery(im, images):
     #### name data is_file
     #print(images[0].keys())
+    if im is None:
+        return images
 
     im_hash = imagehash.average_hash(Image.fromarray(im))
     t2_list = sorted(images, key = lambda imm:
     imagehash.average_hash(Image.open(imm["name"])) - im_hash, reverse = False)
     return list(map(lambda x: x["name"], t2_list))
 
-
-def lexica(prompt, limit_size = 128, resize = (512, 512)):
+def lexica(prompt, limit_size = 128, ratio_size = 256):
     lex = Lexica(query=prompt).images()
     lex = lex[:limit_size]
     lex = list(map(lambda ele: Image.open(
                     requests.get(ele, stream = True).raw
                     ), lex))
-    lex = list(map(lambda x: x.resize(resize), lex))
+    lex = list(map(lambda x: min_dim_to_size(x, ratio_size)[1], lex))
     return lex
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Tabs():
             inputs = gr.Textbox(label = 'Enter prompt to search Lexica.art')
-            text_button = gr.Button("Retrieve Images")
             #gr.Slider(label='Number of images ', minimum = 4, maximum = 20, step = 1, value = 4)]
-        with gr.Tabs():
             i = gr.Image()
+            text_button = gr.Button("Retrieve Images")
+        with gr.Tabs():
             outputs = gr.Gallery(lable='Output gallery').style(grid=3,height=768,
                 allow_preview=False)
             #gr.Dataframe(label='prompts for corresponding images')]