Update app.py

app.py

@@ -23,12 +23,12 @@ from Gradio_UI import GradioUI
 
 @tool
 def get_stock_information(stock:str) -> str:
-    ''' this tool provide company information for the input stock ticker 
+    """ this tool provide company information for the input stock ticker 
         Args:
             stock: A string representing a valid str (e.g., 'googl').
         Returns:
             str
-    '''
+    """
     try:
         data=yf.Ticker(stock)
         info1=data.info
@@ -36,34 +36,31 @@ def get_stock_information(stock:str) -> str:
     except Exception as e:
         return f"Error fetching stock info '{stock}': {str(e)}"
         
-#@tool
-#def get_radio_station(input_cr : str) -> list:
-#    ''' this function fetches streams to radio stations for input_cr, it print str urls addresses
-#         Args: 
-#             input_cr (str) :A string representing a valid input_cr (e.g., 'us')
-#         Return: list       
+@tool
+def get_radio_station(input_cr : str) -> list:
+    """ this function fetches streams to radio stations for input_cr, it print str urls addresses
+        Args: 
+             input_cr (str) :A string representing a valid input_cr (e.g., 'us')
+    """
+    try:
+        conn = http.client.HTTPSConnection("50k-radio-stations.p.rapidapi.com")
+
+        headers = {
+            "x-rapidapi-key": "edb18a5d52msh741fb6d677c3be2p1c3c92jsn41ac9a131361",
+            "x-rapidapi-host": "50k-radio-stations.p.rapidapi.com",
+        }
 
-#         e.g. :get_radio_station('us')
-#    '''
-#    try:
-#        conn = http.client.HTTPSConnection("50k-radio-stations.p.rapidapi.com")
-#
-#        headers = {
-#            "x-rapidapi-key": "edb18a5d52msh741fb6d677c3be2p1c3c92jsn41ac9a131361",
-#            "x-rapidapi-host": "50k-radio-stations.p.rapidapi.com",
-#        }
-#
-#        conn.request("GET", f"/radios/popular/{input_cr}?limit=6", headers=headers)
-#        res = conn.getresponse()
-#        dt = json.loads(res.read().decode("utf-8"))
-#
-#        return [
-#            station["streams"][0]["url"]
-#            for station in dt.get("data", [])
-#            if station.get("streams")
-#        ]
-#    except Exception as e:
-#        return [f"Error: {str(e)}"]
+        conn.request("GET", f"/radios/popular/{input_cr}?limit=6", headers=headers)
+        res = conn.getresponse()
+        dt = json.loads(res.read().decode("utf-8"))
+
+        return [
+            station["streams"][0]["url"]
+            for station in dt.get("data", [])
+            if station.get("streams")
+        ]
+    except Exception as e:
+        return [f"Error: {str(e)}"]
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -102,7 +99,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,get_current_time_in_timezone,get_stock_information], ## add your tools here (don't remove final answer)
+    tools=[final_answer,get_current_time_in_timezone,get_stock_information,get_radio_station], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,