Update app.py

app.py

@@ -2,28 +2,61 @@ import gradio as gr
 import main  # main.py에 작성된 전처리 로직 호출
 import os
 import pandas as pd
+import traceback
+import sys
+from io import StringIO
 
 def process_file(file_path, voucher_number):
+    # 상태 메시지와 결과를 함께 반환하기 위한 변수 초기화
+    status_message = ""
+    output_file_path = None
+    
+    # 로그를 캡처하기 위한 StringIO 객체
+    log_capture = StringIO()
+    
     if file_path is None:
-        return None
+        return None, "파일을 업로드해주세요."
 
-    # 파일 확장자 확인
-    ext = os.path.splitext(file_path)[1].lower()
+    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
+        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)
-
-    return output_path
+        # 메인 전처리 함수 호출 (전표번호 넘겨주기)
+        output_path = main.process_rental_company_with_voucher(input_path, voucher_number)
+        output_file_path = output_path
+        
+        # 성공 메시지 작성
+        status_message = "✅ 파일 변환 성공! 아래 버튼을 클릭하여 다운로드하세요."
+        
+    except Exception as e:
+        # 오류 발생 시 상세 오류 내용 캡처
+        error_detail = traceback.format_exc()
+        status_message = f"❌ 오류 발생: {str(e)}\n\n🔍 상세 내용:\n{error_detail}"
+    
+    finally:
+        # 캡처된 로그 가져오기
+        log_output = log_capture.getvalue()
+        sys.stdout = original_stdout  # 원래 표준 출력으로 복구
+        
+        # 로그 출력이 있으면 상태 메시지에 추가
+        if log_output.strip():
+            status_message += f"\n\n📋 처리 로그:\n{log_output}"
+    
+    return output_file_path, status_message
 
 # Gradio 인터페이스 구성
 with gr.Blocks() as demo:
@@ -33,7 +66,7 @@ with gr.Blocks() as demo:
         file_input = gr.File(
             label="렌탈료 파일 업로드 (CSV 또는 Excel)",
             file_types=[".csv", ".xlsx"],
-            type="filepath"   # ⭐ 여기가 중요!
+            type="filepath"
         )
         voucher_input = gr.Textbox(
             label="전표번호 입력",
@@ -41,22 +74,31 @@ with gr.Blocks() as demo:
         )
 
     with gr.Row():
-        submit_btn = gr.Button("제출")
+        submit_btn = gr.Button("제출", variant="primary")
         clear_btn = gr.Button("지우기")
 
+    # 상태 메시지를 표시할 텍스트 영역 추가
+    status_output = gr.Textbox(
+        label="처리 상태",
+        placeholder="파일을 제출하면 처리 상태가 여기에 표시됩니다.",
+        lines=10
+    )
+    
     output_file = gr.File(label="전처리 완료 파일 다운로드")
 
     # 버튼 클릭 이벤트 연결
     submit_btn.click(
         fn=process_file,
         inputs=[file_input, voucher_input],
-        outputs=output_file
+        outputs=[output_file, status_output]
     )
 
     clear_btn.click(
-        fn=lambda: (None, ""),
+        fn=lambda: (None, "", ""),
         inputs=[],
-        outputs=[file_input, voucher_input]
+        outputs=[file_input, voucher_input, status_output]
     )
 
-demo.launch()
+# 시작 메시지 표시
+print("ERP 자동 전표 변환기가 시작되었습니다.")
+demo.launch()