Update app.py

app.py

@@ -1,150 +1,140 @@
-!pip install gradio==3.30.0
-
 import gradio as gr
 
-VAT_RATE = 0.07       # อัตรา VAT 7%
-WITHHOLD_RATE = 0.03  # อัตราหัก ณ ที่จ่าย 3%
-
-def calculate_price(calculation_type, wht_type, input_amount):
-    """
-    calculation_type: เลือก "Forward" (Base -> Net) หรือ "Reverse" (Net -> Base)
-    wht_type: เลือก "WHT from Base" หรือ "WHT from Total"
-    input_amount: จำนวนเงินที่ใส่เข้ามา
-    """
-    
-    # เตรียมตัวแปรผลลัพธ์ (เก็บเป็น dict เพื่อส่งกลับไปแสดง)
-    result = {
-        "ราคาเริ่มต้น (Base)": 0.0,
-        "VAT 7%": 0.0,
-        "ยอดรวม (รวม VAT)": 0.0,
-        "หัก ณ ที่จ่าย": 0.0,
-        "ยอดสุทธิที่ได้รับ (Net)": 0.0
-    }
-    
-    # ------------------------------------------------
-    # 1) กรณีหัก ณ ที่จ่ายจาก "Base"
-    # ------------------------------------------------
-    if wht_type == "WHT จากฐาน (Base)":
-        if calculation_type == "Forward (รู้ Base -> หา Net)":
-            base = input_amount
-            vat_amount = base * VAT_RATE
-            total_with_vat = base + vat_amount
-            wht_amount = base * WITHHOLD_RATE
-            net_amount = total_with_vat - wht_amount
-            
-            result["ราคาเริ่มต้น (Base)"] = base
-            result["VAT 7%"] = vat_amount
-            result["ยอดรวม (รวม VAT)"] = total_with_vat
-            result["หัก ณ ที่จ่าย"] = wht_amount
-            result["ยอดสุทธิที่ได้รับ (Net)"] = net_amount
+vat_rate = 0.07
+withholding_rate = 0.03
 
-        elif calculation_type == "Reverse (รู้ Net -> หา Base)":
-            # Net = base * 1.07 - base * 0.03 = base * 1.04
-            # base = Net / 1.04
-            net = input_amount
-            base = net / (1 + VAT_RATE - WITHHOLD_RATE)  # 1.07 - 0.03 = 1.04
-            vat_amount = base * VAT_RATE
-            total_with_vat = base + vat_amount
-            wht_amount = base * WITHHOLD_RATE
-            
-            result["ราคาเริ่มต้น (Base)"] = base
-            result["VAT 7%"] = vat_amount
-            result["ยอดรวม (รวม VAT)"] = total_with_vat
-            result["หัก ณ ที่จ่าย"] = wht_amount
-            result["ยอดสุทธิที่ได้รับ (Net)"] = net
+def debug_print(data, label=""):
+    """ฟังก์ชันสำหรับ debug ข้อมูล"""
+    print(f"\n🔍 [DEBUG] {label}:")
+    for k, v in data.items():
+        print(f"{k}: {v} ({type(v)})")
 
