Update app.py

app.py

@@ -8,47 +8,37 @@ from bertopic import BERTopic
 
 from Gradio_UI import GradioUI
 
-
 @tool
-def my_custom_tool(text:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    try:
-        # Load the pre-trained BERTopic model
-        topic_model = BERTopic.load("MaartenGr/BERTopic_ArXiv")
-        
-        # Generate topics and their probabilities
-        topics, _ = topic_model.transform(text)
+def my_custom_tool(text: str, arg2: int) -> str:
+    """A tool that processes input text and returns a modified version.
 
-        topic_words = topic_model.get_topic(topics[0])
-        
-        # Retrieve the actual topic words
-        topic_words = [word[0] for word in topic_words ]
-        topic_words  = ' '.join(topic_words)
+    Args:
+        text (str): The input text to process.
+        arg2 (int): An integer parameter for additional processing.
 
-    except Exception as e:
-        return f"Error fetching the topic of the text"
-    
-    return "Here is the topic of the text: ", topic_words
+    Returns:
+        str: The processed text output.
+    """
+    return f"Processed text: {text}"
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
+def extract_main_topic(text: str) -> str:
+    """Extracts the main topic from a given text using BERTopic.
+
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        text (str): The input text to analyze.
+
+    Returns:
+        str: The extracted main topic as keywords.
     """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+    topics, probs = topic_model.transform([text])
+    topic_id = topics[0]
+    topic_words = topic_model.get_topic(topic_id)
+
+    if topic_words:
+        return f"Main Topic: {topic_words}"
+    else:
+        return "No topic detected."
 
 
 final_answer = FinalAnswerTool()