Update app.py (#2)

- Update app.py (260a33a272fe3b40b98bc22cba2cbed314232980)

app.py

@@ -1,96 +1,64 @@
+import streamlit as st
 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.")
+# Function to initialize serial connection
+def initialize_serial_connection():
+    com_port = predict_serial_port(collect_serial_ports())  # Get the correct port from the AI model
+    if com_port:
+        try:
+            ser = serial.Serial(com_port, 9600)
+            time.sleep(2)  # Wait for the connection to initialize
+            print(f"Connected to {com_port}")
+            return ser
+        except serial.SerialException as e:
+            st.error(f"Error opening serial port: {e}")
+            return None
+    else:
+        st.error("No Arduino device 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
+# Initialize serial connection
+ser = initialize_serial_connection()
 
-# 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
+# Function to update bus information
 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()
+    bus_info = {
+        'bus_name': 'N/A',
+        'seat_status': 'N/A',
+        'distance': 'N/A',
+        'led_status': 'OFF',
+        'buzzer_status': 'OFF',
+    }
+
+    while True:
+        if ser.in_waiting > 0:
+            # Read data from serial port
+            data = ser.readline().decode('utf-8', errors='ignore').strip()
+            
+            # Process the incoming 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)  # Delay to avoid flooding the interface
+
+    return bus_info
 
 # 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()
@@ -107,6 +75,7 @@ buzzer_status_display = st.empty()
 
 # Update the information on the web interface every second
 while True:
+    bus_info = display_bus_info()
     bus_name.text(bus_info['bus_name'])
     seat_status.text(bus_info['seat_status'])
     distance_display.text(f"Distance: {bus_info['distance']} cm")