Update app.py

app.py

@@ -1,17 +1,26 @@
+from bs4 import BeautifulSoup
 import gradio as gr
 import requests
-from bs4 import BeautifulSoup
+from html.parser import HTMLParser
+
+class MyHTMLParser(HTMLParser):
+    def __init__(self):
+        super().__init__()
+        self.data = {}
+
+    def handle_starttag(self, tag, attrs):
+        if tag == 'input':
+            d = dict(attrs)
+            if 'name' in d and 'value' in d:
+                self.data[d['name']] = d['value']
 
 def extract_info_from_html(html_content):
-    soup = BeautifulSoup(html_content, 'html.parser')
-    
-    # Example: Extract order number and buyer name
-    order_number = soup.find('input', {'name': 'ordr_idxx'})['value'] if soup.find('input', {'name': 'ordr_idxx'}) else 'Not found'
-    buyer_name = soup.find('input', {'name': 'buyr_name'})['value'] if soup.find('input', {'name': 'buyr_name'}) else 'Not found'
+    parser = MyHTMLParser()
+    parser.feed(html_content)
     
     return {
-        "order_number": order_number,
-        "buyer_name": buyer_name,
+        "order_number": parser.data.get('ordr_idxx', 'Not found'),
+        "buyer_name": parser.data.get('buyr_name', 'Not found'),
         # Add more extracted fields as needed
     }
 
@@ -36,8 +45,8 @@ def request_batch_key(onfftid, pay_type, method, cert_type, order_no, user_nm, u
     }
     
     headers = {
-        'Accept': 'application/json',  # 요청이 JSON을 선호한다고 서버에 알립니다
-        'Content-Type': 'application/x-www-form-urlencoded',  # 폼 데이터로 전송
+        'Accept': 'application/json',
+        'Content-Type': 'application/x-www-form-urlencoded',
     }
     
     try:
@@ -62,7 +71,50 @@ def request_batch_key(onfftid, pay_type, method, cert_type, order_no, user_nm, u
             "raw_response": getattr(e.response, 'text', None)
         }
 
-# The request_payment function can be updated similarly
+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 demo:
     gr.Markdown("# API 연동 테스트")
@@ -93,6 +145,28 @@ with gr.Blocks() as demo:
             outputs=batch_key_output
         )
     
-    # The "카드 배치키 결제 승인 요청" tab can be updated similarly
+    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
+        )
 
 demo.launch()