from flask import Flask, request, jsonify

app = Flask(__name__)

@app.route('/')
def home():
    return "Hello from Flask API on Hugging Face Spaces!"

@app.route('/predict', methods=['POST'])
def predict():
    data = request.json
    # Replace this with your custom logic
    response = {
        "message": "Your input was processed successfully!",
        "input_data": data
    }
    return jsonify(response)

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