Update app.py

app.py

@@ -1,4 +1,47 @@
-import gradio as gr
+def process_file(file_path, voucher_number):
+    # 상태 메시지와 결과를 함께 반환하기 위한 변수 초기화
+    status_message = ""
+    output_file_path = None
+    
+    # 로그를 캡처하기 위한 StringIO 객체
+    log_capture = StringIO()
+    
+    if file_path is None:
+        return None, "파일을 업로드해주세요."
+
+    try:
+        # 표준 출력을 캡처
+        original_stdout = sys.stdout
+        sys.stdout = log_capture
+        
+        # 파일 확장자 확인
+        ext = os.path.splitext(file_path)[1].lower()
+
+        if ext == ".xlsx":
+            # 엑셀 파일을 CSV로 변환
+            df = pd.read_excel(file_path)
+            csv_path = file_path.replace(".xlsx", ".csv")
+            df.to_csv(csv_path, index=False)
+            input_path = csv_path
+        else:
+            # 이미 CSV 파일이면 그대로 사용
+            input_path = file_path
+
+        # 메인 전처리 함수 호출 (전표번호 넘겨주기)
+        output_path = main.process_rental_company_with_voucher(input_path, voucher_number)
+        output_file_path = output_path
+        
+        # 성공 메시지 작성
+        status_message = "✅ 파일 변환 성공! 위 버튼을 클릭하여 다운로드하세요."
+        
+    except Exception as e:
+        # 오류 발생 시 간단한 오류 메시지
+        status_message = f"❌ 오류 발생: {str(e)}"
+        
+        # 캡처된 로그 가져오기
+        log_output = log_capture.getvalue()
+        
+        # 필요한 로그 정보만import gradio as gr
 import main  # main.py에 작성된 전처리 로직 호출
 import os
 import pandas as pd
@@ -44,8 +87,7 @@ def process_file(file_path, voucher_number):
         status_message = "✅ 파일 변환 성공! 위 버튼을 클릭하여 다운로드하세요."
         
     except Exception as e:
-        # 오류 발생 시 상세 오류 내용 캡처
-        error_detail = traceback.format_exc()
+        # 오류 발생 시 간단한 오류 메시지
         status_message = f"❌ 오류 발생: {str(e)}"
     
     finally:
@@ -53,7 +95,7 @@ def process_file(file_path, voucher_number):
         log_output = log_capture.getvalue()
         sys.stdout = original_stdout  # 원래 표준 출력으로 복구
         
-        # 필요한 로그 정보만 추출
+        # 주요 정보 추출 (오류 발생 여부와 상관없이)
         important_info = []
         
         # 금액 필드 정보 추출
@@ -92,7 +134,7 @@ def process_file(file_path, voucher_number):
             important_info.extend(amt_info)
         
         # 매핑 오류 정보 추출 (오류가 있을 경우에만)
-        if "매핑되지 않은 팀명" in log_output:
+        if not output_file_path and "매핑되지 않은 팀명" in log_output:  # 오류 발생 시에만 매핑 정보 표시
             unmapped_section = re.search(r"매핑되지 않은 팀명 (\d+)개:(.*?)(?=\n\n|\Z)", log_output, re.DOTALL)
             if unmapped_section:
                 unmapped_count = unmapped_section.group(1)
@@ -101,9 +143,9 @@ def process_file(file_path, voucher_number):
                 for team in unmapped_teams:
                     important_info.append(f"- '{team}'")
         
-        # 중요 정보를 상태 메시지에 추가
-        if important_info and not output_file_path:  # 오류 발생 시에만 중요 정보 표시
-            status_message += "\n\n🔍 주요 정보:\n" + "\n".join(important_info)
+        # 주요 정보를 상태 메시지에 추가 (항상)
+        if important_info:
+            status_message += "\n\n" + "\n".join(important_info)
     
     return output_file_path, status_message