Update app.py

app.py

@@ -7,7 +7,7 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 import os
 api_key = os.getenv("OPENAI_API_KEY")
-
+momo_key = os.getenv("MomoKey")
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: 
@@ -70,11 +70,31 @@ def calculate_expenses(sallery: int) -> str:
 
 @tool
 def encode_text(text: str) -> str:
-    """A tool that encodes a given text using the sabonzo method.
+    """Encodes text using XOR with a key and Base64 encoding.
+    
     Args:
-        text: The text to be encoded.
+        text: The text to encode.
+    
+    Returns:
+        Encoded Base64 string.
+    """
+    key_cycle = (ord(k) for k in momo_key)  # Cycle through key chars
+    encoded_bytes = bytes([ord(c) ^ next(key_cycle) for c in text])  # XOR each char
+    return base64.b64encode(encoded_bytes).decode()
+
+@tool
+def decode_text(encoded_text: str) -> str:
+    """Decodes a Base64-encoded XOR encrypted text using the given key.
+    
+    Args:
+        encoded_text: The encoded Base64 string.
+    
+    Returns:
+        Decoded plain text.
     """
-    return ''.join(chr(ord(char) + 2) for char in text)
+    key_cycle = (ord(k) for k in momo_key)
+    decoded_bytes = base64.b64decode(encoded_text)  # Decode Base64
+    return ''.join(chr(b ^ next(key_cycle)) for b in decoded_bytes)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -117,7 +137,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, calculate_expenses, image_generation_tool, encode_text, get_sura, search_quran],
+    tools=[final_answer, calculate_expenses, image_generation_tool, encode_text, decode_text, get_sura, search_quran],
     max_steps=6,
     verbosity_level=1,
     grammar=None,