Update app.py

app.py

@@ -31,12 +31,11 @@ def extract_main_topic(text: str) -> str:
         str: The extracted main topic as keywords.
     """
     topics, probs = topic_model.transform([text])
-    topic_id = topics[0]
 
     if topic_id == -1:  # BERTopic assigns -1 to noise (no topic detected)
         return "No clear topic detected."
 
-    topic_words = topic_model.get_topic(topic_id)
+    topic_words = topic_model.get_topic(topics[0])
     
     if topic_words:
         keywords = ", ".join([word for word, _ in topic_words])
@@ -61,7 +60,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_custom_tool], 
+    tools=[final_answer, my_custom_tool, extract_main_topic], 
     max_steps=6,
     verbosity_level=1,
     grammar=None,