Update app.py

added humerous jedi mind trick tool

app.py

@@ -99,7 +99,31 @@ def execute_bash_command(command: str) -> Tuple[str, str]:
         return ("", f"Error: Command timed out after 10 seconds.")
     except Exception as e:
         return ("", f"An unexpected error occurred: {e}")
+
+@tool
+def jedi_mind_trick(question: str) -> str:
+    # it's import to specify the return type
+    """A powerful tool that uses the classic Jedi Mind Trick phrase to dismiss queries about droids.
+    
+     Args:
+        question: The user's full question or query, which should be about droids (or C3PO/R2D2).
+    """
     
+    # Convert the question to lowercase for case-insensitive matching
+    lower_question = question.lower()
+    
+    # Keywords to trigger the mind trick
+    # Uses regex word boundaries (\b) for 'droid' to avoid matching 'android', 
+    # but allows exact matching for character names.
+    droid_keywords = re.compile(r'\b(droid|droids)\b|c3po|r2d2')
+    
+    if droid_keywords.search(lower_question):
+        # The trick works!
+        return "These aren't the droids you are looking for"
+    else:
+        # For non-droid questions, return a neutral response
+        return "You don't need to use the Force for this query."
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -121,7 +145,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_os_info, DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, get_os_info, jedi_mind_trick, DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,