plot failures separately

app.py

@@ -28,8 +28,7 @@ models = [
     "mistralai/Mistral-7B-Instruct-v0.3",
     "mistralai/Mixtral-8x7B-Instruct-v0.1",
 ]
-LOG_FILE = "api_logs.json"
-CHECK_INTERVAL = 30  # 1 minute
+LOG_FILE = "/data/api_logs.json"
 
 
 client = AsyncInferenceClient(token=os.environ["HF_INFERENCE_API_TOKEN"])
@@ -75,7 +74,7 @@ async def check_apis():
 @app.on_event("startup")
 async def start_scheduler():
     scheduler = AsyncIOScheduler()
-    scheduler.add_job(check_apis, 'interval', minutes=1)
+    scheduler.add_job(check_apis, 'interval', minutes=10)
     scheduler.start()
 
 @app.get("/")
@@ -88,7 +87,7 @@ async def get_logs():
         logs = json.load(f)
     return logs
 
-@app.get("/api/chart-data", response_model=Dict[str, Dict[str, List]])
+@app.get("/api/chart-data", response_model=Dict[str, Dict[str, Dict[str, List]]])
 async def get_chart_data():
     with open(LOG_FILE, "r") as f:
         logs = json.load(f)
@@ -97,9 +96,14 @@ async def get_chart_data():
     for log in logs:
         model = log['model']
         if model not in chart_data:
-            chart_data[model] = {'x': [], 'y': []}
-        chart_data[model]['x'].append(log['timestamp'])
-        chart_data[model]['y'].append(1 if log['success'] else 0)
+            chart_data[model] = {
+                'success': {'x': [], 'y': []},
+                'failure': {'x': [], 'y': []}
+            }
+        
+        status = 'success' if log['success'] else 'failure'
+        chart_data[model][status]['x'].append(log['timestamp'])
+        chart_data[model][status]['y'].append(1)
     
     return chart_data