from flask import Flask, request, jsonify, render_template
from transformers import pipeline

app = Flask(__name__)

# تحميل نموذج توليد النصوص
generator = pipeline("text-generation", model="gpt2")  # يمكنك تعديل اسم النموذج حسب احتياجك

@app.route("/")
def index():
    return render_template("index.html")

@app.route("/generate-response", methods=["POST"])
def generate_response():
    data = request.get_json()
    user_input = data.get("text", "")

    # توليد استجابة من النموذج
    result = generator(user_input, max_length=100, num_return_sequences=1)
    response = result[0]["generated_text"]

    return jsonify({"response": response})

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