Update app.py

app.py

@@ -1,186 +1,72 @@
-from bs4 import BeautifulSoup
 import gradio as gr
 import requests
-from html.parser import HTMLParser
 
-
-class MyHTMLParser(HTMLParser):
-    def __init__(self):
-        super().__init__()
-        self.data = {}
-        self.current_tag = None
-
-    def handle_starttag(self, tag, attrs):
-        self.current_tag = tag
-        if tag == 'input':
-            d = dict(attrs)
-            if 'name' in d and 'value' in d:
-                self.data[d['name']] = d['value']
-
-    def handle_data(self, data):
-        if self.current_tag == 'script':
-            if 'fix_key' in data:
-                for line in data.split('\n'):
-                    if 'fix_key' in line:
-                        key, value = line.split('=')
-                        self.data['fix_key'] = value.strip().strip("';")
-
-def extract_info_from_html(html_content):
-    parser = MyHTMLParser()
-    parser.feed(html_content)
-    
-    return {
-        "order_number": parser.data.get('ordr_idxx', 'Not found'),
-        "buyer_name": parser.data.get('buyr_name', 'Not found'),
-        "fix_key": parser.data.get('fix_key', 'Not found'),
-        # Add more extracted fields as needed
-    }
-    
-
-
-def request_batch_key(onfftid, pay_type, method, cert_type, order_no, user_nm, user_phone2, product_nm, 
-                      card_user_type, card_no, expire_date, auth_value, password, tot_amt):
-    url = "https://store.onoffkorea.co.kr/fix/index.php"
-    payload = {
+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"
+    data = {
         "onfftid": onfftid,
-        "pay_type": pay_type,
-        "method": method,
-        "cert_type": cert_type,
-        "order_no": order_no,
+        "tot_amt": tot_amt,
+        "com_tax_amt": com_tax_amt,
+        "com_free_amt": com_free_amt,
+        "com_vat_amt": com_vat_amt,
+        "card_no": card_no,
+        "install_period": install_period,
         "user_nm": user_nm,
         "user_phone2": user_phone2,
         "product_nm": product_nm,
-        "card_user_type": card_user_type,
-        "card_no": card_no,
         "expire_date": expire_date,
+        "cert_type": cert_type,
+        "card_user_type": card_user_type,
         "auth_value": auth_value,
         "password": password,
-        "tot_amt": tot_amt
-    }
-    
-    headers = {
-        'Accept': 'application/json',
-        'Content-Type': 'application/x-www-form-urlencoded',
+        "card_nm": card_nm,
+        "order_no": order_no,
+        "pay_type": pay_type
     }
     
-    try:
-        response = requests.post(url, data=payload, headers=headers)
-        response.raise_for_status()
-        
-        # Check if the response is JSON
+    response = requests.post(url, data=data)
+    if response.status_code == 200:
         try:
             return response.json()
-        except requests.exceptions.JSONDecodeError:
-            # If not JSON, assume it's HTML and extract information
-            extracted_info = extract_info_from_html(response.text)
-            return {
-                "status": "HTML response received",
-                "extracted_info": extracted_info,
-                "raw_html": response.text[:1000]  # First 1000 characters of HTML for reference
-            }
-    except requests.exceptions.RequestException as e:
-        return {
-            "error": str(e),
-            "status_code": getattr(e.response, 'status_code', None),
-            "raw_response": getattr(e.response, 'text', None)
-        }
+        except ValueError:
+            return {"error": "Non-JSON response", "details": response.text}
+    else:
+        return {"error": "Request failed", "status_code": response.status_code}
 
-def request_payment(onfftid, fix_key, tot_amt, card_user_type, auth_value, install_period, user_nm, 
-                    user_phone2, product_nm, pay_type, method, order_no):
-    url = "https://store.onoffkorea.co.kr/fix/index.php"
-    payload = {
-        "onfftid": onfftid,
-        "fix_key": fix_key,
-        "tot_amt": tot_amt,
-        "card_user_type": card_user_type,
-        "auth_value": auth_value,
-        "install_period": install_period,
-        "user_nm": user_nm,
-        "user_phone2": user_phone2,
-        "product_nm": product_nm,
-        "pay_type": pay_type,
-        "method": method,
-        "order_no": order_no
-    }
-    
-    headers = {
-        'Accept': 'application/json',
-        'Content-Type': 'application/x-www-form-urlencoded',
-    }
-    
-    try:
-        response = requests.post(url, data=payload, headers=headers)
-        response.raise_for_status()
-        
-        # Check if the response is JSON
-        try:
-            return response.json()
-        except requests.exceptions.JSONDecodeError:
-            # If not JSON, assume it's HTML and extract information
-            extracted_info = extract_info_from_html(response.text)
-            return {
-                "status": "HTML response received",
-                "extracted_info": extracted_info,
-                "raw_html": response.text[:1000]  # First 1000 characters of HTML for reference
-            }
-    except requests.exceptions.RequestException as e:
-        return {
-            "error": str(e),
-            "status_code": getattr(e.response, 'status_code', None),
-            "raw_response": getattr(e.response, 'text', None)
-        }
+with gr.Blocks() as app:
+    with gr.Tab("카드 결제 승인 요청"):
+        with gr.Row():
+            with gr.Column(scale=1):
+                onfftid = gr.Textbox(label="온오프코리아 TID", value="OFPT000000011017", readonly=True)
+                tot_amt = gr.Number(label="결제금액", placeholder="금액을 입력하세요")
+                com_tax_amt = gr.Number(label="과세승인금액", optional=True)
+                com_free_amt = gr.Number(label="비과세승인금액", optional=True)
+                com_vat_amt = gr.Number(label="부가세", optional=True)
+                card_no = gr.Textbox(label="카드번호", placeholder="카드번호를 입력하세요")
+                install_period = gr.Dropdown([str(i).zfill(2) for i in range(13)], label="할부기간 선택")
+                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")
+                cert_type = gr.Radio(choices=["0 - 인증", "1 - 비인증"], label="인증여부 선택")
+                card_user_type = gr.Radio(choices=["0 - 개인카드", "1 - 법인카드"], label="카드유형 선택")
+                auth_value = gr.Textbox(label="인증번호", placeholder="개인은 주민번호 앞 6자리, 법인은 사업자등록번호")
+                password = gr.Textbox(label="카드 비밀번호 앞 2자리", placeholder="비밀번호 앞 2자리")
+                card_nm = gr.Textbox(label="카드사명", optional=True, placeholder="카드사명을 입력하세요")
+                order_no = gr.Textbox(label="주문번호", optional=True, placeholder="주문번호를 입력하세요")
+                pay_type = gr.Textbox(label="결제타입", value="card", readonly=True)
 
-with gr.Blocks() as demo:
-    gr.Markdown("# API 연동 테스트")
-    
-    with gr.Tab("카드 배치키 발급 요청"):
-        onfftid = gr.Textbox(label="온오프코리아 TID")
-        pay_type = gr.Textbox(label="결제 타입", value="fixKey")
-        method = gr.Textbox(label="메소드", value="request")
-        cert_type = gr.Radio(["0", "1"], label="인증 여부", value="0")
-        order_no = gr.Textbox(label="주문번호")
-        user_nm = gr.Textbox(label="주문자 이름")
-        user_phone2 = gr.Textbox(label="주문자 연락처")
-        product_nm = gr.Textbox(label="상품명")
-        card_user_type = gr.Radio(["0", "1"], label="카드 타입", value="0")
-        card_no = gr.Textbox(label="카드번호")
-        expire_date = gr.Textbox(label="유효기간(YYMM)")
-        auth_value = gr.Textbox(label="주민(사업자)등록번호")
-        password = gr.Textbox(label="카드비밀번호 앞 2자리")
-        tot_amt = gr.Number(label="결제금액")
-        
-        batch_key_button = gr.Button("배치키 발급 요청")
-        batch_key_output = gr.JSON(label="응답 결과")
-        
-        batch_key_button.click(
-            request_batch_key,
-            inputs=[onfftid, pay_type, method, cert_type, order_no, user_nm, user_phone2, product_nm, 
-                    card_user_type, card_no, expire_date, auth_value, password, tot_amt],
-            outputs=batch_key_output
-        )
-    
-    with gr.Tab("카드 배치키 결제 승인 요청"):
-        onfftid_pay = gr.Textbox(label="온오프코리아 TID")
-        fix_key = gr.Textbox(label="카드 배치 키")
-        tot_amt_pay = gr.Number(label="결제금액")
-        card_user_type_pay = gr.Radio(["0", "1"], label="카드 타입", value="0")
-        auth_value_pay = gr.Textbox(label="주민(사업자)등록번호")
-        install_period = gr.Dropdown(["00", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12"], label="할부기간")
-        user_nm_pay = gr.Textbox(label="고객명")
-        user_phone2_pay = gr.Textbox(label="고객연락처")
-        product_nm_pay = gr.Textbox(label="상품명")
-        pay_type_pay = gr.Textbox(label="결제 타입", value="fixKey")
-        method_pay = gr.Textbox(label="메소드", value="authTran")
-        order_no_pay = gr.Textbox(label="주문번호")
-        
-        payment_button = gr.Button("결제 승인 요청")
-        payment_output = gr.JSON(label="응답 결과")
-        
-        payment_button.click(
-            request_payment,
-            inputs=[onfftid_pay, fix_key, tot_amt_pay, card_user_type_pay, auth_value_pay, install_period, 
-                    user_nm_pay, user_phone2_pay, product_nm_pay, pay_type_pay, method_pay, order_no_pay],
-            outputs=payment_output
+                submit_button = gr.Button("결제 승인 요청")
+                result_area = gr.JSON(label="결과")
+
+        submit_button.click(
+            fn=card_authorization,
+            inputs=[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],
+            outputs=result_area
         )
 
-demo.launch()
+app.launch()