Update Webpage.py

Webpage.py

@@ -42,10 +42,17 @@ def calculate_areas(prompts, brand_num, pictorial_num, text_num):
     ad_size += np.abs((x1-x2)*(y1-y2))
     ad_image = image_entire[int(y1):int(y2), int(x1):int(x2), :]
     left_margin = x1; right_margin = w-x2
-    if left_margin >= right_margin:
-        context_image = image_entire[:, :int(x1), :]
+    if left_margin <=100 and right_margin <= 100:
+        upper_margin = y1; lower_margin = h-y2
+        if upper_margin >= lower_margin:
+            context_image = image_entire[:int(y1), :, :]
+        else:
+            context_image = image_entire[int(y2):, :, :]
     else:
-        context_image = image_entire[:, int(x2):, :]
+        if left_margin >= right_margin:
+            context_image = image_entire[:, :int(x1), :]
+        else:
+            context_image = image_entire[:, int(x2):, :]
 
     whole_size = 0
     whole_size += w*h
@@ -113,8 +120,8 @@ demo = gr.Interface(
                         &nbsp;&nbsp;&nbsp;(c) Text element(s) (skip if N.A.)  
                         &nbsp;&nbsp;&nbsp;(d) The advertisement.  
                         NOTE: Each ad element can have more than 1 boxes."""),
-            gr.DownloadButton(label="Example Magazine Ad Image Download", value='Demo/Magazine_Ad_Example.jpg'),
-            gr.DownloadButton(label="Example Web Ad Image Download", value='Demo/Web_Ad_Example.jpg'),
+            gr.DownloadButton(label="Download Example Magazine Ad Image", value='Demo/Magazine_Ad_Example.jpg'),
+            gr.DownloadButton(label="Download Example Web Ad Image", value='Demo/Web_Ad_Example.jpg'),
             ImagePrompter(label="Upload Entire (Ad+Context) Image, and Draw Bounding Boxes", sources=['upload'], type="pil"),
             gr.Number(label="Number of brand bounding boxes drawn"),
             gr.Number(label="Number of pictorial bounding boxes drawn"),