Update app.py

app.py

@@ -1,28 +1,44 @@
 import gradio as gr
-import flake8
+import flake8.main.application
 import autopep8
+import tempfile
+import os
 
 def lint_code(code):
-    # Initialize Flake8 style guide
-    style_guide = flake8.get_style_guide()
-    # Create a checker for the provided code
-    checker = style_guide.check_files([code])
-    # Collect linting issues
-    issues = [f"{issue.filename}:{issue.line_number}: {issue.text}" for issue in checker.get_statistics('')]
-    # Format the code using AutoPEP8
-    formatted_code = autopep8.fix_code(code)
-    return "\n".join(issues), formatted_code
+    # Create a temporary file to store the code
+    with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.py') as temp_file:
+        temp_file.write(code)
+        temp_file_path = temp_file.name
+
+    try:
+        # Run Flake8
+        app = flake8.main.application.Application()
+        app.initialize(['--format=default', temp_file_path])
+        app.run_checks()
+        app.generate_reports()
+        flake8_output = app.results_manager.stdout.getvalue().strip()
+
+        # Format the code using AutoPEP8
+        formatted_code = autopep8.fix_code(code)
+
+        return flake8_output, formatted_code
+
+    except Exception as e:
+        return f"An error occurred: {str(e)}", code  # Return original code on error
+
+    finally:
+        # Clean up the temporary file
+        os.remove(temp_file_path)
 
 def main():
     with gr.Interface(
         fn=lint_code,
-        inputs="text",
-        outputs=["text", "text"],
+        inputs=gr.Textbox(lines=10, placeholder="Enter your Python code here..."),
+        outputs=[gr.Textbox(label="Flake8 Output"), gr.Code(label="Formatted Code", language="python")],
         title="PyLintPro",
         description="A Gradio app that helps users improve Python code to meet Flake8 and PEP 8 standards.",
-        live=True
     ) as demo:
         demo.launch()
 
 if __name__ == "__main__":
-    main()
+    main()