added save state

app.py

@@ -3,9 +3,19 @@ import json
 from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
 from urllib.parse import parse_qs, urlparse
 from inference import generate_excuse
-from dataset import load_leaderboard, save_leaderboard, update_leaderboard_votes
 
 API_TOKEN = os.getenv("HF_TOKEN")
+LEADERBOARD_FILE = "leaderboard.json"
+
+def save_leaderboard(leaderboard):
+    with open(LEADERBOARD_FILE, "w") as f:
+        json.dump(leaderboard, f)
+
+def load_leaderboard():
+    if os.path.exists(LEADERBOARD_FILE):
+        with open(LEADERBOARD_FILE, "r") as f:
+            return json.load(f)
+    return []
 
 class RequestHandler(SimpleHTTPRequestHandler):
     def do_GET(self):
@@ -31,19 +41,30 @@ class RequestHandler(SimpleHTTPRequestHandler):
                 "max_new_tokens": 100
             })
             leaderboard.sort(key=lambda x: x["votes"], reverse=True)
-            save_leaderboard(leaderboard[:10])  # Keep only top 10
+            leaderboard = leaderboard[:10]  # Keep only top 10
+            save_leaderboard(leaderboard)
             self.send_response(200)
             self.send_header('Content-type', 'application/json')
             self.end_headers()
-            self.wfile.write(json.dumps({'excuse': excuse, 'leaderboard': leaderboard[:10]}).encode())
+            self.wfile.write(json.dumps({'excuse': excuse, 'leaderboard': leaderboard}).encode())
         elif self.path == '/vote':
             index = int(parsed_data['index'][0])
             vote_value = int(parsed_data['vote'][0])
-            updated_leaderboard = update_leaderboard_votes(index, vote_value)
+            leaderboard = load_leaderboard()
+            if 0 <= index < len(leaderboard):
+                leaderboard[index]["votes"] += vote_value
+                leaderboard.sort(key=lambda x: x["votes"], reverse=True)
+                save_leaderboard(leaderboard)
+            self.send_response(200)
+            self.send_header('Content-type', 'application/json')
+            self.end_headers()
+            self.wfile.write(json.dumps({'leaderboard': leaderboard}).encode())
+        elif self.path == '/leaderboard':
+            leaderboard = load_leaderboard()
             self.send_response(200)
             self.send_header('Content-type', 'application/json')
             self.end_headers()
-            self.wfile.write(json.dumps({'leaderboard': updated_leaderboard}).encode())
+            self.wfile.write(json.dumps({'leaderboard': leaderboard}).encode())
         else:
             self.send_error(404, 'Not Found')