Input Sanitization

app.py

@@ -6,6 +6,7 @@ import mediapipe as mp
 import time
 import gradio as gr
 import glob
+import os
 
 width_, height_ = 144, 96
 
@@ -16,8 +17,20 @@ output_frames = []
 
 
 def is_hex(hexq):
-    if hexq[0] == '#' and len(hexq) == 7 and hexq[1:7].isalnum():
-        return True
+    valid = ['0','1','2','3','4','5','6',
+             '7','8','9','A','a','B','b',
+             'C','c','D','d','E','e','F',
+             'f']
+    hexq = str(hexq)
+    if len(hexq) == 7:
+        if hexq[0] == '#':
+            for h in hexq[1:]:
+                if h in valid:
+                    return True
+                else:
+                    return False
+        else:
+            return False
     else:
         return False
 
@@ -80,7 +93,7 @@ def clear(landmarks):  # brute force finger orientation checking
         return False
 
 
-def show(video, dominant_hand, hex_color): # main
+def show(video, dominant_hand, hex_color='#FFFFFF'): # main
     cam = cv2.VideoCapture(video)  # get the video file from path
     width = cam.get(cv2.CAP_PROP_FRAME_WIDTH)
     height = cam.get(cv2.CAP_PROP_FRAME_HEIGHT)
@@ -190,6 +203,7 @@ def show(video, dominant_hand, hex_color): # main
     for filename in glob.glob('*.png'):
         imggg = cv2.imread(filename)
         img_array.append(imggg)
+        os.remove(filename)
 
     video_output = cv2.VideoWriter('any.webm', cv2.VideoWriter_fourcc(*'VP80'), 30, (640, 480))
 
@@ -199,12 +213,15 @@ def show(video, dominant_hand, hex_color): # main
 
     return 'any.webm'
 
+
 title = 'Air Draw'
-desc = 'A mediapipe hands wrapper for drawing in the air. Position your fingers in a "drawing" form to draw, and open up your hand to lift the "pen" from the "paper". Use a "thumbs up" gesture to clear the drawing paper.'
+desc = 'A mediapipe hands wrapper for drawing in the air. Draw holding an invisible pen, ' \
+       'and open up your hand to lift the "pen" from the "paper". Use a "thumbs up" gesture to clear the drawing ' \
+       'paper. '
 iface = gr.Interface(
     fn=show,
     inputs=[
-        gr.inputs.Video(source="webcam", label="Record yourself drawing in the air!"),
+        gr.inputs.Video(source="webcam", label="Record yourself drawing holding an invisible pen!"),
         gr.inputs.Radio(['Right', 'Left'], label="Dominant Hand"),
         gr.inputs.Textbox(placeholder="#355C7D", label="Hex Color")
     ],