Merge pull request #269 from nyu-mlab/hot-fix

Need to explicitly add matplotlib

Author: crazyideas21

iot-inspector.nsi

@@ -94,6 +94,10 @@ Section "Uninstall"
     ; Execute the dedicated uninstall script to handle all cleanup tasks
     ExecWait 'powershell.exe -ExecutionPolicy ByPass -File "$INSTDIR\uninstall.ps1"'
 
+    ; Remove shortcuts created at install (if present)
+    Delete "$SMPROGRAMS\${APP_NAME}.lnk"
+    Delete "$DESKTOP\${APP_NAME}.lnk"
+
     ; Remove files
     RMDir /r "$INSTDIR"
 SectionEnd

pyproject.toml

@@ -11,6 +11,7 @@ requires-python = "==3.13.*"
 dependencies = [
     "libinspector==1.0.12",
     "streamlit>=1.50.0",
+    "matplotlib==3.10.3",
 ]
 authors = [
     { name = "Danny Huang", email = "[email protected]" },

start.ps1

@@ -44,22 +44,17 @@ $CurrentDir = $PSScriptRoot
 # Check if uv is available. If it's not, we assume the initial setup needs to run.
 $NpcapFound = Test-NpcapInstalled
 if (-not (Get-Command uv -ErrorAction SilentlyContinue) -or -not $NpcapFound) {
-    Write-Warning "'uv' not found. Launching initial setup script: $DependencyScript"
+    Write-Warning "Launching initial setup script: $DependencyScript"
 
     # Execute install.ps1 directly (since we are already running as Admin)
     # The dependency script handles uv installation and Npcap checks.
     & "$CurrentDir\$DependencyScript"
 
-    Write-Host "Setup finished. Re-verifying installation..."
-
     # Crucial: After UV is installed, we must start a new shell session
     # for the PATH variable (where uv is located) to be properly updated.
-    if (-not (Get-Command uv -ErrorAction SilentlyContinue)) {
-        Write-Host "UV installation successful, but the PATH variable needs refreshing."
-        Write-Host "Please close this window and run the desktop shortcut again."
-        Read-Host "Press Enter to exit and relaunch"
-        exit
-    }
+    Write-Host "Setup finished!!"
+    Read-Host "Please close this window by clicking the X and click IoT Inspector again."
+    exit
 }
 
 # --- 3. Virtual Environment Setup (Non-Admin system tasks) ---

uninstall.ps1

@@ -37,8 +37,7 @@ $filesToDelete = @(
     "setup.py",
     "uv.lock",
     "README.md",
-    "LICENSE",
-    "npcap-1.76.exe"
+    "LICENSE"
 )
 
 foreach ($file in $filesToDelete) {