Update app.py

app.py

@@ -6,7 +6,15 @@ def face_similarity(img1, img2):
     similarity = result["distance"]  # 낮을수록 더 유사
     verified = result["verified"]
     
-    return f"유사도: {similarity}, 매칭 결과: {verified}"
+    # 유사도에 따른 메시지 조건 분기
+    if similarity >= 0.5:
+        message = "두 분은 천생연분!"
+    elif 0.3 <= similarity < 0.5:
+        message = "두 분은 내꺼인듯 내꺼아닌 내꺼같은...?"
+    else:
+        message = "오히려 좋아! 정반대의 사람이 끌릴지도 몰라요"
+    
+    return f"유사도: {similarity}, 매칭 결과: {verified}\n{message}"
 
 iface = gr.Interface(fn=face_similarity, 
                      inputs=[gr.inputs.Image(shape=(224, 224)), gr.inputs.Image(shape=(224, 224))], 
@@ -16,6 +24,26 @@ iface = gr.Interface(fn=face_similarity,
 
 iface.launch()
 
+
+
+# import gradio as gr
+# from deepface import DeepFace
+
+# def face_similarity(img1, img2):
+#     result = DeepFace.verify(img1, img2)
+#     similarity = result["distance"]  # 낮을수록 더 유사
+#     verified = result["verified"]
+    
+#     return f"유사도: {similarity}, 매칭 결과: {verified}"
+
+# iface = gr.Interface(fn=face_similarity, 
+#                      inputs=[gr.inputs.Image(shape=(224, 224)), gr.inputs.Image(shape=(224, 224))], 
+#                      outputs="text",
+#                      title="얼굴 유사도 측정",
+#                      description="두 얼굴 이미지를 업로드해서 유사도를 측정합니다.")
+
+# iface.launch()
+
 # import gradio as gr
 # from deepface import DeepFace
 # import pandas as pd