added temp graph which auto reloads after set time, improvements to the handling of data and differnt time intervals needs to be added

app.py

@@ -3,10 +3,18 @@ import paho.mqtt.client as mqtt
 import json
 import time
 import os
+import asyncio
+import pandas as pd
+
+from datetime import datetime
+import pytz
 
 
 experiment = None
 running = []
+temp_graph = None
+
+REQUEST_INTERVAL = 60
 
 if "experiment" not in st.session_state:
     st.session_state["experiment"] = None
@@ -56,7 +64,7 @@ else:
 
 
 # Create the MQTT client
-@st.cache_resource
+# @st.cache_resource
 def create_mqtt_client(hivemq_host, hivemq_port, hivemq_username, hivemq_password):
     client = mqtt.Client()
 
@@ -83,6 +91,40 @@ def on_message_worker(client, userdata, message):
     experiment = data.get("experiment", None)
     running = data.get("running", [])
 
+def on_message(client, userdata, message):
+    payload = message.payload.decode("utf-8")
+    data = json.loads(payload)
+    global temp_graph
+    temp_graph = data.get("data", None)
+
+async def publish_temperature_request(client, experiment, filter_mod_N, lookback, reactor):
+        """Publish a temperature request to the request topic."""
+        payload = {
+            "command": "get_temperature_readings",
+            "experiment": experiment,
+            "filter_mod": filter_mod_N,
+            "lookback": lookback,
+            "reactor": reactor
+        }
+        payload_str = json.dumps(payload)
+        client.publish("pioreactor/control", payload_str)
+
+async def periodic_temp_reading(client, reactor, experiment):
+    try:
+        await publish_temperature_request(client, experiment, 1, 100000, reactor)
+    except asyncio.CancelledError:
+        pass
+
+async def run_mqtt_client(client):
+    client.subscribe(f"pioreactor/{reactor}/temp_reading")
+    client.subscribe(f"pioreactor/{reactor}/od_reading")
+    client.subscribe(f"pioreactor/{reactor}/growth_rate")
+
+    client.on_message = on_message
+
+    client.loop_start()
+    await periodic_temp_reading(client, reactor, st.session_state["experiment"])
+
 if submit_button:
     client = create_mqtt_client(hivemq_host, hivemq_port, hivemq_username, hivemq_password)
 
@@ -315,7 +357,7 @@ if st.session_state["experiment"] is not None:
                 if temp_auto == "Heat To Temp":
                     if st.session_state.get("temp", None) is not None:
                         st.write(f"Current Temperature: {st.session_state['temp']} °C")
-                    temp = st.slider("Temperature", min_value=0.0, max_value=60.0, step=0.1, value=st.session_state["temp"] if st.session_state["temp"] is not None else 30.0)
+                    temp = st.slider("Temperature", min_value=0, max_value=60, step=1, value=st.session_state["temp"] if st.session_state["temp"] is not None else 30)
                     cols = st.columns(2)
                     with cols[0]:
                         temp_submit = st.form_submit_button("Update Temperature Automation")
@@ -329,10 +371,11 @@ if st.session_state["experiment"] is not None:
                         temp_stop = st.form_submit_button("Stop Temperature Automation")
 
     else:
+        temp = None
         temp_auto = st.selectbox("Temperature Automation", ["Record Temp Only", "Heat To Temp"], placeholder="Select an option")
         with st.form("temperature_form"):
             if temp_auto == "Heat To Temp":
-                temp = st.slider("Temperature", min_value=0.0, max_value=60.0, step=0.1, value=30.0)
+                temp = st.slider("Temperature", min_value=0, max_value=60, step=1, value=30)
                 temp_submit = st.form_submit_button("Start Temperature Automation")
                 temp_stop = False
             elif temp_auto == "Record Temp Only":
@@ -475,4 +518,60 @@ if st.session_state["experiment"] is not None:
         time.sleep(3)
         st.rerun()
 
-    st.divider()
+    st.divider()
+
+    # Display the temperature graph 
+    # Using asyncio to run the loop in the background and periodically update the graph
+
+
+
+
+
+    st.header("Graphs")
+
+    payload = {
+        "command": "get_temperature_readings",
+        "experiment": st.session_state["experiment"],
+        "filter_mod": 1,
+        "lookback": 100,
+        "reactor": reactor
+    }
+
+    st.session_state["client"].loop_start()
+    st.session_state["client"].on_message = on_message
+    payload_str = json.dumps(payload)
+    st.session_state["client"].subscribe(f"pioreactor/{reactor}/temperature")
+
+    temp_graph_actual = []
+
+    st.subheader("Temperature Graph")
+    placeholder = st.empty()
+
+    while True:
+        temp_graph = None
+
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        timeout = 30
+        start_time = time.time()
+
+        while temp_graph is None and (time.time() - start_time) < timeout:
+            time.sleep(1)
+
+        temp_graph = temp_graph[0]
+
+        for temp in temp_graph:
+            utc_time = datetime.strptime(temp["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
+            local_tz = pytz.timezone("America/New_York")
+            utc_time = utc_time.replace(tzinfo=pytz.utc)
+            local_time = utc_time.astimezone(local_tz)
+            temp["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+
+        df = pd.DataFrame(temp_graph)
+        df = df.set_index("x")
+
+        placeholder.line_chart(df, x_label="Time", y_label="Temperature (°C)")
+        
+        time.sleep(REQUEST_INTERVAL)
+    
+