app.py

from flask import Flask, request, jsonify
from flask_cors import CORS
import pandas as pd
import numpy as np
import joblib
import calendar

app = Flask(__name__)
CORS(app)

# Load model and features
model = joblib.load("model.pkl")
features = joblib.load("features.pkl")
df = pd.read_csv("electricity_demand_dataset.csv")

default_vals = {
    "Temperature_C": 28,
    "CDD65": 5,
    "Humidity_Percent": 70,
    "DISCO_IBEDC": 1,
    "DISCO_IE": 0
}

@app.route("/")
def home():
    return "🔌 Electricity Demand Forecast API"

@app.route("/predict", methods=["POST"])
def predict():
    """Single month prediction - existing endpoint"""
    data = request.get_json()
    try:
        year = int(data.get("year"))
        month = int(data.get("month"))
        disco = data.get("disco", "IBEDC").upper()

        month_sin = np.sin(2 * np.pi * month / 12)
        month_cos = np.cos(2 * np.pi * month / 12)

        input_data = {
            "Year": year,
            "Month_sin": month_sin,
            "Month_cos": month_cos,
            "Temperature_C": default_vals["Temperature_C"],
            "CDD65": default_vals["CDD65"],
            "Humidity_Percent": default_vals["Humidity_Percent"],
            "DISCO_IBEDC": 1 if disco == "IBEDC" else 0,
            "DISCO_IE": 1 if disco == "IE" else 0,
        }

        df_input = pd.DataFrame([input_data])
        df_input = df_input.reindex(columns=features, fill_value=0)

        prediction = model.predict(df_input)[0]

        # Find actual consumption for same year/month/disco
        actual_row = df[
            (df["Year"] == year) &
            (df["Month"] == month) &
            (df["DISCO"].str.upper() == disco)
        ]

        actual_value = actual_row["Consumption_MWh"].values[0] if not actual_row.empty else None

        return jsonify({
            "forecast": round(prediction, 2),
            "actual": round(actual_value, 2) if actual_value is not None else None
        })

    except Exception as e:
        return jsonify({"error": str(e)}), 400

@app.route("/forecast", methods=["POST"])
def forecast_year():
    """Generate forecast for entire year - new endpoint for chart"""
    data = request.get_json()
    try:
        year = int(data.get("year"))
        disco = data.get("disco", "IBEDC").upper()
        
        results = []
        
        for month in range(1, 13):
            month_sin = np.sin(2 * np.pi * month / 12)
            month_cos = np.cos(2 * np.pi * month / 12)

            input_data = {
                "Year": year,
                "Month_sin": month_sin,
                "Month_cos": month_cos,
                "Temperature_C": default_vals["Temperature_C"],
                "CDD65": default_vals["CDD65"],
                "Humidity_Percent": default_vals["Humidity_Percent"],
                "DISCO_IBEDC": 1 if disco == "IBEDC" else 0,
                "DISCO_IE": 1 if disco == "IE" else 0,
            }

            df_input = pd.DataFrame([input_data])
            df_input = df_input.reindex(columns=features, fill_value=0)

            prediction = model.predict(df_input)[0]

            # Find actual consumption for same year/month/disco
            actual_row = df[
                (df["Year"] == year) &
                (df["Month"] == month) &
                (df["DISCO"].str.upper() == disco)
            ]

            actual_value = actual_row["Consumption_MWh"].values[0] if not actual_row.empty else None

            results.append({
                "month": calendar.month_abbr[month],  # Jan, Feb, Mar, etc.
                "forecast": round(prediction, 2),
                "actual": round(actual_value, 2) if actual_value is not None else None
            })

        return jsonify({
            "year": year,
            "disco": disco,
            "data": results
        })

    except Exception as e:
        return jsonify({"error": str(e)}), 400

@app.route("/forecast-range", methods=["POST"])
def forecast_range():
    """Generate forecast for a range of months - flexible endpoint"""
    data = request.get_json()
    try:
        start_year = int(data.get("start_year"))
        end_year = int(data.get("end_year", start_year))
        start_month = int(data.get("start_month", 1))
        end_month = int(data.get("end_month", 12))
        disco = data.get("disco", "IBEDC").upper()
        
        results = []
        
        for year in range(start_year, end_year + 1):
            month_start = start_month if year == start_year else 1
            month_end = end_month if year == end_year else 12
            
            for month in range(month_start, month_end + 1):
                month_sin = np.sin(2 * np.pi * month / 12)
                month_cos = np.cos(2 * np.pi * month / 12)

                input_data = {
                    "Year": year,
                    "Month_sin": month_sin,
                    "Month_cos": month_cos,
                    "Temperature_C": default_vals["Temperature_C"],
                    "CDD65": default_vals["CDD65"],
                    "Humidity_Percent": default_vals["Humidity_Percent"],
                    "DISCO_IBEDC": 1 if disco == "IBEDC" else 0,
                    "DISCO_IE": 1 if disco == "IE" else 0,
                }

                df_input = pd.DataFrame([input_data])
                df_input = df_input.reindex(columns=features, fill_value=0)

                prediction = model.predict(df_input)[0]

                # Find actual consumption
                actual_row = df[
                    (df["Year"] == year) &
                    (df["Month"] == month) &
                    (df["DISCO"].str.upper() == disco)
                ]

                actual_value = actual_row["Consumption_MWh"].values[0] if not actual_row.empty else None

                # Create month label (e.g., "Jan 2024" for multi-year ranges)
                month_label = calendar.month_abbr[month]
                if end_year > start_year:
                    month_label += f" {year}"

                results.append({
                    "month": month_label,
                    "forecast": round(prediction, 2),
                    "actual": round(actual_value, 2) if actual_value is not None else None
                })

        return jsonify({
            "start_year": start_year,
            "end_year": end_year,
            "disco": disco,
            "data": results
        })

    except Exception as e:
        return jsonify({"error": str(e)}), 400
    
@app.route("/api/login", methods=["POST"])
def login():
    data = request.get_json()
    email = data.get("email")
    password = data.get("password")

    if email == "admin@example.com" and password == "admin123":
        return jsonify({"email": email}), 200
    else:
        return jsonify({"message": "Invalid credentials"}), 401


if __name__ == "__main__":
    app.run(debug=True)