Zsirak's picture
Update app.py
ebc2d8b verified
from flask import Flask, request, render_template
from predict_yield import predict_yield
import os
app = Flask(__name__)
@app.route('/', methods=['GET', 'POST'])
def predict():
if request.method == 'POST':
year = int(request.form['year'])
hectares = float(request.form['hectares'])
# Predict yield using the predict_yield function
result = predict_yield(year, hectares)
# Round values for display
result['predicted_yield_per_ha_kg'] = round(result['predicted_yield_per_ha_kg'], 2)
result['total_predicted_yield_kg'] = round(result['total_predicted_yield_kg'], 2)
result['confidence_interval_per_ha_lower'] = round(result['confidence_interval_per_ha_lower'], 2)
result['confidence_interval_per_ha_upper'] = round(result['confidence_interval_per_ha_upper'], 2)
result['total_confidence_interval_lower'] = round(result['total_confidence_interval_lower'], 2)
result['total_confidence_interval_upper'] = round(result['total_confidence_interval_upper'], 2)
return render_template('predict.html', result=result)
return render_template('predict.html', result=None)
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=8080)