Spaces:
Sleeping
Sleeping
| import os | |
| import uuid | |
| from datetime import datetime | |
| from flask import Flask, render_template, request, flash, url_for | |
| from models.config import Config | |
| from models.data_loader import DataManager | |
| from models.temporal_forecaster import TemporalForecaster | |
| from models.rotation_optimizer import CropRotationOptimizer | |
| from models.image_generator import CropImageGenerator | |
| def create_app(): | |
| app = Flask(__name__) | |
| app.config["SECRET_KEY"] = os.environ.get("FLASK_SECRET_KEY", "dev-secret-key") | |
| app.config["UPLOAD_FOLDER"] = os.path.join("static", "uploads") | |
| app.config["GENERATED_FOLDER"] = os.path.join("static", "generated") | |
| os.makedirs(app.config["UPLOAD_FOLDER"], exist_ok=True) | |
| os.makedirs(app.config["GENERATED_FOLDER"], exist_ok=True) | |
| data_manager = DataManager(Config) | |
| forecaster = TemporalForecaster(Config) | |
| optimizer = CropRotationOptimizer(Config) | |
| image_generator = CropImageGenerator(Config) | |
| def index(): | |
| if request.method == "POST": | |
| try: | |
| region = request.form.get("region") or "Field-001" | |
| crop = request.form.get("crop") or "Maize" | |
| seasons = int(request.form.get("seasons") or 2) | |
| scenario = request.form.get("scenario") or "baseline" | |
| start_date_str = request.form.get("start_date") or datetime.today().strftime("%Y-%m-%d") | |
| start_date = datetime.strptime(start_date_str, "%Y-%m-%d") | |
| rotation_plan_text = (request.form.get("rotation_plan") or "").strip() | |
| rotation_file = request.files.get("rotation_file") | |
| rotation_file_path = None | |
| if rotation_file and rotation_file.filename: | |
| filename = f"{uuid.uuid4().hex}_{rotation_file.filename}" | |
| rotation_file_path = os.path.join(app.config["UPLOAD_FOLDER"], filename) | |
| rotation_file.save(rotation_file_path) | |
| ndvi_data = data_manager.load_ndvi_series(region=region) | |
| weather_forecast = data_manager.load_weather_forecast(region=region) | |
| soil_terrain = data_manager.load_soil_and_terrain(region=region) | |
| commons_data = data_manager.load_ag_commons_metadata() | |
| growth_forecast = forecaster.forecast_growth( | |
| ndvi_data=ndvi_data, | |
| weather=weather_forecast, | |
| soil=soil_terrain, | |
| crop=crop, | |
| start_date=start_date, | |
| seasons=seasons, | |
| scenario=scenario, | |
| ) | |
| optimized_rotation, rl_metrics = optimizer.optimize_rotation( | |
| base_plan_text=rotation_plan_text, | |
| uploaded_plan_path=rotation_file_path, | |
| crop=crop, | |
| seasons=seasons, | |
| scenario=scenario, | |
| ) | |
| img_filename = f"crop_sim_{uuid.uuid4().hex}.png" | |
| img_path = os.path.join(app.config["GENERATED_FOLDER"], img_filename) | |
| rel_img_path = os.path.join("generated", img_filename) | |
| image_generator.generate_growth_visualization( | |
| ndvi_series=ndvi_data, | |
| forecast=growth_forecast, | |
| rotation_plan=optimized_rotation, | |
| output_path=img_path, | |
| scenario=scenario, | |
| ) | |
| metrics = { | |
| "yield_index": growth_forecast.get("yield_index", 0.0), | |
| "stress_risk": growth_forecast.get("stress_risk", 0.0), | |
| "weather_volatility": growth_forecast.get("weather_volatility", 0.0), | |
| "soil_score": growth_forecast.get("soil_score", 0.0), | |
| } | |
| return render_template( | |
| "results.html", | |
| region=region, | |
| crop=crop, | |
| seasons=seasons, | |
| scenario=scenario, | |
| start_date=start_date_str, | |
| metrics=metrics, | |
| rotation_plan=optimized_rotation, | |
| rl_metrics=rl_metrics, | |
| growth_forecast=growth_forecast, | |
| image_path=rel_img_path, | |
| commons_data=commons_data, | |
| ) | |
| except Exception as e: | |
| flash(f"Simulation failed: {e}", "danger") | |
| regions = ["Field-001", "Field-002", "Field-Highland", "Field-Lowland"] | |
| crops = ["Maize", "Wheat", "Rice", "Soybean", "Cotton"] | |
| scenarios = [ | |
| ("baseline", "Baseline"), | |
| ("wet_year", "Wet year (↑ rainfall)"), | |
| ("dry_year", "Dry year (↓ rainfall)"), | |
| ("heatwave", "Heatwave risk"), | |
| ("low_inputs", "Low fertilizer / low inputs"), | |
| ] | |
| return render_template( | |
| "index.html", | |
| regions=regions, | |
| crops=crops, | |
| scenarios=scenarios, | |
| today=datetime.today().strftime("%Y-%m-%d"), | |
| ) | |
| def health(): | |
| return {"status": "ok", "feature": "future_crop_visualization"} | |
| return app | |
| if __name__ == "__main__": | |
| app = create_app() | |
| app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 5003)), debug=True) | |