from flask import Flask, request, jsonify
from transformers import pipeline
from flask_cors import CORS

app = Flask(__name__)
CORS(app)  # Frontend'ten gelen istekleri kabul etmek için

qa_pipeline = pipeline("question-answering", model="deepset/tinyroberta-squad2")

@app.route("/predict", methods=["POST"])
def predict():
    data = request.get_json()
    question = data.get("question", "")
    context = (
        "Diş ağrısı, diş eti hastalıkları, çürükler, diş apseleri gibi pek çok nedenden kaynaklanabilir. "
        "Düzenli diş fırçalama, diş ipi kullanımı ve hekim kontrolleri önemlidir. "
        "Ciddi ağrılarda mutlaka bir diş hekimine başvurulmalıdır."
    )
    result = qa_pipeline(question=question, context=context)
    return jsonify({"answer": result["answer"]})

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