Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+from IPython.display import Audio
 from Gradio_UI import GradioUI
 
 # # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -18,6 +18,31 @@ from Gradio_UI import GradioUI
 #     """
 #     return "What magic will you build ?"
 
+leak_warning_img = "leak_warning.png"
+safe_status_img = "safe_status.png"
+alarm_sound_file = "alarm_sound.wav"
+
+@tool
+def leakage_alarm_checker(current_leakage_level: float, safe_threshold: float):
+    """
+    Checks if the leakage level has exceeded a safe threshold and triggers an alarm.
+    
+    Args:
+        current_leakage_level: The measured leakage level.
+        safe_threshold: The maximum allowable leakage before triggering an alert.
+    """
+    try:
+        dif_leakage = current_leakage_level - safe_threshold
+        if dif_leakage > 0:  
+            alert_message = f"🚨 ALERT: Leakage level is {current_leakage_level} (Threshold: {safe_threshold}). IMMEDIATE ACTION REQUIRED!"
+            warning_image = Image.open(leak_warning_img)
+            return alert_message, warning_image, Audio(alarm_sound_file, autoplay=True)
+        else:
+            alert_message = f"✅ SAFE: Leakage level is {current_leakage_level}, within the safe limit of {safe_threshold}. System is operating normally."
+            safe_image = Image.open(safe_status_img)
+            return alert_message, safe_image, None
+    except Exception as e:
+        return f"Error processing leakage levels: {str(e)}", None, None
 
 @tool
 def alarm_comparator_degrees(weather_average_degrees:float, optimal_fermentation_degrees:float)-> str: #it's import to specify the return type
@@ -138,7 +163,8 @@ agent = CodeAgent(
         get_current_time_in_timezone,  # Your custom time tool
         daily_gold_oil_updates,
         daily_weather_search,
-        alarm_comparator_degrees
+        alarm_comparator_degrees,
+        leakage_alarm_checker
     ],
     max_steps=6,
     verbosity_level=1,