Update routes/recommend.py

routes/recommend.py

@@ -226,32 +226,25 @@ def recommend_movie():
 def recommend_book():
     try:
         data = request.get_json()
-
         if not data:
             return jsonify({"error": "No data provided"}), 400
-
         rec_type = data.get('type')
         genre = data.get('genre')
         top_k = data.get('top_k', 10)
-
         if not rec_type or not genre:
             return jsonify({"error": "Missing 'type' or 'genre'"}), 400
-
         if rec_type not in RECOMMENDER_ENDPOINTS or not RECOMMENDER_ENDPOINTS[rec_type]:
             return jsonify({"error": "Invalid or missing recommender URL for type."}), 400
-
         # JWT Authentication
         auth_header = request.headers.get('Authorization')
         if not auth_header or not auth_header.startswith('Bearer '):
             return jsonify({"error": "Missing or invalid Authorization header"}), 401
-
         token = auth_header.split(" ")[1]
         try:
             user_data = decode_jwt(token)
             user_id = user_data.get("user_id")
         except Exception as e:
             return jsonify({"error": "Invalid or expired token"}), 401
-
         # Call microservice
         try:
             recommender_url = RECOMMENDER_ENDPOINTS[rec_type]
@@ -260,20 +253,24 @@ def recommend_book():
                 json={"genre": genre},  # Single string as expected
                 timeout=10
             )
-
             if not response.ok:
                 return jsonify({
                     "error": f"{rec_type} service error", 
                     "details": response.text,
                     "status_code": response.status_code
                 }), 500
-
             result = response.json()
-            if result.get("status") != "success":
-                return jsonify({"error": result.get("message", "Unknown error")}), 500
-
-            raw_items = result.get(RESPONSE_KEYS.get(rec_type, "items"), [])
-
+            
+            # Handle different response formats
+            if isinstance(result, list):
+                # Microservice returns a direct list of books
+                raw_items = result
+            else:
+                # Microservice returns structured response with status
+                if result.get("status") != "success":
+                    return jsonify({"error": result.get("message", "Unknown error")}), 500
+                raw_items = result.get(RESPONSE_KEYS.get(rec_type, "items"), [])
+            
             # Normalize response format
             normalized = []
             for item in raw_items:
@@ -285,7 +282,6 @@ def recommend_book():
                     "genre": item.get("genre", []),
                     "rating": item.get("rating")
                 })
-
             # Save to history
             try:
                 history = History(
@@ -300,7 +296,6 @@ def recommend_book():
             except Exception as e:
                 print(f"Failed to save history: {e}")
                 # Don't fail the request if history saving fails
-
             return jsonify({
                 "status": "success",
                 "recommendations": normalized,
@@ -308,12 +303,10 @@ def recommend_book():
                 "type": rec_type,
                 "genres": [genre]  # ✅ Fixed: wrap single genre in list for consistency
             }), 200
-
         except requests.exceptions.Timeout:
             return jsonify({"error": f"{rec_type} service timeout"}), 504
         except requests.exceptions.RequestException as e:
             return jsonify({"error": f"Failed to connect to {rec_type} service", "details": str(e)}), 503
-
     except Exception as e:
         print(f"Recommend error: {e}")
         return jsonify({"error": "Internal server error"}), 500