Update app.py

app.py

@@ -1,5 +1,19 @@
 import gradio as gr
 import requests
+from bs4 import BeautifulSoup
+
+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'
+    
+    return {
+        "order_number": order_number,
+        "buyer_name": buyer_name,
+        # 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):
@@ -21,56 +35,25 @@ def request_batch_key(onfftid, pay_type, method, cert_type, order_no, user_nm, u
         "tot_amt": tot_amt
     }
     
-    try:
-        response = requests.post(url, data=payload)
-        response.raise_for_status()  # Raises an HTTPError for bad responses
-        
-        # Try to parse JSON, if fails, return the raw text
-        try:
-            return response.json()
-        except requests.exceptions.JSONDecodeError:
-            return {
-                "error": "Invalid JSON response",
-                "status_code": response.status_code,
-                "raw_response": response.text
-            }
-    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)
-        }
-
-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',  # 요청이 JSON을 선호한다고 서버에 알립니다
+        'Content-Type': 'application/x-www-form-urlencoded',  # 폼 데이터로 전송
     }
     
     try:
-        response = requests.post(url, data=payload)
-        response.raise_for_status()  # Raises an HTTPError for bad responses
+        response = requests.post(url, data=payload, headers=headers)
+        response.raise_for_status()
         
-        # Try to parse JSON, if fails, return the raw text
+        # 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 {
-                "error": "Invalid JSON response",
-                "status_code": response.status_code,
-                "raw_response": response.text
+                "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 {
@@ -79,6 +62,8 @@ def request_payment(onfftid, fix_key, tot_amt, card_user_type, auth_value, insta
             "raw_response": getattr(e.response, 'text', None)
         }
 
+# The request_payment function can be updated similarly
+
 with gr.Blocks() as demo:
     gr.Markdown("# API 연동 테스트")
     
@@ -108,28 +93,6 @@ with gr.Blocks() as demo:
             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
-        )
+    # The "카드 배치키 결제 승인 요청" tab can be updated similarly
 
 demo.launch()