Update app.py

app.py

@@ -265,11 +265,17 @@ def process_image(input_img, w=0.7):
 
         face_helper.get_inverse_affine(None)
         final_img = face_helper.paste_faces_to_input_image(upsample_img=bg_img, draw_box=False)
+
+        # Chuyển BGR (OpenCV) sang RGB để xử lý với PIL và hiển thị trên UI
+        final_img_rgb = cv2.cvtColor(final_img, cv2.COLOR_BGR2RGB)
+        final_img_pil = Image.fromarray(final_img_rgb)
         
-        final_img_pil = Image.fromarray(cv2.cvtColor(final_img, cv2.COLOR_BGR2RGB))
+        # Thực hiện Enhance (nếu cần)
         final_img_pil = ImageEnhance.Color(final_img_pil).enhance(1.0) 
         final_img_pil = ImageEnhance.Contrast(final_img_pil).enhance(1.0)
-        res_hybrid = np.array(final_img)
+        
+        # Chuyển về mảng numpy để Gradio hiển thị đúng màu
+        res_hybrid = np.array(final_img_pil)
         
     except Exception as e:
         print(f"CodeFormer Error/No Face: {e}")