Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,6 +19,8 @@ import time
 import openai
 import dotenv 
 import os 
+import site
+
 dotenv.load_dotenv()
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 
@@ -126,18 +128,19 @@ st.sidebar.subheader("📦 Install Python Library")
 with st.sidebar.form("install_lib_form"):
     lib_name = st.text_input("Library name (e.g., numpy, pandas)")
     install_btn = st.form_submit_button("Install with pip")
+
     if install_btn and lib_name.strip():
         with st.spinner(f"Installing {lib_name}..."):
             try:
-                # Define target directory for installation
-                target_dir = "/.local/lib/python3.9/site-packages"
+                # Use user site-packages (safe, writable)
+                target_dir = site.USER_SITE
                 os.makedirs(target_dir, exist_ok=True)
 
-                # Add to sys.path so imports work immediately
+                # Add target_dir to sys.path so imports work
                 if target_dir not in sys.path:
                     sys.path.insert(0, target_dir)
 
-                # Run pip install into target dir
+                # Install package
                 result = subprocess.run(
                     [
                         sys.executable, "-m", "pip", "install",
@@ -147,10 +150,21 @@ with st.sidebar.form("install_lib_form"):
                     ],
                     capture_output=True, text=True
                 )
+
                 if result.returncode == 0:
-                    st.success(f"Successfully installed `{lib_name}`.")
+                    st.success(f"✅ Successfully installed `{lib_name}`.")
+
+                    # Try importing immediately
+                    try:
+                        module = importlib.import_module(lib_name)
+                        importlib.reload(module)
+                        st.info(f"📦 `{lib_name}` is now ready to use!")
+                    except Exception as e:
+                        st.warning(f"Installed but could not import: {e}")
+
                 else:
-                    st.error(f"Error installing `{lib_name}`:\n{result.stderr}")
+                    st.error(f"❌ Error installing `{lib_name}`:\n{result.stderr}")
+
             except Exception as e:
                 st.error(f"Exception: {e}")
 st.title("🧠 Python Code Generator & Runner")