Update app.py

app.py

@@ -21,12 +21,25 @@ def request_batch_key(onfftid, pay_type, method, cert_type, order_no, user_nm, u
         "tot_amt": tot_amt
     }
     
-    response = requests.post(url, data=payload)
-    
-    if response.status_code == 200:
-        return response.json()
-    else:
-        return f"Error: {response.status_code} - {response.text}"
+    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):
@@ -46,12 +59,25 @@ def request_payment(onfftid, fix_key, tot_amt, card_user_type, auth_value, insta
         "order_no": order_no
     }
     
-    response = requests.post(url, data=payload)
-    
-    if response.status_code == 200:
-        return response.json()
-    else:
-        return f"Error: {response.status_code} - {response.text}"
+    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)
+        }
 
 with gr.Blocks() as demo:
     gr.Markdown("# API 연동 테스트")