import gradio as gr

def add(a, b):
    return a + b

def subtract(a, b):
    return a - b

def multiply(a, b):
    return a * b

def divide(a, b):
    if b == 0:
        return "Error! Division by zero."
    return a / b

def calculate(choice, num1, num2):
    if choice == 'Add':
        return add(num1, num2)
    elif choice == 'Subtract':
        return subtract(num1, num2)
    elif choice == 'Multiply':
        return multiply(num1, num2)
    elif choice == 'Divide':
        return divide(num1, num2)
    else:
        return "Invalid input"

demo = gr.Interface(
    fn=calculate,
    inputs=[
        gr.Radio(
            choices=['Add', 'Subtract', 'Multiply', 'Divide'],
            type="value",
            label="Select operation"
        ),
        gr.Number(label="Enter first number"),
        gr.Number(label="Enter second number")
    ],
    outputs=gr.Textbox(label="Result"),
    title="Simple Calculator",
    description="A simple calculator that performs addition, subtraction, multiplication, and division"
)

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