Update interpreter.py

interpreter.py

@@ -1,14 +1,40 @@
 import io
 import sys
+import ast
 from logger import logger
 
 def run_code(code):
+    """
+    Executes user-provided Python code and captures its output.
+    Detects function definitions and provides feedback if functions are not invoked.
+
+    Parameters:
+        code (str): Python code entered by the user.
+
+    Returns:
+        str: Captured output or error messages.
+    """
     # Redirect stdout to capture code output
     old_stdout = sys.stdout
     redirected_output = sys.stdout = io.StringIO()
 
     try:
+        # Parse the code to detect function definitions
+        tree = ast.parse(code)
+        function_names = [
+            node.name for node in ast.walk(tree) if isinstance(node, ast.FunctionDef)
+        ]
+
+        # Execute the code
         exec(code)
+
+        # Check if functions are defined but not called
+        if function_names:
+            captured_output = redirected_output.getvalue()
+            captured_output += f"\n\nDefined functions: {', '.join(function_names)}\n"
+            captured_output += "Note: Functions need to be explicitly called to see their output."
+            return captured_output
+
     except Exception as e:
         logger.error(f"Execution error: {e}")
         return f"Error: {e}"
@@ -16,4 +42,5 @@ def run_code(code):
         # Reset stdout
         sys.stdout = old_stdout
 
+    # Return the captured output
     return redirected_output.getvalue()