Update app.py

app.py

@@ -17,10 +17,10 @@ def load_code(filename: str) -> str:
     try:
         with open(filename, 'r', encoding='utf-8') as file:
             return file.read()
-    except FileNotFoundError:
-        return f"{filename} 파일을 찾을 수 없습니다."
-    except Exception as e:
-        return f"파일을 읽는 중 오류가 발생했습니다: {str(e)}"
+        except FileNotFoundError:
+            return f"{filename} 파일을 찾을 수 없습니다."
+        except Exception as e:
+            return f"파일을 읽는 중 오류가 발생했습니다: {str(e)}"
 
 def load_parquet(filename: str) -> str:
     try:
@@ -127,7 +127,9 @@ def text_to_parquet(text: str) -> Tuple[str, str, str]:
             dtype=str,
             quoting=csv.QUOTE_ALL,  # 모든 필드를 큰따옴표로 감싸는 것으로 처리
             escapechar='\\',        # 이스케이프 문자 설정
-            engine='python'         # Python 엔진 사용
+            engine='python',        # Python 엔진 사용
+            header=None,            # 첫 번째 행을 열 이름으로 사용하지 않음
+            names=['id', 'text', 'label', 'metadata']  # 열 이름 지정
         )
         # 데이터 유형 최적화
         df = df.astype({'id': 'int32', 'text': 'string', 'label': 'string', 'metadata': 'string'})
@@ -402,3 +404,4 @@ if __name__ == "__main__":
 
 
 
+