from flask import Flask, request, jsonify

app = Flask(__name__)

# دیکشنری برای ذخیره انتخاب‌های کاربر (به جای st.session_state)
user_selections = {}

@app.route('/update_selection', methods=['POST'])
def update_selection():
    data = request.get_json()
    key = data.get('key')
    value = data.get('value')
    user_selections[key] = value
    return jsonify({'status': 'success', 'key': key, 'value': value})

@app.route('/get_selection/<key>', methods=['GET'])
def get_selection(key):
    value = user_selections.get(key, 0)
    return jsonify({'key': key, 'value': value})

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, debug=True)