Update app.py

app.py

@@ -1,105 +1,115 @@
-import serial
-import serial.tools.list_ports
-import time
-import streamlit as st
-import threading
-
-# Global variables to store bus information and hardware statuses
-bus_info = {
-    'bus_name': 'N/A',
-    'seat_status': 'N/A',
-    'distance': 'N/A',
-    'led_status': 'OFF',
-    'buzzer_status': 'OFF',
-}
-
-# Function to identify the correct COM port on Windows
-def get_serial_port():
-    ports = serial.tools.list_ports.comports()
-    if not ports:
-        print("No available ports found.")
-        return None
-    for port in ports:
-        print(f"Port: {port.device}, Description: {port.description}")
-    com_port = input("Enter the correct COM port (e.g., COM3): ")
-    return com_port
-
-# Open serial connection to Arduino (use the correct COM port)
-try:
-    com_port = get_serial_port()  # Prompt user to select the correct COM port
-    ser = serial.Serial(com_port, 9600)  # Open serial port with the selected COM port
-    time.sleep(2)  # Wait for the serial connection to initialize
-    print(f"Connected to {com_port}")
-except serial.SerialException as e:
-    print(f"Error opening serial port: {e}")
-    exit()
-
-# Function to update the bus information based on received serial data
-def display_bus_info():
-    global bus_info
-    try:
-        while True:
-            if ser.in_waiting > 0:
-                # Read data from serial port
-                data = ser.readline().decode('utf-8', errors='ignore').strip()  # Decode and remove extra spaces
-                
-                # Debug print to see the raw serial data
-                print(f"Raw Data: {data}")
-
-                if "Bus Name:" in data:
-                    bus_info['bus_name'] = data.split("Bus Name:")[1].strip()
-
-                elif "Seat Status:" in data:
-                    bus_info['seat_status'] = data.split("Seat Status:")[1].strip()
-
-                elif "Distance:" in data:
-                    bus_info['distance'] = data.split("Distance:")[1].strip()
-
-                elif "LED:" in data:
-                    bus_info['led_status'] = data.split("LED:")[1].strip()
-
-                elif "Buzzer:" in data:
-                    bus_info['buzzer_status'] = data.split("Buzzer:")[1].strip()
-
-            time.sleep(1)  # Small delay to prevent flooding the terminal
-
-    except KeyboardInterrupt:
-        print("Exiting...")
-
-# Start the bus information display in a separate thread
-bus_info_thread = threading.Thread(target=display_bus_info)
-bus_info_thread.daemon = True
-bus_info_thread.start()
-
-# Streamlit Web Interface
-st.title('Bus System Status')
-
-# Display the bus information
-col1, col2 = st.columns(2)
-
-with col1:
-    st.subheader('Bus Name')
-    bus_name = st.empty()
-
-with col2:
-    st.subheader('Seat Status')
-    seat_status = st.empty()
-
-st.subheader('Distance to Bus')
-distance_display = st.empty()
-
-# Display LED and Buzzer status
-st.subheader('LED Status')
-led_status_display = st.empty()
-
-st.subheader('Buzzer Status')
-buzzer_status_display = st.empty()
-
-# Update the information on the web interface every second
-while True:
-    bus_name.text(bus_info['bus_name'])
-    seat_status.text(bus_info['seat_status'])
-    distance_display.text(f"Distance: {bus_info['distance']} cm")
-    led_status_display.text(f"LED: {bus_info['led_status']}")
-    buzzer_status_display.text(f"Buzzer: {bus_info['buzzer_status']}")
-    time.sleep(1)
+import serial
+import serial.tools.list_ports
+import time
+import streamlit as st
+import threading
+
+# Global variables to store bus information and hardware statuses
+bus_info = {
+    'bus_name': 'N/A',
+    'seat_status': 'N/A',
+    'distance': 'N/A',
+    'led_status': 'OFF',
+    'buzzer_status': 'OFF',
+}
+
+# Function to identify the correct COM port (Arduino-like device) automatically
+def get_serial_port():
+    ports = serial.tools.list_ports.comports()
+    if not ports:
+        print("No available ports found.")
+        return None
+
+    # Search for Arduino-like device based on description
+    for port in ports:
+        if 'Arduino' in port.description:  # Matching the description of Arduino devices
+            print(f"Automatically selected port: {port.device} ({port.description})")
+            return port.device
+
+    # If no Arduino-like device found, return None
+    print("No Arduino device found.")
+    return None
+
+# Open serial connection to Arduino (use the correct COM port)
+com_port = get_serial_port()
+if com_port:
+    try:
+        ser = serial.Serial(com_port, 9600)  # Open serial port with the selected COM port
+        time.sleep(2)  # Wait for the serial connection to initialize
+        print(f"Connected to {com_port}")
+    except serial.SerialException as e:
+        print(f"Error opening serial port: {e}")
+        st.error("Error opening serial port.")
+else:
+    st.error("Failed to detect Arduino serial port.")
+    exit()
+
+# Function to update the bus information based on received serial data
+def display_bus_info():
+    global bus_info
+    try:
+        while True:
+            if ser.in_waiting > 0:
+                # Read data from serial port
+                data = ser.readline().decode('utf-8', errors='ignore').strip()  # Decode and remove extra spaces
+                
+                # Debug print to see the raw serial data
+                print(f"Raw Data: {data}")
+
+                if "Bus Name:" in data:
+                    bus_info['bus_name'] = data.split("Bus Name:")[1].strip()
+
+                elif "Seat Status:" in data:
+                    bus_info['seat_status'] = data.split("Seat Status:")[1].strip()
+
+                elif "Distance:" in data:
+                    bus_info['distance'] = data.split("Distance:")[1].strip()
+
+                elif "LED:" in data:
+                    bus_info['led_status'] = data.split("LED:")[1].strip()
+
+                elif "Buzzer:" in data:
+                    bus_info['buzzer_status'] = data.split("Buzzer:")[1].strip()
+
+            time.sleep(1)  # Small delay to prevent flooding the terminal
+
+    except KeyboardInterrupt:
+        print("Exiting...")
+
+# Start the bus information display in a separate thread
+bus_info_thread = threading.Thread(target=display_bus_info)
+bus_info_thread.daemon = True
+bus_info_thread.start()
+
+# Streamlit Web Interface
+st.title('Bus System Status')
+
+# Display the bus information
+col1, col2 = st.columns(2)
+
+with col1:
+    st.subheader('Bus Name')
+    bus_name = st.empty()
+
+with col2:
+    st.subheader('Seat Status')
+    seat_status = st.empty()
+
+st.subheader('Distance to Bus')
+distance_display = st.empty()
+
+# Display LED and Buzzer status
+st.subheader('LED Status')
+led_status_display = st.empty()
+
+st.subheader('Buzzer Status')
+buzzer_status_display = st.empty()
+
+# Update the information on the web interface every second
+while True:
+    bus_name.text(bus_info['bus_name'])
+    seat_status.text(bus_info['seat_status'])
+    distance_display.text(f"Distance: {bus_info['distance']} cm")
+    led_status_display.text(f"LED: {bus_info['led_status']}")
+    buzzer_status_display.text(f"Buzzer: {bus_info['buzzer_status']}")
+    time.sleep(1)