Update app.py

app.py

@@ -8,19 +8,29 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 import ast
+import io
 import pyflakes.api
 import pyflakes.reporter
-import io
+from flake8.api import legacy as flake8
+from mccabe import get_code_complexity
 
-def detect_code_issues(code: str) -> list:
-    """Analyzes Python code for common issues using AST and Pyflakes.
+@tool
+def analyze_code_complexity(code: str) -> list:
+    """Analyzes the cyclomatic complexity of functions in the code."""
+    tree = ast.parse(code)
+    issues = []
     
-    Args:
-        code: The Python code as a string.
-
-    Returns:
-        A list of detected issues.
-    """
+    for node in ast.walk(tree):
+        if isinstance(node, ast.FunctionDef):
+            complexity = get_code_complexity(node, max_complexity=10)
+            if complexity > 10:
+                issues.append(f"High complexity: Function '{node.name}' has a cyclomatic complexity of {complexity}")
+    
+    return issues
+    
+@tool
+def detect_code_issues(code: str) -> list:
+    """Analyzes Python code for issues using AST, Pyflakes, Flake8, and McCabe."""
     issues = []
 
     # Check for syntax errors
@@ -29,17 +39,31 @@ def detect_code_issues(code: str) -> list:
     except SyntaxError as e:
         issues.append(f"SyntaxError: {e.msg} at line {e.lineno}")
 
-    # Use Pyflakes to check for other issues (e.g., unused imports, undefined variables)
+    # Use Pyflakes to detect undefined variables and unused imports
     output_stream = io.StringIO()
     reporter = pyflakes.reporter.Reporter(output_stream, output_stream)
     pyflakes.api.check(code, filename="<user_code>", reporter=reporter)
-    output = output_stream.getvalue()
+    pyflakes_output = output_stream.getvalue().strip()
+    
+    if pyflakes_output:
+        issues.extend(pyflakes_output.split("\n"))
 
-    if output:
-        issues.extend(output.strip().split("\n"))
+    # Use Flake8 for style violations (PEP8)
+    flake8_style = flake8.get_style_guide(ignore=['E501'])  # Ignore line length warnings
+    report = flake8_style.input_file(filename="<user_code>", lines=code.split("\n"))
+    if report.total_errors > 0:
+        issues.append(f"Flake8: Found {report.total_errors} style violations.")
 
-    return issues
+    # Analyze function complexity
+    issues.extend(analyze_code_complexity(code))
 
+    # Check for missing docstrings
+    tree = ast.parse(code)
+    for node in ast.walk(tree):
+        if isinstance(node, (ast.FunctionDef, ast.ClassDef)) and not ast.get_docstring(node):
+            issues.append(f"Missing docstring in {node.__class__.__name__}: '{node.name}'")
+
+    return issues
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: