from fastapi import FastAPI
import requests

app = FastAPI()

# ---------- SIMPLE AGENT ----------

def solve(question: str):
    # TEMPORARY: basic logic first
    return question  # we improve this later


# ---------- API ----------
@app.get("/")
def home():
    return {"status": "agent running"}

@app.get("/predict")
def predict(question: str):
    answer = solve(question)
    return {
        "answer": answer
    }