Update app.py

app.py

@@ -4,19 +4,40 @@ import cv2
 
 def filter_img(input_img,filter_type):
     img = input_img
+    filtered_img = np.zeros(img.shape, dtype=int)
+    
     if filter_type == 'vintage':
         sepia_filter = np.array(
             [[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]]
         )
         filtered_img = img.dot(sepia_filter.T)
         filtered_img /= filtered_img.max()
-        return filtered_img
-        
-        
     elif filter_type == 'gray-sharp':
         img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
-    filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
-    
+        filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
+    elif filter_type == 'sharp':
+        filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
+    elif filter_type == 'gray':
+        filtered_img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
+    elif filter_type == 'B&W':
+        filtered_img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
+        _,filtered_img = cv2.threshold(filtered_img,127,255,cv2.THRESH_BINARY)
+    elif filter_type == 'Magenta':
+        filtered_img[:,:,0] = img[:,:,0]
+        filtered_img[:,:,2] = img[:,:,2]
+    elif filter_type == 'Cyan':
+        filtered_img[:,:,1] = img[:,:,1]
+        filtered_img[:,:,2] = img[:,:,2]
+    elif filter_type == 'Yellow':
+        filtered_img[:,:,0] = img[:,:,0]
+        filtered_img[:,:,1] = img[:,:,1]
+    elif filter_type == 'Blue':
+        filtered_img[:,:,2] = img[:,:,2]
+    elif filter_type == 'Green':
+        filtered_img[:,:,1] = img[:,:,1]
+    elif filter_type == 'Red':
+        filtered_img[:,:,0] = img[:,:,0]
+
     return filtered_img
 
 demo = gr.Interface(filter_img,