from flask import Flask, request, jsonify
import joblib
import numpy as np

# Inisialisasi Flask app
app = Flask(__name__)

# Load model dari file model.pkl
model = joblib.load('model.pkl')

# Route utama hanya untuk cek apakah server hidup
@app.route('/')
def home():
    return "Model API is running!"

# Route untuk prediksi
@app.route('/predict', methods=['POST'])
def predict():
    try:
        # Ambil data dari permintaan JSON
        data = request.get_json()

        # Ambil inputan fitur dan ubah ke array 2D
        features = np.array(data['features']).reshape(1, -1)

        # Lakukan prediksi
        prediction = model.predict(features)

        # Kirim hasil ke user
        return jsonify({
            'prediction': prediction.tolist()
        })

    except Exception as e:
        return jsonify({'error': str(e)})

# Jalankan Flask server
if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7860)