update

db_utils.py

@@ -50,23 +50,25 @@ def get_latest_prediction():
     return None
 
 
-
-
-
-
 def get_all_predictions():
     conn = sqlite3.connect(DB_FILE)
-    cursor = conn.cursor()
-    cursor.execute("SELECT * FROM predictions")  
-    rows = cursor.fetchall()
+    c = conn.cursor()
+    c.execute('''
+        SELECT input, predicted_class, probability, timestamp
+        FROM predictions
+        ORDER BY id DESC
+        LIMIT 2
+    ''')
+    rows = c.fetchall()
     conn.close()
+
     results = []
     for row in rows:
-        if row:
-            results.append({
-                "input": json.loads(row[0]),
-                "class": row[1],
-                "probability": row[2],
-                "timestamp": row[3]
-            })
+        results.append({
+            "input": json.loads(row[0]),
+            "class": row[1],
+            "probability": row[2],
+            "timestamp": row[3]
+        })
+
     return results