Update handler.py

handler.py

@@ -37,10 +37,15 @@ class EndpointHandler:
             for topic, prob in zip(topics, probabilities):
                 topic_info = self.topic_model.get_topic(topic)
                 topic_words = [word for word, _ in topic_info] if topic_info else []
+
+                # Get custom label for the topic
+                custom_label = self.topic_model.custom_labels_[topic + 1] if hasattr(self.topic_model, "custom_labels_") else f"Topic {topic}"
+
                 results.append({
                     "topic": int(topic),
                     "probability": float(prob),
-                    "top_words": topic_words[:5]  # Top 5 words
+                    "top_words": topic_words[:5],  # Top 5 words
+                    "customLabel": custom_label  # Add custom label
                 })
 
             return results