Update app.py

app.py

@@ -11,7 +11,6 @@ from stable_baselines3 import PPO
 app = FastAPI(title="Transaction Classifier API", description="API for classifying banking transactions.")
 print("FastAPI app initialized...")
 
-# Lazy-loaded resources
 model = None
 tokenizer = None
 le_category = None
@@ -114,7 +113,6 @@ async def predict(request: TransactionRequest):
         model, tokenizer, le_category, le_subcategory = load_resources()
         num_subcategories = len(le_subcategory.classes_)
         
-        # Fixed: Use request.description instead of description
         seq = tokenizer.texts_to_sequences([request.description])
         pad = pad_sequences(seq, maxlen=max_len, padding='post')
         pred = model.predict(pad, verbose=0)
@@ -123,8 +121,10 @@ async def predict(request: TransactionRequest):
             ppo = load_ppo_model()
             obs = pad[0]
             action, _ = ppo.predict(obs)
+            print(f"RL Action: {action}, Observation: {obs}")
             cat_idx = action // num_subcategories
             subcat_idx = action % num_subcategories
+            print(f"RL cat_idx: {cat_idx}, subcat_idx: {subcat_idx}")
         else:
             cat_probs = pred[0][0]
             subcat_probs = pred[1][0]
@@ -136,6 +136,7 @@ async def predict(request: TransactionRequest):
         cat_conf = float(pred[0][0][cat_idx] * 100)
         subcat_conf = float(pred[1][0][subcat_idx] * 100)
         
+        print(f"Predicted: category={cat_pred}, subcategory={subcat_pred}")
         return {
             "category": cat_pred,
             "subcategory": subcat_pred,