predictor/batch_handler.py

import pandas as pd
import requests

def get_single_prediction(record: dict) -> float:
    response = requests.post(
        "https://Pushpak21-SuperKart-Sales-Forecast-Backend.hf.space/predict",
        json=record
    )

    if response.status_code == 200:
        return response.json().get("prediction", 0.0)
    else:
        raise ValueError(f"API Error {response.status_code}: {response.text}")

def get_predictions(df: pd.DataFrame) -> pd.DataFrame:
    records = df.to_dict(orient="records")

    response = requests.post(
        "https://Pushpak21-SuperKart-Sales-Forecast-Backend.hf.space/predict_batch",
        json=records
    )

    if response.status_code == 200:
        predictions = response.json().get("predictions", [])
        df["Predicted_Sales"] = predictions
        return df
    else:
        raise ValueError(f"API Error {response.status_code}: {response.text}")