combine hues

app.py

@@ -12,6 +12,8 @@ def inference(img, template, angel):
     HSV_image = cv2.cvtColor(color_image, cv2.COLOR_BGR2HSV)
     selected_harmomic_scheme = HarmonicScheme(str(template), int(angel))
     new_HSV_image = selected_harmomic_scheme.hue_shifted(HSV_image, num_superpixels=-1)
+    # Convert HSV to BGR
+    result_image = cv2.cvtColor(new_HSV_image, cv2.COLOR_HSV2BGR)
 
     # Compute shifted histogram
     histo_1 = count_hue_histogram(HSV_image)
@@ -22,15 +24,14 @@ def inference(img, template, angel):
     fig_1_cv = get_img_from_fig(fig1)
     fig2 = plothis(histo_2, selected_harmomic_scheme, "Target Hue")
     fig_2_cv = get_img_from_fig(fig2)
-    # Convert HSV to BGR
-    result_image = cv2.cvtColor(new_HSV_image, cv2.COLOR_HSV2BGR)
 
-    cv2.imwrite('fig_1_cv.jpg', fig_1_cv)
-    cv2.imwrite('fig_2_cv.jpg', fig_2_cv)
+    hue_plots = np.concatenate((fig_1_cv, fig_2_cv), axis=1)
+
+    cv2.imwrite('hue.jpg', hue_plots)
     cv2.imwrite('result_image.jpg', result_image)
 
 
-    return ['result_image.jpg', 'fig_1_cv.jpg', 'fig_2_cv.jpg']
+    return ['result_image.jpg', 'hue.jpg']
 
 title = 'Color Harmonization'
 description = 'Compute Color Harmonization with Different Templates'
@@ -46,8 +47,7 @@ gr.Interface(
                         label="Template"),
      gr.inputs.Slider(0, 359, label="Angle")],
     [gr.outputs.Image(type='file', label='Color Harmonization of Output Image'),
-     gr.outputs.Image(type='file', label='Source Hue'),
-     gr.outputs.Image(type='file', label='Target Hue'),],
+     gr.outputs.Image(type='file', label='Hue'),
     title=title,
     description=description,
     article=article,