chore: Update RGB control panel with Streamlit and MQTT implementation

app.py

@@ -53,11 +53,17 @@ R = st.slider("Select the Red value:", min_value=0, max_value=max_value, value=0
 G = st.slider("Select the Green value:", min_value=0, max_value=max_value, value=0)
 B = st.slider("Select the Blue value:", min_value=0, max_value=max_value, value=0)
 
-# Initialize session state for messages and lock state
+# # Initialize session state for messages and lock state
 if "messages" not in st.session_state:
     st.session_state.messages = []
-if "locked" not in st.session_state:
-    st.session_state.locked = False
+
+# if "locked" not in st.session_state:
+#     st.session_state.locked = False
+
+# if "command_history" not in st.session_state:
+#     st.session_state.command_history = []
+# if "sensor_data_history" not in st.session_state:
+#     st.session_state.sensor_data_history = []
 
 # Queue to hold sensor data
 sensor_data_queue = queue.Queue()
@@ -69,7 +75,9 @@ def get_paho_client(
     sensor_data_topic, hostname, username, password=None, port=8883, tls=True
 ):
 
-    client = mqtt.Client(protocol=mqtt.MQTTv5)  # create new instance
+    client = mqtt.Client(
+        mqtt.CallbackAPIVersion.VERSION2, protocol=mqtt.MQTTv5
+    )  # create new instance
 
     def on_message(client, userdata, msg):
         sensor_data_queue.put(json.loads(msg.payload))
@@ -92,14 +100,24 @@ def get_paho_client(
 
 
 def send_and_receive(client, command_topic, msg, queue_timeout=60):
-    client.publish(command_topic, json.dumps(msg), qos=2)
+    print("Sending command...")
+    result = client.publish(command_topic, json.dumps(msg), qos=2)
+    result.wait_for_publish()  # Ensure the message is sent
 
-    while True:
+    if result.rc == mqtt.MQTT_ERR_SUCCESS:
+        print(f"Command sent: {msg} to topic {command_topic}")
+    else:
+        print(f"Failed to send command: {result.rc}")
+
+    try:
         sensor_data = sensor_data_queue.get(True, queue_timeout)
         return sensor_data
+    except queue.Empty:
+        st.error("No sensor data received within the timeout period.")
+        return None
 
 
-# helper function to plot discrete spectral sensor data
+# Helper function to plot discrete spectral sensor data
 def plot_spectra(sensor_data):
     """https://chatgpt.com/share/210d2fee-ca64-45a5-866e-e6df6e56bd1c"""
     wavelengths = np.array([410, 440, 470, 510, 550, 583, 620, 670])
@@ -148,23 +166,28 @@ def plot_spectra(sensor_data):
     for wavelength, intensity in zip(wavelengths, intensities):
         ax.vlines(wavelength, 0, intensity, color="gray", linestyle="--", linewidth=1)
 
+    # Adjust limits and labels with larger font size
     ax.set_xlim(wavelengths.min() - 10, wavelengths.max() + 10)
-    ax.set_ylim(0, max(intensities) + 10)  # Ensure the lower y limit is 0
+    ax.set_ylim(
+        0, max(intensities) + 15
+    )  # Ensure the lower y limit is 0 and add buffer
     ax.set_xticks(wavelengths)
-    ax.set_xlabel("Wavelength (nm)")
-    ax.set_ylabel("Intensity")
-    ax.set_title("Spectral Intensity vs. Wavelength")
+    ax.set_xlabel("Wavelength (nm)", fontsize=14)
+    ax.set_ylabel("Intensity", fontsize=14)
+    ax.set_title("Spectral Intensity vs. Wavelength", fontsize=16)
+    ax.tick_params(axis="both", which="major", labelsize=12)
 
     st.pyplot(fig)
 
 
 # Publish button
-if st.button("Send RGB Command", disabled=st.session_state.locked):
+# if st.button("Send RGB Command", disabled=st.session_state.locked):
+if st.button("Send RGB Command"):
     if not PICO_ID or not HIVEMQ_HOST or not HIVEMQ_USERNAME or not HIVEMQ_PASSWORD:
         st.error("Please enter all required fields.")
     else:
-        st.session_state.locked = True
-        st.success("Please wait while the command is sent...")
+        # st.session_state.locked = True
+        st.info("Please wait while the command is sent...")
 
         client = get_paho_client(
             sensor_data_topic,
@@ -177,14 +200,14 @@ if st.button("Send RGB Command", disabled=st.session_state.locked):
 
         command_msg = {"R": R, "G": G, "B": B}
         sensor_data = send_and_receive(
-            client, command_topic, command_msg, queue_timeout=30
+            client, command_topic, command_msg, queue_timeout=15
         )
 
-        st.session_state.locked = False
-        st.success("Command sent successfully!")
-        plot_spectra(sensor_data)
-        st.write("Sensor Data Received:", sensor_data)
+        # st.session_state.locked = False
+
+        print("Waiting for sensor data...")
 
-# Display received messages
-for message in st.session_state.messages:
-    st.write(f"Received message: {message}")
+        if sensor_data:
+            st.success("Command sent successfully!")
+            plot_spectra(sensor_data)
+            st.write("Sensor Data Received:", sensor_data)