def calculator(num1, num2, operation):
  if operation == "add":
    return num1 + num2
  elif operation == "subtract":
    return num1 - num2
  elif operation == "multiply":
    return num1 * num2
  elif operation == "divide":
    if num2 == 0:
      return "Division by zero!"
    return num1 / num2

# prompt: using gradio to create two numbers input side by side. below it have 4 buttons for add, subtract, multiply, and dive, add title Calculator

import gradio as gr

with gr.Blocks() as calculator_app:
    gr.Markdown("## Calculator")
    with gr.Row():
        num1 = gr.Number(label="Number 1")
        num2 = gr.Number(label="Number 2")
    with gr.Row():
        add_btn = gr.Button("Add")
        subtract_btn = gr.Button("Subtract")
        multiply_btn = gr.Button("Multiply")
        divide_btn = gr.Button("Divide")
    output = gr.Textbox(label="Output")

    add_btn.click(calculator, inputs=[num1, num2, gr.Textbox(value="add", visible=False)], outputs=output)
    subtract_btn.click(calculator, inputs=[num1, num2, gr.Textbox(value="subtract", visible=False)], outputs=output)
    multiply_btn.click(calculator, inputs=[num1, num2, gr.Textbox(value="multiply", visible=False)], outputs=output)
    divide_btn.click(calculator, inputs=[num1, num2, gr.Textbox(value="divide", visible=False)], outputs=output)

calculator_app.launch()