Update handler.py

handler.py

@@ -43,7 +43,12 @@ class EndpointHandler:
     
             # Rank topics by their contribution in descending order
             ranked_topics = sorted(enumerate(doc_topic_distribution), key=lambda x: x[1], reverse=True)[:10]
-    
+            customlabels = self.topic_model.custom_labels_
+
+            distributedtopics = [
+                [f"Topic {topic_idx}", customlabels[topic_idx + 1], f"{contribution * 100}%"] 
+                for topic_idx, contribution in ranked_topics
+            ]    
             # Prepare the results
             results = []
             for topic, prob in zip(topics, probabilities):
@@ -57,14 +62,14 @@ class EndpointHandler:
                     custom_label = f"Topic {topic}"  # Fallback label
     
                 # Get the contribution from approximate distribution
-                contribution = next((contribution for idx, contribution in ranked_topics if idx == topic), 0.0)
+                # contribution = next((contribution for idx, contribution in ranked_topics if idx == topic), 0.0)
     
                 results.append({
                     "topic": int(topic),
                     "probability": float(prob),
                     "top_words": topic_words[:5],  # Top 5 words
                     "customLabel": custom_label,  # Add custom label
-                    "contribution": ranked_topics
+                    "contribution": distributedtopics
                 })
     
             return results