Update app.py

app.py

@@ -3,7 +3,7 @@ import google.generativeai as genai
 import requests
 import subprocess
 import os
-import pylint.epylint as lint
+import pylint
 import pandas as pd
 from sklearn.model_selection import train_test_split
 from sklearn.ensemble import RandomForestClassifier
@@ -39,7 +39,10 @@ def generate_response(user_input):
 def optimize_code(code):
     # Placeholder for advanced code optimization logic
     # This could involve using external tools or libraries for static analysis and optimization
-    (pylint_stdout, pylint_stderr) = lint.py_run(code, return_std=True)
+    with open("temp_code.py", "w") as file:
+        file.write(code)
+    result = subprocess.run(["pylint", "temp_code.py"], capture_output=True, text=True)
+    os.remove("temp_code.py")
     return code
 
 def fetch_from_github(query):