-    # ------------------------------------------------
-    # 2) กรณีหัก ณ ที่จ่ายจาก "Total (base+VAT)"
-    # ------------------------------------------------
-    elif wht_type == "WHT จากยอดรวม (Total)":
-        if calculation_type == "Forward (รู้ Base -> หา Net)":
-            base = input_amount
-            vat_amount = base * VAT_RATE
-            total_with_vat = base + vat_amount
-            wht_amount = total_with_vat * WITHHOLD_RATE
-            net_amount = total_with_vat - wht_amount
-            
-            result["ราคาเริ่มต้น (Base)"] = base
-            result["VAT 7%"] = vat_amount
-            result["ยอดรวม (รวม VAT)"] = total_with_vat
-            result["หัก ณ ที่จ่าย"] = wht_amount
-            result["ยอดสุทธิที่ได้รับ (Net)"] = net_amount
+def calculate_tax(
+    base: float = None,
+    after_vat: float = None,
+    after_vat_withholding: float = None,
+    after_withholding: float = None
+):
+    try:
+        # ตรวจสอบประเภทข้อมูลและค่าที่ได้รับ
+        inputs = {
+            "base": base,
+            "after_vat": after_vat,
+            "after_vat_withholding": after_vat_withholding,
+            "after_withholding": after_withholding
+        }
+        debug_print(inputs, "Inputs Received")
+        
+        # เช็คค่าที่กรอกเข้ามา
+        filled = 0
+        for key, value in inputs.items():
+            if value not in [None, 0.0, 0]:
+                filled += 1
+                active_field = key
+        debug_print({"filled": filled, "active_field": active_field}, "Validation")
+        
+        if filled != 1:
+            raise ValueError(f"กรุณากรอกข้อมูลเพียง 1 ช่อง (ตรวจพบ {filled} ช่อง)")
+        
+        # คำนวณจากช่องที่กรอก
+        if active_field == "base":
+            base = float(base)
+            vat = base * vat_rate
+            withholding = base * withholding_rate
+            results = {
+                "after_vat": base + vat,
+                "after_vat_withholding": (base + vat) - withholding,
+                "after_withholding": base - withholding,
+                "vat": vat,
+                "withholding": withholding
+            }
+        elif active_field == "after_vat":
+            after_vat = float(after_vat)
+            base = after_vat / (1 + vat_rate)
+            vat = base * vat_rate
+            withholding = base * withholding_rate
+            results = {
+                "base": base,
+                "after_vat_withholding": after_vat - withholding,
+                "after_withholding": base - withholding,
+                "vat": vat,
+                "withholding": withholding
+            }
+        elif active_field == "after_vat_withholding":
+            after_vat_withholding = float(after_vat_withholding)
+            base = after_vat_withholding / (1 + vat_rate - withholding_rate)
+            vat = base * vat_rate
+            withholding = base * withholding_rate
+            results = {
+                "base": base,
+                "after_vat": base + vat,
+                "after_withholding": base - withholding,
+                "vat": vat,
+                "withholding": withholding
+            }
+        elif active_field == "after_withholding":
+            after_withholding = float(after_withholding)
+            base = after_withholding / (1 - withholding_rate)
+            vat = base * vat_rate
+            withholding = base * withholding_rate
+            results = {
+                "base": base,
+                "after_vat": base + vat,
+                "after_vat_withholding": (base + vat) - withholding,
+                "vat": vat,
+                "withholding": withholding
+            }
+        
+        # ปัดเศษและแปลงรูปแบบ
+        formatted = {k: round(v, 2) for k, v in results.items()}
+        formatted["error"] = ""
+        debug_print(formatted, "Results")
+        return formatted
+        
+    except Exception as e:
+        error_msg = f"⚠️ เกิดข้อผิดพลาด: {str(e)}"
+        debug_print({"error": error_msg}, "Error")
+        return {"error": error_msg}
 
-        elif calculation_type == "Reverse (รู้ Net -> หา Base)":
-            # Net = (base * (1+VAT_RATE)) - ( (base*(1+VAT_RATE)) * WITHHOLD_RATE )
-            # Net = (base * (1+VAT_RATE)) * (1 - WITHHOLD_RATE)
-            # base = Net / [ (1+VAT_RATE)*(1 - WITHHOLD_RATE) ]
-            net = input_amount
-            base = net / ((1 + VAT_RATE) * (1 - WITHHOLD_RATE))
-            vat_amount = base * VAT_RATE
-            total_with_vat = base + vat_amount
-            wht_amount = total_with_vat * WITHHOLD_RATE
-            
-            result["ราคาเริ่มต้น (Base)"] = base
-            result["VAT 7%"] = vat_amount
-            result["ยอดรวม (รวม VAT)"] = total_with_vat
-            result["หัก ณ ที่จ่าย"] = wht_amount
-            result["ยอดสุทธิที่ได้รับ (Net)"] = net
+def update_outputs(**kwargs):
+    results = calculate_tax(**kwargs)
+    outputs = {
+        "base": results.get("base", None),
+        "after_vat": results.get("after_vat", None),
+        "after_vat_withholding": results.get("after_vat_withholding", None),
+        "after_withholding": results.get("after_withholding", None),
+        "vat": results.get("vat", None),
+        "withholding": results.get("withholding", None),
+        "error_msg": results.get("error", "")
+    }
+    debug_print(outputs, "Final Outputs")
+    return outputs
 
