Update setup.sh

setup.sh

@@ -1,15 +1,29 @@
-echo "dtoverlay=dwc2,dr_mode=peripheral" >> /boot/config.txt
+#!/bin/bash
+
+# Enable USB gadget mode
+# Fix or add dwc2 overlay
+if grep -q "^dtoverlay=dwc2" /boot/config.txt; then
+    sudo sed -i 's/^dtoverlay=dwc2.*/dtoverlay=dwc2,dr_mode=peripheral/' /boot/config.txt
+else
+    echo "dtoverlay=dwc2,dr_mode=peripheral" | sudo tee -a /boot/config.txt
+fi
+
+# Load required kernel modules
 echo dwc2 | sudo tee /etc/modules-load.d/dwc2.conf
 echo libcomposite | sudo tee /etc/modules-load.d/libcomposite.conf
 
-cp ./hid-setup.sh /usr/local/bin/hid-setup.sh
-cp ./hidg.service /etc/systemd/system/hidg.service
-
-chmod +x /usr/local/bin/hid-setup.sh
+# Install HID gadget script and service
+sudo cp ./hid-setup.sh /usr/local/bin/hid-setup.sh
+sudo cp ./hidg.service /etc/systemd/system/hidg.service
+sudo chmod +x /usr/local/bin/hid-setup.sh
 
+# Enable systemd service
+sudo systemctl daemon-reload
 sudo systemctl enable hidg.service
 
-apt install python3 python3-pip git -y
+# Install Python + venv
+sudo apt install python3 python3-pip python3-venv git -y
 
+# Create virtual environment and install web UI dependencies
 python3 -m venv ~/rpikvm
-~/rpikvm/bin/pip install flask evdev
+~/rpikvm/bin/pip install flask evdev