Update app.py

app.py

@@ -9,12 +9,11 @@ from bertopic import BERTopic
 from Gradio_UI import GradioUI
 
 @tool
-def my_custom_tool(text: str, arg2: int) -> str:
+def my_custom_tool(text: str) -> str:
     """A tool that processes input text and returns a modified version.
 
     Args:
         text (str): The input text to process.
-        arg2 (int): An integer parameter for additional processing.
 
     Returns:
         str: The processed text output.
@@ -33,14 +32,18 @@ def extract_main_topic(text: str) -> str:
     """
     topics, probs = topic_model.transform([text])
     topic_id = topics[0]
-    topic_words = topic_model.get_topic(topic_id)
 
+    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)
+    
     if topic_words:
-        return f"Main Topic: {topic_words}"
+        keywords = ", ".join([word for word, _ in topic_words])
+        return f"Main Topic Keywords: {keywords}"
     else:
         return "No topic detected."
 
-
 final_answer = FinalAnswerTool()
 
 model = HfApiModel(