Update app.py

app.py

@@ -2,23 +2,37 @@ import gradio as gr
 from flake8.api import legacy as flake8
 import autopep8
 
-def lint_code(code):
-    # Run Flake8 for linting
-    style_guide = flake8.get_style_guide(ignore=["E501"])  # Ignore line-length violations
+def lint_code(code: str) -> tuple[str, str]:
+    """Lint the provided Python code and return the fixed code with linting report."""
+    # Run Flake8 for linting and ignore line-length violations
+    style_guide = flake8.get_style_guide(ignore=["E501"])
     report = style_guide.check_files(None, lines=code.splitlines())
     
-    # Auto-fix with autopep8
+    # Auto-fix the code with autopep8
     fixed_code = autopep8.fix_code(code)
-    return fixed_code, "\n".join(msg for msg in report.get_statistics())
+    
+    # Return fixed code and linting report as a tuple
+    lint_report = "\n".join(msg for msg in report.get_statistics())
+    return fixed_code, lint_report
 
-def process_file(file):
-    # Read file contents
+def process_file(file) -> tuple[str, str]:
+    """Process the uploaded file and return the fixed code and linting report."""
+    # Read the file content and decode it to a string
     code = file.read().decode("utf-8")
-    linted_code, report = lint_code(code)
-    return linted_code, report
+    
+    # Lint the code and get the report
+    return lint_code(code)
+
+def handle_input(code: str, file) -> tuple[str, str]:
+    """Handle both code inputs (text and file) and return fixed code and linting report."""
+    # Process file if provided, else process the pasted code
+    if file:
+        return process_file(file)
+    return lint_code(code)
 
+# Create the Gradio interface
 interface = gr.Interface(
-    fn=lint_code,
+    fn=handle_input,
     inputs=[
         gr.Textbox(lines=20, placeholder="Paste your Python code here...", label="Python Code"),
         gr.File(label="Upload a Python (.py) file", file_types=[".py"])
@@ -31,4 +45,5 @@ interface = gr.Interface(
     description="Paste your Python code or upload a .py file to get a Flake8-compliant, linted version."
 )
 
-interface.launch()
+# Launch the interface
+interface.launch()