Update results.py

results.py

@@ -1,7 +1,25 @@
 from fasthtml.common import *
 from fasthtml.components import *
+import json
+from matplotlib import pyplot
+from fh_matplotlib import matplotlib2fasthtml
 
 
+with open("graph_data.jsonl", 'r') as f:
+    for line in f:
+        for title, meta in json.loads(line).items():
+            _, ax = pyplot.subplots()
+            for i, (label, data, color) in enumerate(meta["data"]):
+                ax.plot(*data, label=label, color=color)
+
+            ax.grid()
+            ax.set_xlabel(meta["x_label"])
+            ax.set_ylabel(meta["y_label"])
+            ax.set_title(title)
+            ax.legend()
+
+fig = pyplot.show()
+
 intro_div = Div(
         H2("Perplexity Evaluation on Duplicate Data"),
         H3("Model based Quality Estimation"),
@@ -79,6 +97,7 @@ def results():
                     intro_div,
                     perp1_div,
                     llama_div,
+                    matplotlib2fasthtml(fig),
                     id="inner-text"
                 )
     )