endpoints/predict.py

import os
from fastapi import APIRouter, Security, Depends, HTTPException
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from pydantic import BaseModel
import jwt
import numpy as np
import tensorflow as tf
from rnn.models.deep_portfolio import DeepPortfolioAI


router = APIRouter()
security = HTTPBearer()
MODEL = DeepPortfolioAI(num_assets=10)
SECRET_KEY = os.getenv("SECRET_KEY")

class PredictionRequest(BaseModel):
    market_data: list
    news_data: list

def verify_jwt(credentials: HTTPAuthorizationCredentials = Security(security)):
    try:
        return jwt.decode(credentials.credentials, SECRET_KEY, algorithms=["HS256"])
    except:
        raise HTTPException(status_code=401, detail="Invalid token")

@router.post("/predict")
async def predict(
    request: PredictionRequest,
    user=Depends(verify_jwt)
):
    try:
        market_tensor = tf.convert_to_tensor(request.market_data, dtype=tf.float32)
        prediction = MODEL([market_tensor, request.news_data])
        return {
            "success": True,
            "prediction": prediction.numpy().tolist(),
            "timestamp": np.datetime64('now').astype(str)
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))