Update app.py

app.py

@@ -5,7 +5,9 @@ from langchain_community.tools import DuckDuckGoSearchRun
 from langchain.tools import tool
 from langchain_core.output_parsers import JsonOutputParser
 from langchain.agents import AgentExecutor, create_react_agent, initialize_agent, AgentType
-from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound, TranscriptNotFoundError
+import traceback # For detailed error logging
+import timeout_decorator # Optional: for execute_code timeout (pip install timeout-decorator)
 import whisper
 import chess, chess.engine
 from bs4 import BeautifulSoup
@@ -271,7 +273,7 @@ def classify_vegetables(items: list) -> str:
 
 @tool
 # Optional: Add timeout to prevent runaway code execution
-# @timeout_decorator.timeout(10, timeout_exception=TimeoutError) # Limit execution to 10 seconds
+@timeout_decorator.timeout(10, timeout_exception=TimeoutError) # Limit execution to 10 seconds
 def execute_code(code: str) -> str:
     """
     Executes a given Python code snippet and returns the value of the 'output' variable.
@@ -302,8 +304,8 @@ def execute_code(code: str) -> str:
         if output_val is None:
             return "Result: Code executed, but no variable named 'output' was assigned."
         return str(output_val)
-    # except TimeoutError:
-    #     return "Error: Code execution timed out."
+    except TimeoutError:
+        return "Error: Code execution timed out."
     except Exception as e:
         # Capture and return execution errors
         error_details = traceback.format_exc()