import gradio as gr
from transformers import pipeline
calculator = pipeline("zero-shot-classification")

def calculate_expression(expression):
    result = calculator(expression, candidate_labels=["addition", "subtraction", "multiplication", "division"])
    if result["labels"][0] == "addition":
        return eval(expression)
    elif result["labels"][0] == "subtraction":
        return eval(expression)
    elif result["labels"][0] == "multiplication":
        return eval(expression)
    elif result["labels"][0] == "division":
        return eval(expression)