Update app.py

app.py

@@ -7,7 +7,7 @@ import sys
 import re
 from io import StringIO
 
-def process_file(file_path, voucher_number):
+def process_file(file_path, voucher_number, employee_number):
     # 상태 메시지와 결과를 함께 반환하기 위한 변수 초기화
     status_message = ""
     output_file_path = None
@@ -15,8 +15,15 @@ def process_file(file_path, voucher_number):
     # 로그를 캡처하기 위한 StringIO 객체
     log_capture = StringIO()
     
+    # 입력값 검증
     if file_path is None:
         return None, "파일을 업로드해주세요."
+    
+    if not voucher_number or not voucher_number.strip():
+        return None, "전표번호를 입력해주세요."
+        
+    if not employee_number or not employee_number.strip():
+        return None, "사원번호를 입력해주세요. 사원번호는 필수 입력값입니다."
 
     try:
         # 표준 출력을 캡처
@@ -36,8 +43,8 @@ def process_file(file_path, voucher_number):
             # 이미 CSV 파일이면 그대로 사용
             input_path = file_path
 
-        # 메인 전처리 함수 호출 (전표번호 넘겨주기)
-        output_path = main.process_rental_company_with_voucher(input_path, voucher_number)
+        # 메인 전처리 함수 호출 (전표번호와 사원번호 넘겨주기)
+        output_path = main.process_rental_company_with_voucher(input_path, voucher_number, employee_number)
         output_file_path = output_path
         
         # 성공 메시지 작성
@@ -108,7 +115,7 @@ def process_file(file_path, voucher_number):
 
 # Gradio 인터페이스 구성
 with gr.Blocks() as demo:
-    gr.Markdown("# ERP 자동 전표 변환기\n\n업로드할 파일과 전표번호를 입력하세요.")
+    gr.Markdown("# ERP 자동 전표 변환기\n\n업로드할 파일과 전표번호, 사원번호를 입력하세요.")
 
     with gr.Row():
         file_input = gr.File(
@@ -116,10 +123,17 @@ with gr.Blocks() as demo:
             file_types=[".csv", ".xlsx"],
             type="filepath"
         )
+        
+    with gr.Row():
         voucher_input = gr.Textbox(
-            label="전표번호 입력",
+            label="전표번호 입력 (필수)",
             placeholder="예: 20250427001"
         )
+        
+        employee_input = gr.Textbox(
+            label="사원번호 입력 (필수)",
+            placeholder="예: 00616"
+        )
 
     with gr.Row():
         submit_btn = gr.Button("제출", variant="primary")
@@ -137,14 +151,14 @@ with gr.Blocks() as demo:
     # 버튼 클릭 이벤트 연결
     submit_btn.click(
         fn=process_file,
-        inputs=[file_input, voucher_input],
+        inputs=[file_input, voucher_input, employee_input],
         outputs=[output_file, status_output]
     )
 
     clear_btn.click(
-        fn=lambda: (None, "", ""),
+        fn=lambda: (None, "", "", ""),
         inputs=[],
-        outputs=[file_input, voucher_input, status_output]
+        outputs=[file_input, voucher_input, employee_input, status_output]
     )
 
 # 시작 메시지 표시