Update results.py

results.py

@@ -1,25 +1,35 @@
 from fasthtml.common import *
 from fasthtml.components import *
 import json
-from matplotlib import pyplot
-from fh_matplotlib import matplotlib2fasthtml
+from fh_plotly import plotly2fasthtml
+import pandas as pd
+import plotly.express as px
+
+#Perplexity Across Different Buckets (global)
+DATA = {
+    "2014": "1-1": 17.410227605477868,
+    "2014": "2-5":   16.11176217183986,
+    "2014": "6-10":  15.632757662414805,
+    "2014": "11-100": 15.446116676532212,
+    "2014": "101-1000": 16.716943171826703,
+    "2014": "1001-30000000": 18.156821563322765,
+}
+Perplexity_Across_Different_Buckets_global = pd.DataFrame(
+    DATA, columns=["Year", "Bucket", "Perplexity"]
+)
+
+fig = px.line(
+    Perplexity_Across_Different_Buckets_global,
+    x="Bucket",
+    y="Perplexity",
+    labels={"Bucket": "Bucket", "Perplexity": "Perplexity"},
+)
+
+Perplexity_Across_Different_Buckets_global_graph = fig
+
 
 
-with open("data/graph_data.jsonl", "r") as f:
-    #if title == "Perplexity Across Different Buckets (global)":
-    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"),
@@ -98,7 +108,8 @@ def results():
                     intro_div,
                     perp1_div,
                     llama_div,
-                    matplotlib2fasthtml(fig),
+                    P("test plotly"),
+                    plotly2fasthtml(Perplexity_Across_Different_Buckets_global_graph),
                     id="inner-text"
                 )
     )