Logic Fix

app.py

@@ -92,6 +92,9 @@ def analyze_trend(latest_row):
     }
 
 
+# ===============================
+# Endpoint: /analyze (grafik & data)
+# ===============================
 # ===============================
 # Endpoint: /analyze (grafik & data)
 # ===============================
@@ -101,16 +104,20 @@ def analyze_ema(input_data: DateRange):
         start_date = pd.to_datetime(input_data.start_date)
         end_date = pd.to_datetime(input_data.end_date)
 
-        if end_date <= start_date:
-            return {"status": "error", "message": "Tanggal akhir harus lebih besar dari tanggal awal"}
+        # Tambahkan 60 hari ke belakang agar cukup untuk EMA50
+        extended_start = start_date - timedelta(days=60)
 
-        df = yf.download(PAIR, start=start_date, end=end_date + timedelta(days=1))
+        df = yf.download(PAIR, start=extended_start, end=end_date + timedelta(days=1))
         if df.empty:
             return {"status": "error", "message": "Data tidak ditemukan untuk rentang tanggal tersebut"}
 
         df = df.reset_index()[["Date", "Close"]]
         df.rename(columns={"Date": "date", "Close": "close"}, inplace=True)
 
+        # Pastikan minimal 50 data
+        if len(df) < 50:
+            return {"status": "error", "message": f"Data terlalu sedikit ({len(df)} hari). Butuh minimal 50 hari untuk EMA50."}
+
         df["EMA20"] = ema_manual(df["close"], 20)
         df["EMA50"] = ema_manual(df["close"], 50)
         df = df.dropna().reset_index(drop=True)
@@ -147,12 +154,20 @@ def analyze_ema(input_data: DateRange):
 @app.post("/summary")
 def ema_summary(input_data: DateRange):
     try:
-        df = yf.download(PAIR, start=input_data.start_date, end=input_data.end_date)
+        start_date = pd.to_datetime(input_data.start_date)
+        end_date = pd.to_datetime(input_data.end_date)
+        extended_start = start_date - timedelta(days=60)
+
+        df = yf.download(PAIR, start=extended_start, end=end_date + timedelta(days=1))
         if df.empty:
             return {"status": "error", "message": "Data tidak ditemukan"}
 
         df = df.reset_index()[["Date", "Close"]]
         df.rename(columns={"Date": "date", "Close": "close"}, inplace=True)
+
+        if len(df) < 50:
+            return {"status": "error", "message": f"Data terlalu sedikit ({len(df)} hari). Butuh minimal 50 hari untuk EMA50."}
+
         df["EMA20"] = ema_manual(df["close"], 20)
         df["EMA50"] = ema_manual(df["close"], 50)
         df = df.dropna().reset_index(drop=True)
@@ -172,8 +187,3 @@ def ema_summary(input_data: DateRange):
 
     except Exception as e:
         return {"status": "error", "message": str(e)}
-
-
-@app.get("/")
-def root():
-    return {"message": "Model B API (EMA + Trend Summary) aktif 🚀"}