add glass shape recommendation

app.py

@@ -38,6 +38,15 @@ def determine_face_shape(landmarks):
     else:
         return "Square"
 
+# Recommend glass shape based on face shape
+def recommend_glass_shape(face_shape):
+    if face_shape == "Round":
+        return "Square"
+    elif face_shape == "Oval":
+        return "Round"
+    else:
+        return "Square"
+
 directory_path = 'glasses'
 total_glass_num = count_files_in_directory(directory_path)
 
@@ -62,16 +71,12 @@ def process_frame(frame):
     results = face_mesh.process(frame_rgb)
 
     face_shape = "Unknown"
+    glass_shape = "Unknown"
     if faces is not None and results.multi_face_landmarks:
         for face in faces:
             x, y, w, h = face[:4].astype(int)
             face_landmarks = face[4:14].reshape(5, 2).astype(int)
 
-            # Determine face shape
-            # face_shape = determine_face_shape(landmarks)
-
-            # Adjust the overlay size and position
-            # nose_x, nose_y = face_landmarks[2].astype(int)
             left_eye_x, left_eye_y = face_landmarks[0].astype(int)
             right_eye_x, right_eye_y = face_landmarks[1].astype(int)
 
@@ -102,10 +107,10 @@ def process_frame(frame):
 
             for face_landmarks_mp in results.multi_face_landmarks:
                 landmarks = np.array([(lm.x * frame.shape[1], lm.y * frame.shape[0]) for lm in face_landmarks_mp.landmark])
-                # Determine face shape
                 face_shape = determine_face_shape(landmarks)
+                glass_shape = recommend_glass_shape(face_shape)
 
-    return frame, face_shape
+    return frame, face_shape, glass_shape
 
 # Transform function
 def transform_cv2(frame, transform):
@@ -195,9 +200,9 @@ def save_frame(frame):
 
 # Gradio webcam input
 def webcam_input(frame, transform):
-    frame, face_shape = process_frame(frame)
+    frame, face_shape, glass_shape = process_frame(frame)
     frame = transform_cv2(frame, transform)
-    return frame, face_shape
+    return frame, face_shape, glass_shape
 
 # Gradio Interface
 with gr.Blocks() as demo:
@@ -207,10 +212,11 @@ with gr.Blocks() as demo:
                                     value="none", label="Transformation")
             input_img = gr.Image(sources=["webcam"], type="numpy", streaming=True)
             face_shape_output = gr.Textbox(label="Detected Face Shape")
+            glass_shape_output = gr.Textbox(label="Recommended Glass Shape")
             next_button = gr.Button("Next Glasses")
             save_button = gr.Button("Save as a Picture")
 
-    input_img.stream(webcam_input, [input_img, transform], [input_img, face_shape_output], time_limit=30, stream_every=0.1)
+    input_img.stream(webcam_input, [input_img, transform], [input_img, face_shape_output, glass_shape_output], time_limit=30, stream_every=0.1)
     with gr.Row():
         next_button.click(change_glasses, [], [])
     with gr.Row():