graphs?

app.py

@@ -3,6 +3,11 @@ import paho.mqtt.client as mqtt
 import os
 import json
 import time
+import matplotlib.pyplot as plt
+import numpy as np
+import datetime
+import pytz
+import pandas as pd
 
 HOST = os.environ.get("host")
 PORT = int(os.environ.get("port"))
@@ -20,6 +25,11 @@ rpm_mqtt = None
 led_mqtt = None
 experiments = []
 
+temp_graph = None
+od_graph = None
+norm_od_graph = None
+growth_rate_graph = None
+
 
 def on_connect(client, userdata, flags, rc):
     print("Connected with result code "+str(rc))
@@ -609,7 +619,145 @@ def change_experiment(ch_exp, host, port, username, password, pioreactor, exp):
         "reactor": pioreactor,
     }
     client_custom.publish(f"pioreactor/control", json.dumps(payload))
-
+    
+def on_message(client, userdata, message):
+    payload = message.payload.decode("utf-8")
+    data = json.loads(payload)
+    global temp_graph
+    global od_graph
+    global norm_od_graph
+    global growth_rate_graph
+    temp_graph = data.get("temperature", None)
+    od_graph = data.get("od", None)
+    norm_od_graph = data.get("normalized_od", None) 
+    growth_rate_graph = data.get("growth_rate", None)
+    
+def get_data_default(time_scale, exp):
+    global client
+    
+    payload = {
+        "command": "get_readings",
+        "experiment": exp,
+        "filter_mod": 1,
+        "lookback": 1000000,
+        "filter_mod2": 1,
+        "lookback2": 1000000,
+        "filter_mod3": 1,
+        "lookback3": 1000000,
+        "filter_mod4": 1,
+        "lookback4": 1000000,
+        "amount": time_scale,
+        "amount2": time_scale,
+        "amount3": time_scale,
+        "amount4": time_scale,
+    }
+    
+    client.publish(f"pioreactor/control", json.dumps(payload))
+    
+    client.on_message = on_message
+    client.subscribe(f"pioreactor/{PIOREACTOR}/readings")
+    
+    timeout = 10
+    start = time.time()
+    
+    global temp_graph
+    global od_graph
+    global norm_od_graph
+    global growth_rate_graph
+    
+    temp_graph = None
+    od_graph = None
+    norm_od_graph = None
+    growth_rate_graph = None
+    
+    while temp_graph is None and (time.time() - start) < timeout:
+        time.sleep(0.1)
+        
+    client.unsubscribe(f"pioreactor/{PIOREACTOR}/readings")
+    
+    if len(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")
+        
+        plt.figure()
+        plt.plot(df)
+        plt.xlabel("Time")
+        plt.ylabel("Temperature (C)")
+        plt.title("Temperature vs Time")
+        plt.legend()
+        plot1 = plt.gcf()
+    else:
+        plot1 = None
+        
+    if len(od_graph) != 0:
+        for od in od_graph:
+            utc_time = datetime.strptime(od["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)
+            od["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+        df = pd.DataFrame(od_graph)
+        df = df.set_index("x")
+        
+        plt.figure()
+        plt.plot(df)
+        plt.xlabel("Time")
+        plt.ylabel("OD")
+        plt.title("OD vs Time")
+        plt.legend()
+        plot2 = plt.gcf()
+    else:
+        plot2 = None
+    
+    if len(norm_od_graph) != 0:
+        for od in norm_od_graph:
+            utc_time = datetime.strptime(od["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)
+            od["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+        df = pd.DataFrame(norm_od_graph)
+        df = df.set_index("x")
+        
+        plt.figure()
+        plt.plot(df)
+        plt.xlabel("Time")
+        plt.ylabel("Normalized OD")
+        plt.title("Normalized OD vs Time")
+        plt.legend()
+        plot3 = plt.gcf()
+    else:
+        plot3 = None
+        
+    if len(growth_rate_graph) != 0:
+        for gr in growth_rate_graph:
+            utc_time = datetime.strptime(gr["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)
+            gr["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+        df = pd.DataFrame(growth_rate_graph)
+        df = df.set_index("x")
+        
+        plt.figure()
+        plt.plot(df)
+        plt.xlabel("Time")
+        plt.ylabel("Growth Rate")
+        plt.title("Growth Rate vs Time")
+        plt.legend()
+        plot4 = plt.gcf()
+    else:
+        plot4 = None
+    
+    return plot1, plot2, plot3, plot4    
+    
+    
 # Define the interface components
 with gr.Blocks() as demo:
     with gr.Tab("Default"):
@@ -813,6 +961,25 @@ with gr.Blocks() as demo:
                         fn=cycle_media_default,
                         inputs=[cycle_media_sec, experiment_input]
                     )
+                    
+    with gr.Tab("Default Graphs"):
+        with gr.Row():
+            time_scale = gr.RadioButton(label="Time Scale", choices=["1 hour", "24 hour", "All Data"], default="24 hour")
+            get_graphs = gr.Button("Get Graphs")
+            
+        with gr.Row():
+            plot1 = gr.Plot()
+            plot2 = gr.Plot()
+        with gr.Row():
+            plot3 = gr.Plot()
+            plot4 = gr.Plot()
+            
+        get_graphs.click(
+            fn=get_data_default,
+            inputs=[time_scale, experiment_input],
+            outputs=[plot1, plot2, plot3, plot4]
+        )
+            
 
     with gr.Tab("Custom"):
         # Input components

requirements.txt

@@ -1 +1,3 @@
 paho-mqtt>=1.6.1
+matplotlib>=3.3.4
+numpy>=1.20.1