Update model_handler.py

model_handler.py

@@ -1,11 +1,9 @@
 import numpy as np
 import torch
-# Menggunakan ChronosPipeline untuk pemuatan dan inferensi yang efisien
 from chronos import ChronosPipeline 
 
 class ModelHandler:
     def __init__(self):
-        # Mengganti model lama dengan Chronos-2 yang lebih canggih
         self.model_name = "amazon/chronos-2" 
         self.pipeline = None
         # Penentuan device: "cuda" jika ada GPU, jika tidak "cpu"
@@ -34,11 +32,12 @@ class ModelHandler:
         """Generate predictions using Chronos-2 or fallback. 'data' must be the dict from data_processor.prepare_for_chronos."""
         try:
             # Cek data: memastikan data yang masuk adalah dictionary yang valid
+            # Fix error 'original' in fallback logic by ensuring proper data check
             if data is None or not isinstance(data, dict) or 'original' not in data or len(data['original']) < 20:
                 return np.array([0] * horizon)
             
             if self.pipeline is None:
-                # --- Fallback Logic (Menggunakan data['original']) ---
+                # --- Fallback Logic ---
                 values = data['original']
                 recent_trend = np.polyfit(range(len(values[-20:])), values[-20:], 1)[0]
                 
@@ -46,9 +45,7 @@ class ModelHandler:
                 last_value = values[-1]
                 
                 for i in range(horizon):
-                    # Add trend with some noise
                     next_value = last_value + recent_trend * (i + 1)
-                    # Use .get('std', 1.0) for safety
                     noise = np.random.normal(0, data.get('std', 1.0) * 0.1)
                     predictions.append(next_value + noise)