-    # จัดรูปแบบเลขให้อ่านง่ายขึ้น (เช่น ปัดทศนิยม 2 ตำแหน่ง)
-    for k, v in result.items():
-        result[k] = f"{v:,.2f}"
+with gr.Blocks(title="Tax Calculator Debug", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 🧮 เครื่องคิดเลขภาษี (Debug Mode)")
     
-    # สร้างข้อความสรุป หรือจะส่งออกเป็น dict เพื่อให้ Gradio แสดงก็ได้
-    output_text = (
-        f"ราคาเริ่มต้น (Base): {result['ราคาเริ่มต้น (Base)']} บาท\n"
-        f"VAT 7%: {result['VAT 7%']} บาท\n"
-        f"ยอดรวม (รวม VAT): {result['ยอดรวม (รวม VAT)']} บาท\n"
-        f"หัก ณ ที่จ่าย: {result['หัก ณ ที่จ่าย']} บาท\n"
-        f"ยอดสุทธิที่ได้รับ (Net): {result['ยอดสุทธิที่ได้รับ (Net)']} บาท\n"
-    )
+    with gr.Row():
+        base = gr.Number(label="1. เงินต้น (ก่อน VAT)", precision=2)
+        after_vat = gr.Number(label="2. ยอดรวม VAT 7%", precision=2)
+        after_vat_withholding = gr.Number(label="3. ยอดสุทธิ (หลัง VAT และหัก ณ ที่จ่าย)", precision=2)
+        after_withholding = gr.Number(label="4. ยอดหลังหัก ณ ที่จ่าย (ไม่มี VAT)", precision=2)
     
-    return output_text
-
-
-# -------------------------------------------------------
-# สร้าง UI ด้วย Gradio
-# -------------------------------------------------------
-with gr.Blocks() as demo:
-    gr.Markdown(
-        """
-        # โปรแกรมคำนวณราคา (VAT 7% และ หัก ณ ที่จ่าย 3%)
-        เลือกเงื่อนไขและกรอกตัวเลขเพื่อดูผลการคำนวณได้ทั้งแบบ "เดินหน้า" (รู้ Base -> หา Net)
-        หรือ "ถอยหลัง" (รู้ Net -> หา Base) และยังเลือกวิธีการคำนวณหัก ณ ที่จ่ายได้
-        ว่าจะหักจาก "Base" หรือจาก "Total (Base+VAT)"
-        """
-    )
+    calculate_btn = gr.Button("คำนวณ", variant="primary")
+    reset_btn = gr.Button("รีเซ็ต")
     
-    with gr.Box():
-        calculation_type = gr.Radio(
-            label="ประเภทการคำนวณ",
-            choices=["Forward (รู้ Base -> หา Net)", "Reverse (รู้ Net -> หา Base)"],
-            value="Forward (รู้ Base -> หา Net)"
-        )
-        
-        wht_type = gr.Radio(
-            label="วิธีการหัก ณ ที่จ่าย",
-            choices=["WHT จากฐาน (Base)", "WHT จากยอดรวม (Total)"],
-            value="WHT จากฐาน (Base)"
-        )
-        
-        input_amount = gr.Number(
-            label="กรอกจำนวนเงิน (บาท)",
-            value=6990.00,
-            precision=2
-        )
-        
-        calc_button = gr.Button("คำนวณ")
-        
-    output_area = gr.Textbox(
-        label="ผลลัพธ์การคำนวณ",
-        lines=6
+    vat = gr.Number(label="ภาษีมูลค่าเพิ่ม (7%)", interactive=False, precision=2)
+    withholding = gr.Number(label="ภาษีหัก ณ ที่จ่าย (3%)", interactive=False, precision=2)
+    error_msg = gr.Textbox(label="สถานะ", interactive=False)
+    
+    calculate_btn.click(
+        update_outputs,
+        inputs=[base, after_vat, after_vat_withholding, after_withholding],
+        outputs=[base, after_vat, after_vat_withholding, after_withholding, vat, withholding, error_msg]
     )
     
-    calc_button.click(
-        fn=calculate_price,
-        inputs=[calculation_type, wht_type, input_amount],
-        outputs=output_area
+    reset_btn.click(
+        lambda: [None]*4 + [None]*2 + [""],
+        outputs=[base, after_vat, after_vat_withholding, after_withholding, vat, withholding, error_msg]
     )
 
-demo.launch()
+demo.launch()