Update app.py

app.py

@@ -41,7 +41,15 @@ def apply_edge_enhance(image):
     return np.clip(enhanced_edge_image, 0, 255).astype(np.uint8)
 
 def brightness_control(image, value):
-    return cv2.convertScaleAbs(image, alpha=1, beta=value)
+    img_float = np.float32(image)
+    img_float += value
+    
+    img_float = np.clip(img_float, 0, 255)
+    
+    brightened_image = np.uint8(img_float)
+    
+    return brightened_image
+
 
 def final(option1,option2, frame):
     option1_map = {
@@ -63,9 +71,10 @@ def final(option1,option2, frame):
     option1 = option1_map.get(option1, 1)
     option2 = option2_map.get(option2, 1)
 
-    if option1 == 1:
+    if option1 == 1:  # Apply effects to the person
         person = segmen.removeBG(frame, (0, 0, 0), cutThreshold=0.8)
         background = cv2.subtract(frame, person)
+        
         if option2 == 1:
             person = cv2.GaussianBlur(person, (15, 15), 0)
         elif option2 == 2:
@@ -77,11 +86,13 @@ def final(option1,option2, frame):
         elif option2 == 5:
             person = apply_edge_enhance(person)
         elif option2 == 'i':
-            person = brightness_control(person,10)
-            background = brightness_control(background,-10)
+            # person = brightness_control(person, 30)
+            background = brightness_control(background, -30)
         elif option2 == 'd':
-            background = brightness_control(background, -10)
+            person = brightness_control(person, -30)
+        
         frame = cv2.add(person, background)
+
     elif option1 == 2:
         person = segmen.removeBG(frame, (0, 0, 0), cutThreshold=0.8)
         background = cv2.subtract(frame, person)
@@ -96,10 +107,10 @@ def final(option1,option2, frame):
         elif option2 == 5:
             background = apply_edge_enhance(background)
         elif option2 == 'i':
-            person = brightness_control(person,-30)
-            background = brightness_control(background, 30)
+            person = brightness_control(person, -30)
+            # background = brightness_control(background, 30)
         elif option2 == 'd':
-            background = brightness_control(background, -10)
+            background = brightness_control(background, -30)
         frame = cv2.add(person, background)
     elif option1 == 3:
         if option2 == 1:
@@ -113,9 +124,9 @@ def final(option1,option2, frame):
         elif option2 == 5:
             frame = apply_edge_enhance(frame)
         elif option2 == 'i':
-            frame = brightness_control(frame, 30)
+            frame = brightness_control(frame, 50)
         elif option2 == 'd':
-            frame = brightness_control(frame, -30)
+            frame = brightness_control(frame, -50)
     return frame
 
 def process_image(img, option1, option2):