Merge branch 'fe/fix/upload' into 'fe/develop'

Refactor: 퍼센테이지 복구

See merge request s11-s-project/S11P21S002!301

frontend/src/components/ImagePreSignedForm/index.tsx

@@ -5,7 +5,6 @@ import useAuthStore from '@/stores/useAuthStore';
 import { CircleCheckBig, CircleDashed, CircleX, X } from 'lucide-react';
 import { FixedSizeList } from 'react-window';
 import useUploadFiles from '@/hooks/useUploadFiles';
-import useUploadImagePresignedQuery from '@/queries/images/useUploadImagePresignedQuery';
 import { unzipFilesWithPath, extractFilesRecursivelyWithPath } from '@/utils/fileUtils';
 
 interface ImagePreSignedFormProps {
@@ -34,20 +33,16 @@ export default function ImagePreSignedForm({
   const [isUploading, setIsUploading] = useState<boolean>(false);
   const [isUploaded, setIsUploaded] = useState<boolean>(false);
   const [isFailed, setIsFailed] = useState<boolean>(false);
-  const [progress, setProgress] = useState<number>(0);
-  const [uploadStatus, setUploadStatus] = useState<(boolean | null)[]>([]);
+  const [uploadStatus, setUploadStatus] = useState<('uploading' | 'success' | 'failed' | null)[]>([]);
 
-  // Ensure to destructure the uploadFiles function properly from the hook
   const { uploadFiles } = useUploadFiles();
-  const uploadImageFile = useUploadImagePresignedQuery();
+
   const handleClose = () => {
     onClose();
     setFiles([]);
     setInputKey((prevKey) => prevKey + 1);
-    setIsUploading(false);
-    setIsUploaded(false);
     setIsFailed(false);
-    setProgress(0);
+    setIsUploaded(false);
     setUploadStatus([]);
   };
 
@@ -92,9 +87,10 @@ export default function ImagePreSignedForm({
     event.stopPropagation();
     setIsDragging(false);
 
+    let processedFiles: { path: string; file: File }[] = [];
+
     if (uploadType === 'folder') {
       const droppedItems = event.dataTransfer.items;
-      let processedFiles: { path: string; file: File }[] = [];
 
       for (let i = 0; i < droppedItems.length; i++) {
         const item = droppedItems[i];
@@ -106,20 +102,17 @@ export default function ImagePreSignedForm({
           }
         }
       }
-
-      setFiles((prevFiles) => [...prevFiles, ...processedFiles]);
-      setUploadStatus((prevStatus) => [...prevStatus, ...processedFiles.map(() => null)]);
     } else {
       const droppedFiles = event.dataTransfer.files;
       if (droppedFiles) {
-        const processedFiles: { path: string; file: File }[] = [];
         for (const file of Array.from(droppedFiles)) {
           processedFiles.push({ path: file.name, file });
         }
-        setFiles((prevFiles) => [...prevFiles, ...processedFiles]);
-        setUploadStatus((prevStatus) => [...prevStatus, ...processedFiles.map(() => null)]);
       }
     }
+
+    setFiles((prevFiles) => [...prevFiles, ...processedFiles]);
+    setUploadStatus((prevStatus) => [...prevStatus, ...processedFiles.map(() => null)]);
   };
 
   const handleRemoveFile = (index: number) => {
@@ -130,72 +123,51 @@ export default function ImagePreSignedForm({
   const handleUpload = async () => {
     if (files.length > 0) {
       setIsUploading(true);
-      setIsUploaded(false);
       setIsFailed(false);
+      setIsUploaded(false);
+
+      setUploadStatus(files.map(() => 'uploading'));
 
       let finalFiles: { path: string; file: File }[] = [];
 
       for (const file of files) {
         if (file.file.type === 'application/zip' || file.file.type === 'application/x-zip-compressed') {
-          console.log('업로드 전에 ZIP 파일 해제:', file.file.name);
           const unzippedFiles = await unzipFilesWithPath(file.file);
-          console.log('해제된 파일:', unzippedFiles);
           finalFiles = [...finalFiles, ...unzippedFiles];
         } else {
           finalFiles.push(file);
         }
       }
 
-      if (uploadType === 'file') {
-        uploadImageFile.mutate(
-          {
-            memberId,
-            projectId,
-            folderId,
-            files: finalFiles.map(({ file }) => file), // Extract only the file
-            progressCallback: (index: number) => {
-              setUploadStatus((prevStatus) => {
-                const newStatus = [...prevStatus];
-                newStatus[index] = true; // Mark as uploaded
-                return newStatus;
-              });
-            },
+      try {
+        await uploadFiles({
+          files: finalFiles,
+          projectId,
+          folderId,
+          memberId,
+          onProgress: (progress) => {
+            setUploadStatus((prevStatus) => {
+              const completedFiles = Math.round((progress / 100) * files.length);
+              const newStatus = prevStatus.map((status, index) => (index < completedFiles ? 'success' : status));
+              return newStatus;
+            });
           },
-          {
-            onSuccess: () => {
-              handleRefetch();
-              setIsUploaded(true);
-            },
-            onError: () => {
-              setIsFailed(true);
-              setUploadStatus((prevStatus) => prevStatus.map((status) => (status === null ? false : status)));
-            },
-          }
-        );
-      } else {
-        try {
-          await uploadFiles({
-            files: finalFiles,
-            projectId,
-            folderId,
-            memberId,
-            onProgress: (progressValue: number) => {
-              setProgress(progressValue);
-            },
-          });
+          useSingleUpload: uploadType === 'file',
+        });
 
-          setUploadStatus(finalFiles.map(() => true));
-          setIsUploaded(true);
-          handleRefetch();
-        } catch (error) {
-          setIsFailed(true);
-          setUploadStatus(finalFiles.map(() => false));
-          console.error('업로드 실패:', error);
-        }
+        setUploadStatus((prevStatus) => prevStatus.map(() => 'success'));
+        setIsUploaded(true);
+        handleRefetch();
+      } catch (error) {
+        setUploadStatus((prevStatus) => prevStatus.map((status) => (status === 'uploading' ? 'failed' : status)));
+        setIsFailed(true);
+        console.error('업로드 실패:', error);
       }
     }
   };
 
+  const totalProgress = Math.round((uploadStatus.filter((status) => status !== null).length / files.length) * 100);
+
   useEffect(() => {
     onFileCount(files.length);
   }, [files, onFileCount]);
@@ -237,46 +209,44 @@ export default function ImagePreSignedForm({
         </div>
       )}
       {files.length > 0 && (
-        <ul className="m-0 max-h-[260px] list-none overflow-y-auto p-0">
-          <FixedSizeList
-            height={260}
-            itemCount={files.length}
-            itemSize={40}
-            width="100%"
-          >
-            {({ index, style }) => (
-              <li
-                key={index}
-                className="flex items-center justify-between p-1"
-                style={style}
-              >
-                <span className="truncate">{files[index].path}</span>
-                {isUploading ? (
-                  <div className="p-2">
-                    {uploadStatus[index] === true ? (
-                      <CircleCheckBig
-                        className="stroke-green-500"
-                        size={16}
-                        strokeWidth="2"
-                      />
-                    ) : uploadStatus[index] === false ? (
-                      <CircleX
-                        className="stroke-red-500"
-                        size={16}
-                        strokeWidth="2"
-                      />
-                    ) : (
-                      <CircleDashed
-                        className="stroke-gray-500"
-                        size={16}
-                        strokeWidth="2"
-                      />
-                    )}
-                  </div>
-                ) : (
+        <FixedSizeList
+          height={260}
+          itemCount={files.length}
+          itemSize={40}
+          width="100%"
+        >
+          {({ index, style }) => (
+            <div
+              key={index}
+              className="flex items-center justify-between border-b border-gray-200 p-2"
+              style={style}
+            >
+              <span className="truncate">{files[index].path}</span>
+              <div className="flex items-center">
+                {uploadStatus[index] === 'success' ? (
+                  <CircleCheckBig
+                    className="stroke-green-500"
+                    size={16}
+                    strokeWidth="2"
+                  />
+                ) : uploadStatus[index] === 'failed' ? (
+                  <CircleX
+                    className="stroke-red-500"
+                    size={16}
+                    strokeWidth="2"
+                  />
+                ) : uploadStatus[index] === 'uploading' ? (
+                  <CircleDashed
+                    className="animate-spin stroke-gray-500"
+                    size={16}
+                    strokeWidth="2"
+                  />
+                ) : null}
+                {!isUploading && (
                   <button
-                    className="cursor-pointer p-2"
+                    className="ml-2 cursor-pointer p-1"
                     onClick={() => handleRemoveFile(index)}
+                    disabled={uploadStatus[index] === 'success'}
                   >
                     <X
                       color="red"
@@ -285,18 +255,23 @@ export default function ImagePreSignedForm({
                     />
                   </button>
                 )}
-              </li>
-            )}
-          </FixedSizeList>
-        </ul>
+              </div>
+            </div>
+          )}
+        </FixedSizeList>
       )}
       {isUploading ? (
         <Button
           onClick={handleClose}
           variant={isFailed ? 'red' : 'blue'}
-          disabled={!isUploaded && !isFailed}
         >
-          {isFailed ? '업로드 실패 (닫기)' : isUploaded ? '업로드 완료 (닫기)' : `업로드 중... ${progress}%`}
+          {isFailed
+            ? '업로드 실패 (닫기)'
+            : isUploaded
+              ? '업로드 완료 (닫기)'
+              : totalProgress === 0
+                ? '업로드 준비 중...'
+                : `업로드 중... ${totalProgress}%`}
         </Button>
       ) : (
         <Button

frontend/src/hooks/useUploadFiles.ts

@@ -11,55 +11,74 @@ export default function useUploadFiles() {
     folderId,
     memberId,
     onProgress,
+    useSingleUpload = false,
   }: {
     files: { path: string; file: File }[];
     projectId: number;
     folderId: number;
     memberId: number;
     onProgress: (progress: number) => void;
+    useSingleUpload?: boolean;
   }) => {
     const folderIdMap: { [path: string]: number } = { '': folderId };
 
-    const foldersToCreate = Array.from(new Set(files.map(({ path }) => path.split('/').slice(0, -1).join('/'))));
-    foldersToCreate.sort();
+    if (!useSingleUpload) {
+      const foldersToCreate = Array.from(new Set(files.map(({ path }) => path.split('/').slice(0, -1).join('/'))));
+      foldersToCreate.sort();
 
-    for (const folderPath of foldersToCreate) {
-      if (folderPath) {
-        const pathSegments = folderPath.split('/');
-        const parentPath = pathSegments.slice(0, -1).join('/');
-        const folderName = pathSegments[pathSegments.length - 1];
+      for (const folderPath of foldersToCreate) {
+        if (folderPath) {
+          const pathSegments = folderPath.split('/');
+          const parentPath = pathSegments.slice(0, -1).join('/');
+          const folderName = pathSegments[pathSegments.length - 1];
 
-        const parentId = folderIdMap[parentPath] || folderId;
+          const parentId = folderIdMap[parentPath] || folderId;
 
-        const newFolder = await createFolderMutation.mutateAsync({
-          projectId,
-          folderData: {
-            title: folderName,
-            parentId: parentId,
-          },
-        });
+          const newFolder = await createFolderMutation.mutateAsync({
+            projectId,
+            folderData: {
+              title: folderName,
+              parentId: parentId,
+            },
+          });
 
-        folderIdMap[folderPath] = newFolder.id;
+          folderIdMap[folderPath] = newFolder.id;
+        }
       }
     }
 
-    let progress = 0;
+    let completedFiles = 0;
     const totalFiles = files.length;
 
-    for (const { path, file } of files) {
-      const folderPath = path.split('/').slice(0, -1).join('/');
-      const targetFolderId = folderIdMap[folderPath] || folderId;
-
+    if (useSingleUpload) {
       await uploadImageMutation.mutateAsync({
         memberId,
         projectId,
-        folderId: targetFolderId,
-        files: [file],
-        progressCallback: (value) => {
-          progress += value / totalFiles;
+        folderId,
+        files: files.map(({ file }) => file),
+        progressCallback: (progressValue: number) => {
+          const progress = (progressValue / totalFiles) * 100;
           onProgress(Math.round(progress));
         },
       });
+    } else {
+      for (const { path, file } of files) {
+        const folderPath = path.split('/').slice(0, -1).join('/');
+        const targetFolderId = folderIdMap[folderPath] || folderId;
+
+        await uploadImageMutation.mutateAsync({
+          memberId,
+          projectId,
+          folderId: targetFolderId,
+          files: [file],
+          progressCallback: (progressValue: number) => {
+            const progress = ((completedFiles + progressValue / 100) / totalFiles) * 100;
+            onProgress(Math.round(progress));
+          },
+        });
+
+        completedFiles += 1;
+      }
     }
   };
 

frontend/src/pages/Home.tsx

@@ -1,5 +1,5 @@
 import { Link } from 'react-router-dom';
-import GoogleLogo from '@/assets/icons/web_neutral_rd_ctn@1x.png';
+import GoogleLogo from '@/assets/icons/web_light_rd_ctn@1x.png';
 import useAuthStore from '@/stores/useAuthStore';
 import { Button } from '@/components/ui/button';
 
@@ -9,48 +9,45 @@ export default function Home() {
   const { accessToken } = useAuthStore();
 
   return (
-    <div className="flex h-full flex-col items-center justify-center bg-gray-50 p-8">
-      <div className="mb-6 max-w-xl rounded-lg bg-white p-6 shadow-lg">
-        <h2 className="mb-4 text-2xl font-bold text-gray-900">서비스 설명</h2>
-        <p className="mb-4 text-base text-gray-700">
-          본 서비스는 인공 지능(AI) 모델의 학습을 지원하기 위해 웹 기반의 자동 라벨링 도구를 개발하는 것을 목표로
-          합니다.
+    <div className="flex h-full w-full flex-col items-center justify-center bg-gray-50">
+      <div className="text-center">
+        <p className="text-5xl font-semibold leading-[60px] text-black">
+          <span className="text-primary">웹 기반 오토 레이블링</span> 플랫폼
+          <br />
+          <span className="text-primary">WorLabel</span>에 오신 것을 환영합니다
         </p>
-        <p className="mb-4 text-base text-gray-700">
-          기존의 수동적인 방법으로는 대량의 학습 데이터를 처리하는데 시간과 비용이 많이 소모되었습니다. 그러나 본
-          서비스의 결과물인 Auto Labeler를 사용하면, 이러한 문제를 해결할 수 있을 것으로 기대됩니다.
-        </p>
-        <p className="mb-4 text-base text-gray-700">
-          Auto Labeler는 웹 기반으로 동작하므로, 별도의 설치 과정 없이 인터넷 연결 환경에서 쉽게 사용할 수 있습니다.
-          또한, 사용자 친화적인 인터페이스를 제공하여 비전문가도 손쉽게 이용할 수 있도록 설계될 예정입니다.
-        </p>
-        <p className="text-base text-gray-700">
-          본 서비스는 특히 학습 데이터 구축 과정의 효율성과 정확도를 향상시키는 데 중점을 두고 있습니다.
+      </div>
+      <div className="mt-4 text-center">
+        <p className="text-xl font-light leading-[28px] text-black">
+          WorLabel로 레이블링 작업을 간소화하세요.
+          <br />
+          브라우저에서 직접 레이블링을 자동화하여 빠르고 효율적인 워크플로우를 경험하세요.
         </p>
       </div>
 
-      {!accessToken ? (
-        <a
-          href={`${BASE_URL}/login/oauth2/authorization/google`}
-          className="mb-4 transition hover:opacity-90 focus:outline-none focus:ring-2 focus:ring-gray-300 active:opacity-80"
-        >
-          <img
-            src={GoogleLogo}
-            alt="Sign in with Google"
-            className="h-auto w-full"
-          />
-        </a> // 404 에러 방지
-      ) : (
-        <>
+      <div className="mt-8">
+        {!accessToken ? (
+          <a
+            href={`${BASE_URL}/login/oauth2/authorization/google`}
+            className="flex items-center justify-center transition hover:opacity-90 focus:outline-none focus:ring-2 focus:ring-gray-300 active:opacity-80"
+          >
+            <img
+              src={GoogleLogo}
+              alt="Sign in with Google"
+              className="h-auto w-full"
+            />
+          </a>
+        ) : (
           <Button
             asChild
             variant="blue"
             size="lg"
+            className="mt-8"
           >
             <Link to="/browse">시작하기</Link>
           </Button>
-        </>
-      )}
+        )}
+      </div>
     </div>
   );
 }