From 900b54ec0b797fde93d4fedd781b7591636586d0 Mon Sep 17 00:00:00 2001
From: zjamnik <zjamnik@gmail.com>
Date: Fri, 17 Apr 2026 22:38:45 +0000
Subject: [PATCH] Update setup.sh

---
 setup.sh | 28 +++++++++++++++++++++-------
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/setup.sh b/setup.sh
index 1d3adf0..fb57761 100644
--- a/setup.sh
+++ b/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
\ No newline at end of file
+~/rpikvm/bin/pip install flask evdev