Update app.py

app.py

@@ -5,6 +5,7 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from huggingface_hub import InferenceClient
+import language_tool_python
 
 from Gradio_UI import GradioUI
 
@@ -19,6 +20,30 @@ def my_custom_tool(arg1:int, arg2:str)-> str: #it's import to specify the return
     """
     return arg2 * arg1
 
+@tool
+def correct_grammar(sentence: str) -> str:
+    """A tool that checks and corrects grammatical errors in a given sentence.
+    
+    Args:
+        sentence: A string containing the sentence to correct.
+        
+    Returns:
+        A grammatically corrected version of the input sentence.
+    """
+    try:
+        tool = language_tool_python.LanguageTool('en-US')
+        matches = tool.check(sentence)
+        corrected = language_tool_python.utils.correct(sentence, matches)
+        return corrected
+    except Exception as e:
+        return f"Error correcting sentence: {str(e)}"
+
+# Example usage
+if __name__ == "__main__":
+    test_sentence = "He go to school everyday."
+    print("Original:", test_sentence)
+    print("Corrected:", correct_grammar(test_sentence))
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.