Update app.py

app.py

@@ -83,7 +83,52 @@ def extract_code_from_text(text: str):
 
     return code.strip()
 
+import re
+import subprocess
+import sys
+import importlib.util
+
+# List of common standard library modules to skip
+STD_LIBS = {
+    "os", "sys", "math", "re", "json", "time", "pathlib",
+    "subprocess", "itertools", "functools", "collections",
+    "random", "datetime", "threading", "multiprocessing",
+    "shutil", "tempfile", "logging"
+}
+
 def install_packages_from_code(code: str):
+    """
+    Scan Python code for any imports and attempt to install missing packages automatically.
+    Works for:
+      - import X
+      - import X as Y
+      - from X import Y
+      - from X.Y import Z
+    """
+    # Find all import statements
+    matches = re.findall(r'^\s*(?:import|from)\s+([a-zA-Z0-9_\.]+)', code, flags=re.MULTILINE)
+
+    # Convert to top-level package names
+    packages = set()
+    for m in matches:
+        top_pkg = m.split(".")[0]  # take top-level module
+        if top_pkg not in STD_LIBS:
+            packages.add(top_pkg)
+
+    # Install packages via pip
+    for pkg in packages:
+        # Skip if already installed
+        if importlib.util.find_spec(pkg) is not None:
+            continue
+        print(f"📦 Installing {pkg} ...")
+        try:
+            subprocess.run(
+                [sys.executable, "-m", "pip", "install", pkg],
+                check=True
+            )
+        except subprocess.CalledProcessError as e:
+            print(f"⚠️ Failed to install package {pkg}: {e}")
+def install_packages_from_code0(code: str):
     """
     Scan Python code for imports and install missing packages automatically.
     """