Update app.py

app.py

@@ -3,6 +3,7 @@ import subprocess
 import tempfile
 import shutil
 import traceback
+import importlib.util
 
 class CodeExecutor:
     def __init__(self):
@@ -11,17 +12,12 @@ class CodeExecutor:
     def _install_packages(self, packages):
         for package in packages:
             try:
-                subprocess.check_call(["pip", "install", package], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+                spec = importlib.util.find_spec(package)
+                if spec is None:
+                    subprocess.check_call(["pip", "install", package], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
             except subprocess.CalledProcessError as e:
                 raise Exception(f"Error installing package {package}: {e}")
 
-    def _uninstall_packages(self, packages):
-        for package in packages:
-            try:
-                subprocess.check_call(["pip", "uninstall", "-y", package], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-            except subprocess.CalledProcessError as e:
-                raise Exception(f"Error uninstalling package {package}: {e}")
-
     def _execute_code(self, code, inputs):
         temp_file = f"{self.temp_dir.name}/temp.py"
         with open(temp_file, "w") as f:
@@ -54,12 +50,9 @@ class CodeExecutor:
                 packages = []
             else:
                 packages = [p.strip() for p in packages.split(",")]
-            if packages:
-                self._install_packages(packages)
+            self._install_packages(packages)
             input_values = [i.strip() for i in inputs.split(",")] if inputs else []
             output = self._execute_code(code, input_values)
-            if packages:
-                self._uninstall_packages(packages)
             return output
         except Exception as e:
             return f"Error: {str(e)}"