Update app.py

app.py

@@ -1,10 +1,32 @@
 import gradio as gr
 import requests
-
 import xml.etree.ElementTree as ET
 
 def card_authorization(onfftid, tot_amt, com_tax_amt, com_free_amt, com_vat_amt, card_no, install_period, user_nm, user_phone2, product_nm, expire_date, cert_type, card_user_type, auth_value, password, card_nm, order_no, pay_type):
     url = "https://store.onoffkorea.co.kr/payment/index.php"
+    
+    # 필수 파라미터 확인
+    if not all([onfftid, tot_amt, card_no, install_period, user_nm, user_phone2, product_nm, expire_date, cert_type, card_user_type]):
+        return {"error": "필수 파라미터가 누락되었습니다."}
+    
+    # 인증 관련 파라미터 처리
+    if cert_type == "0 - 인증":
+        if not auth_value or not password:
+            return {"error": "인증 결제 시 인증번호와 비밀번호는 필수입니다."}
+        cert_type = "0"
+    elif cert_type == "1 - 비인증":
+        cert_type = "1"
+    else:
+        return {"error": "잘못된 인증 유형입니다."}
+
+    # 카드 유형 처리
+    if card_user_type == "0 - 개인카드":
+        card_user_type = "0"
+    elif card_user_type == "1 - 법인카드":
+        card_user_type = "1"
+    else:
+        return {"error": "잘못된 카드 유형입니다."}
+
     data = {
         "onfftid": onfftid,
         "tot_amt": tot_amt,
@@ -21,48 +43,52 @@ def card_authorization(onfftid, tot_amt, com_tax_amt, com_free_amt, com_vat_amt,
         "card_user_type": card_user_type,
         "auth_value": auth_value,
         "password": password,
-        "card_nm": card_nm,
-        "order_no": order_no,
-        "pay_type": pay_type
+        "card_nm": card_nm or '',
+        "order_no": order_no or '',
+        "pay_type": pay_type or 'card'
     }
+
     response = requests.post(url, data=data)
     if response.status_code == 200:
         try:
-            return response.json()
-        except ValueError:
-            # XML 응답 처리
-            try:
-                root = ET.fromstring(response.text)
-                result = {}
-                for child in root:
-                    result[child.tag] = child.text
-                return result
-            except ET.ParseError:
-                return {"error": "Invalid XML response", "details": response.text}
+            root = ET.fromstring(response.text)
+            result = {}
+            for child in root:
+                result[child.tag] = child.text
+            
+            # 결과 코드 확인
+            if result.get('result_cd') == '0000':
+                result['status'] = '성공'
+            else:
+                result['status'] = '실패'
+            
+            return result
+        except ET.ParseError:
+            return {"error": "응답 파싱 오류", "details": response.text}
     else:
-        return {"error": "Request failed", "status_code": response.status_code}
+        return {"error": "요청 실패", "status_code": response.status_code}
 
 with gr.Blocks() as app:
     with gr.Tab("카드 결제 승인 요청"):
         with gr.Row():
             with gr.Column(scale=1):
                 onfftid = gr.Textbox(label="온오프코리아 TID", value="OFPT000000011017")
-                tot_amt = gr.Number(label="결제금액 (금액을 입력하세요)")
-                com_tax_amt = gr.Number(label="과세승인금액 (필요시 입력)")
-                com_free_amt = gr.Number(label="비과세승인금액 (필요시 입력)")
-                com_vat_amt = gr.Number(label="부가세 (필요시 입력)")
-                card_no = gr.Textbox(label="카드번호", placeholder="카드번호를 입력하세요")
-                install_period = gr.Dropdown([str(i).zfill(2) for i in range(13)], label="할부기간 선택", value="00")
-                user_nm = gr.Textbox(label="결제자명", placeholder="결제자의 이름을 입력하세요")
-                user_phone2 = gr.Textbox(label="결제자 연락처", placeholder="연락처를 입력하세요")
-                product_nm = gr.Textbox(label="상품명", placeholder="상품명을 입력하세요")
-                expire_date = gr.Textbox(label="유효기간(YYMM)", placeholder="예: 2306")
+                tot_amt = gr.Number(label="결제금액", required=True)
+                com_tax_amt = gr.Number(label="과세승인금액")
+                com_free_amt = gr.Number(label="비과세승인금액")
+                com_vat_amt = gr.Number(label="부가세")
+                card_no = gr.Textbox(label="카드번호", required=True)
+                install_period = gr.Dropdown(choices=[f"{str(i).zfill(2)} - {i if i > 0 else '일시불'}개월" for i in range(13)], label="할부기간 선택", value="00 - 일시불")
+                user_nm = gr.Textbox(label="결제자명", required=True)
+                user_phone2 = gr.Textbox(label="결제자 연락처", required=True)
+                product_nm = gr.Textbox(label="상품명", required=True)
+                expire_date = gr.Textbox(label="유효기간(YYMM)", required=True)
                 cert_type = gr.Radio(choices=["0 - 인증", "1 - 비인증"], label="인증여부 선택", value="0 - 인증")
                 card_user_type = gr.Radio(choices=["0 - 개인카드", "1 - 법인카드"], label="카드유형 선택", value="0 - 개인카드")
-                auth_value = gr.Textbox(label="인증번호", placeholder="개인은 주민번호 앞 6자리, 법인은 사업자등록번호")
-                password = gr.Textbox(label="카드 비밀번호 앞 2자리", placeholder="비밀번호 앞 2자리")
-                card_nm = gr.Textbox(label="카드사명 (필요시 입력)", placeholder="카드사명을 입력하세요")
-                order_no = gr.Textbox(label="주문번호 (필요시 입력)", placeholder="주문번호를 입력하세요")
+                auth_value = gr.Textbox(label="인증번호 (인증 시 필수)")
+                password = gr.Textbox(label="카드 비밀번호 앞 2자리 (인증 시 필수)")
+                card_nm = gr.Textbox(label="카드사명")
+                order_no = gr.Textbox(label="주문번호")
                 pay_type = gr.Textbox(label="결제타입", value="card")
                 
         submit_button = gr.Button("결제 승인 요청")