Spaces:

ppaihack
/

PoCInnovation

Sleeping

pierrelissope commited on Sep 28, 2024

Commit

5e0990b

1 Parent(s): 94f7100

feat: c'est carré on a fini

Files changed (4) hide show

front/dist/index.html CHANGED Viewed

@@ -5,8 +5,8 @@
     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Vite + React + TS</title>
-    <script type="module" crossorigin src="/assets/index-Cqst1UVv.js"></script>
-    <link rel="stylesheet" crossorigin href="/assets/index-DBF5kfs0.css">
   </head>
   <body>
     <div id="root"></div>

     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Vite + React + TS</title>
+    <script type="module" crossorigin src="/assets/index-D1zI8xqC.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-FOeRIyQ0.css">
   </head>
   <body>
     <div id="root"></div>

front/src/VerificationScene.tsx CHANGED Viewed

@@ -5,6 +5,8 @@ import VideoWrapper from "@/components/VideoWrapper.tsx";
 import ResultContainer from "@/components/ResultContainer.tsx";
 import DarkVideoWrapper from "@/components/DarkVideoWrapper.tsx";
 enum StepType {
   takeId,
   verifyId,
@@ -39,7 +41,7 @@ export default function VerificationScene() {
   const sendImageToServer = async (
     idCard: string,
     profileImage: string,
-    setLoading: (n: boolean) => void,
   ) => {
     try {
       const response = await fetch("http://0.0.0.0:7860/uploadids", {
@@ -56,10 +58,10 @@ export default function VerificationScene() {
       const data = await response.json();
       console.log("Image envoyée avec succès:", data);
-      setLoading(false);
     } catch (error) {
       console.error("Erreur:", error);
-      setLoading(false);
     }
   };

 import ResultContainer from "@/components/ResultContainer.tsx";
 import DarkVideoWrapper from "@/components/DarkVideoWrapper.tsx";
+export type fetchResult = "Valid" | "Not Valid" | "Fetching" | "Error";
 enum StepType {
   takeId,
   verifyId,
   const sendImageToServer = async (
     idCard: string,
     profileImage: string,
+    setLoading: (n: fetchResult) => void,
   ) => {
     try {
       const response = await fetch("http://0.0.0.0:7860/uploadids", {
       const data = await response.json();
       console.log("Image envoyée avec succès:", data);
+      setLoading(data.message === "Valid" ? "Valid" : "Not Valid");
     } catch (error) {
       console.error("Erreur:", error);
+      setLoading("Error");
     }
   };

front/src/components/ResultContainer.tsx CHANGED Viewed

@@ -1,4 +1,5 @@
 import { useEffect, useState } from "react";
 export default function ResultContainer({
   sendImageToServer,
@@ -8,22 +9,26 @@ export default function ResultContainer({
   sendImageToServer: (
     pic: string,
     pic2: string,
-    setLoading: (n: boolean) => void,
   ) => void;
   idCardPicture: string;
   profileImage: string;
 }) {
-  const [loading, setLoading] = useState<boolean>(true);
   useEffect(() => {
     sendImageToServer(idCardPicture ?? "", profileImage ?? "", setLoading);
   }, []);
   return (
     <div className="flex w-full h-full items-center justify-center">
-      {loading ? (
         <h1 className="text-2xl font-bold text-orange-600">Fetching....</h1>
       ) : (
-        <h1 className="text-2xl font-bold text-green-800">Process Completed</h1>
       )}
     </div>
   );

 import { useEffect, useState } from "react";
+import { fetchResult } from "@/VerificationScene.tsx";
 export default function ResultContainer({
   sendImageToServer,
   sendImageToServer: (
     pic: string,
     pic2: string,
+    setLoading: (n: fetchResult) => void,
   ) => void;
   idCardPicture: string;
   profileImage: string;
 }) {
+  const [loading, setLoading] = useState<fetchResult>("Fetching");
   useEffect(() => {
     sendImageToServer(idCardPicture ?? "", profileImage ?? "", setLoading);
   }, []);
   return (
     <div className="flex w-full h-full items-center justify-center">
+      {loading === "Fetching" ? (
         <h1 className="text-2xl font-bold text-orange-600">Fetching....</h1>
+      ) : loading === "Valid" ? (
+        <h1 className="text-2xl font-bold text-green-800">Valid</h1>
+      ) : loading === "Not Valid" ? (
+        <h1 className="text-2xl font-bold text-green-800">Not Valid</h1>
       ) : (
+        <h1 className="text-2xl text-red-700">Error</h1>
       )}
     </div>
   );

main.py CHANGED Viewed

@@ -7,6 +7,7 @@ from pydantic import BaseModel
 import time
 from facenet_pytorch import InceptionResnetV1, MTCNN
 import warnings
 import face_compare
 warnings.filterwarnings('ignore', category=FutureWarning, module='facenet_pytorch')
@@ -53,15 +54,26 @@ async def upload_pdf(data: ImageData):
 @router.post("/uploadids")
 async def upload_ids(data: ImagesData):
     header, encoded1 = data.idCard.split(',', 1)
     binary_data1 = base64.b64decode(encoded1)
     header, encoded2 = data.profileImage.split(',', 1)
     binary_data2 = base64.b64decode(encoded2)
-    output = face_compare.compare_faces(binary_data1, binary_data2)
     if output > 0.6:
-        return {"message": "Les images correspondent"}
     else:
-        return {"message": "Les images ne correspondent pas"}
 app.include_router(router)

 import time
 from facenet_pytorch import InceptionResnetV1, MTCNN
 import warnings
+import os
 import face_compare
 warnings.filterwarnings('ignore', category=FutureWarning, module='facenet_pytorch')
 @router.post("/uploadids")
 async def upload_ids(data: ImagesData):
     header, encoded1 = data.idCard.split(',', 1)
     binary_data1 = base64.b64decode(encoded1)
     header, encoded2 = data.profileImage.split(',', 1)
     binary_data2 = base64.b64decode(encoded2)
+    with open("id_card_image.png", "wb") as id_card_file:
+        id_card_file.write(binary_data1)  # Save as PNG (or use correct extension based on header)
+    with open("profile_image.png", "wb") as profile_file:
+        profile_file.write(binary_data2)  # Save as PNG (or use correct extension based on header)
+    id_card_abs_path = os.path.abspath("id_card_image.png")
+    profile_image_abs_path = os.path.abspath("profile_image.png")
+    output = face_compare.compare_faces(id_card_abs_path, profile_image_abs_path)
     if output > 0.6:
+        return {"message": "Valid"}
     else:
+        return {"message": "Not Valid"}
 app.include_router(router)