Update app.py

app.py

@@ -123,13 +123,12 @@ class CodeGenerator:
 
 
 
-
 import autopep8
 import black
 import isort
 import pylint.lint
 import autoimport
-
+from yapf.yapflib.yapf_api import FormatCode  # reformat a string of code
 
 class PythonCodeFormatter:
     def __init__(self, code):
@@ -154,6 +153,9 @@ class PythonCodeFormatter:
 
             # Use autopep8 to fix any remaining issues
             formatted_code = autopep8.fix_code(formatted_code)
+
+            formatted_code, changed = FormatCode(formatted_code)
+
             return formatted_code
 
         except RuntimeError as error: