from fastapi import FastAPI
from pydantic import BaseModel
import uvicorn

app = FastAPI()

class Inputs(BaseModel):
    expression: str 

def calculate(expression):
    try:
        result = eval(expression)
        return result
    except Exception as e:
        raise Exception("ERROR")

@app.post("/calculate")
async def calculate_endpoint(request: Inputs):
    try:
        result = calculate(request.expression) 
        return {'result': result}
    except Exception as e:
        return {"error": str(e)}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port = 8000)