Update app.py

app.py

@@ -127,9 +127,9 @@ def classify_zip_and_analyze_color(zip_file):
                     faces = DeepFace.analyze(img_cv2, actions=["age", "gender", "emotion"], enforce_detection=False)
                     if isinstance(faces, list):  # multiple faces
                         for f in faces:
-                            face_info += f"Age: {f['age']}, Gender: {f['gender']}, Gender Confidence: {f['gender_confidence']*100:.2f}, Emotion: {f['dominant_emotion']}; "
+                            face_info += f"Age: {f['age']}, Gender: {f['gender']}, Emotion: {f['dominant_emotion']}; "
                     else:  # single face
-                        face_info = f"Age: {faces['age']}, Gender: {faces['gender']}, Gender Confidence: {faces['gender_confidence']*100:.2f}, Emotion: {faces['dominant_emotion']}"
+                        face_info = f"Age: {faces['age']}, Gender: {faces['gender']}, Emotion: {faces['dominant_emotion']}"
                 except Exception as e:
                     face_info = "No face detected"
 
@@ -181,45 +181,32 @@ def classify_zip_and_analyze_color(zip_file):
     plot2_img = Image.open(buf2)
 
     # ---------------------------
-    # Extract age and gender (confidence ≤ 80%)
+    # Extract age and gender
     # ---------------------------
     ages = []
-    gender_confidence = {"Man": 0, "Woman": 0}
+    genders = []
 
     for info in df["Face Info"]:
         if info != "No face detected":
             for face_str in info.split(";"):
                 face_str = face_str.strip()
                 if face_str:
-                    # Age
                     age_part = face_str.split(",")[0]
                     age = int(age_part.replace("Age:", "").strip())
                     ages.append(age)
                     
-                    # Gender and confidence
                     gender_part = face_str.split(",")[1]
                     gender = gender_part.replace("Gender:", "").strip()
-                    
-                    # Extract confidence
-                    conf = 1.0
-                    for part in face_str.split(","):
-                        if "Gender Confidence:" in part:
-                            conf = float(part.split("Gender Confidence:")[1].strip()) / 100  # convert % to 0-1
-
-                    # Only include if confidence ≤ 0.8
-                    if conf <= 0.8:
-                        if gender in gender_confidence:
-                            gender_confidence[gender] += conf
-                        else:
-                            gender_confidence[gender] = conf
+                    genders.append(gender)
 
     # ---------------------------
-    # Plot 3: Gender distribution (confidence ≤ 80%)
+    # Plot 3: Gender distribution
     # ---------------------------
     fig3, ax3 = plt.subplots()
-    ax3.bar(gender_confidence.keys(), gender_confidence.values(), color=["lightblue", "pink"])
-    ax3.set_title("Gender Distribution (Confidence ≤ 80%)")
-    ax3.set_ylabel("Sum of Confidence")
+    gender_counts = pd.Series(genders).value_counts()
+    ax3.bar(gender_counts.index, gender_counts.values, color=["lightblue", "pink"])
+    ax3.set_title("Gender Distribution")
+    ax3.set_ylabel("Count")
     buf3 = io.BytesIO()
     plt.savefig(buf3, format="png")
     plt.close(fig3)
@@ -253,7 +240,7 @@ demo = gr.Interface(
         gr.File(label="Download XLSX"),
         gr.Image(type="pil", label="Basic Color Frequency"),
         gr.Image(type="pil", label="Top Prediction Distribution"),
-        gr.Image(type="pil", label="Gender Distribution (≤80% Confidence)"),
+        gr.Image(type="pil", label="Gender Distribution"),
         gr.Image(type="pil", label="Age Distribution"),
     ],
     title="Image Classifier with Color & Face Analysis",
@@ -262,4 +249,3 @@ demo = gr.Interface(
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)
-