Update app.py

app.py

@@ -152,12 +152,10 @@ async def predict_category(transaction: TransactionInput, request: Request):
         
         # Prepare response
         response = {
-            "original_description": transaction.description,
-            "cleaned_description": cleaned_description,
-            "main_category": main_category,
-            "main_confidence": round(main_confidence, 4),
-            "sub_category": sub_category,
-            "sub_confidence": round(sub_confidence, 4)
+            "category": main_category,
+            "subcategory": sub_category,
+            "category_confidence": round(main_confidence, 4),
+            "subcategory_confidence": round(sub_confidence, 4)
         }
         logger.info(f"Response: {response}")
         return response