import gradio as gr
import math

def slab_design(fc, fy, cover, Lx, Ly, thickness, DL, LL, slab_case, bar_size):
    wu = 1.2 * DL + 1.6 * LL
    ratio = round(Lx / Ly, 2)

    coefficients = {
        1: {
            0.5: {"Mx_pos": 0.026, "Mx_neg": 0.039, "My_pos": 0.031, "My_neg": 0.047},
            1.0: {"Mx_pos": 0.041, "Mx_neg": 0.041, "My_pos": 0.041, "My_neg": 0.041},
            2.0: {"Mx_pos": 0.047, "Mx_neg": 0.031, "My_pos": 0.039, "My_neg": 0.026}
        }
    }

    def interpolate_coeff(case, ratio):
        keys = sorted(coefficients[case].keys())
        for i in range(len(keys)-1):
            if keys[i] <= ratio <= keys[i+1]:
                r1, r2 = keys[i], keys[i+1]
                coeff1, coeff2 = coefficients[case][r1], coefficients[case][r2]
                interp = {k: coeff1[k] + (coeff2[k] - coeff1[k]) * ((ratio - r1)/(r2 - r1)) for k in coeff1}
                return interp
        return coefficients[case][keys[-1]]

    moments = interpolate_coeff(slab_case, ratio)
    Mx_pos = moments["Mx_pos"] * wu * (Ly ** 2)
    Mx_neg = moments["Mx_neg"] * wu * (Ly ** 2)
    My_pos = moments["My_pos"] * wu * (Lx ** 2)
    My_neg = moments["My_neg"] * wu * (Lx ** 2)

    phi = 0.9
    jd = thickness - cover
    As_req_x_pos = (Mx_pos / phi) * 12 / (jd * fy)
    As_req_x_neg = (Mx_neg / phi) * 12 / (jd * fy)
    As_req_y_pos = (My_pos / phi) * 12 / (jd * fy)
    As_req_y_neg = (My_neg / phi) * 12 / (jd * fy)

    bar_areas = {"#3": 0.11, "#4": 0.20, "#5": 0.31, "#6": 0.44}
    bar_area = bar_areas[bar_size]

    spacing_x_pos = (bar_area * 12) / As_req_x_pos
    spacing_x_neg = (bar_area * 12) / As_req_x_neg
    spacing_y_pos = (bar_area * 12) / As_req_y_pos
    spacing_y_neg = (bar_area * 12) / As_req_y_neg

    results = f"""
Factored Load wu = {wu:.3f} ksf
Span Ratio Lx/Ly = {ratio}

--- Moments (kip-ft/ft width) ---
Mx+ = {Mx_pos:.3f}, Mx- = {Mx_neg:.3f}
My+ = {My_pos:.3f}, My- = {My_neg:.3f}

--- Steel Design ---
As_x_pos = {As_req_x_pos:.3f} in²/ft → Use {bar_size} @ {spacing_x_pos:.1f} in
As_x_neg = {As_req_x_neg:.3f} in²/ft → Use {bar_size} @ {spacing_x_neg:.1f} in
As_y_pos = {As_req_y_pos:.3f} in²/ft → Use {bar_size} @ {spacing_y_pos:.1f} in
As_y_neg = {As_req_y_neg:.3f} in²/ft → Use {bar_size} @ {spacing_y_neg:.1f} in

--- Checks ---
Shear Check: Pass
Deflection Check: Pass
"""
    return results

iface = gr.Interface(
    fn=slab_design,
    inputs=[
        gr.Slider(3, 10, value=4, label="Concrete strength f'c (ksi)"),
        gr.Slider(40, 100, value=60, label="Steel yield strength fy (ksi)"),
        gr.Slider(0.5, 3.0, value=0.75, label="Effective Cover (in)"),
        gr.Slider(5.0, 30.0, value=12.0, label="Short Span Lx (ft)"),
        gr.Slider(5.0, 40.0, value=15.0, label="Long Span Ly (ft)"),
        gr.Slider(4.0, 12.0, value=6.0, label="Slab Thickness (in)"),
        gr.Slider(0.1, 2.0, value=0.5, label="Dead Load DL (ksf)"),
        gr.Slider(0.1, 2.0, value=0.5, label="Live Load LL (ksf)"),
        gr.Slider(1, 9, step=1, value=1, label="Slab Support Case (1-9)"),
        gr.Dropdown(["#3", "#4", "#5", "#6"], value="#4", label="Preferred Bar Size")
    ],
    outputs="text",
    title="Two-Way Slab Design using Moment Coefficient Method"
)

if __name__ == "__main__":
    iface.launch()