Upload 2 files

server/models/Dataset.js

@@ -4,16 +4,34 @@ const db = require('../config/db');
 class Dataset {
   // Get all datasets
   static async getAll() {
-    const query = 'SELECT * FROM datasets ORDER BY id';
-    const { rows } = await db.query(query);
-    return rows;
+    try {
+      if (process.env.HF_SPACES === 'true') {
+        return db.mockDatasets;
+      }
+      
+      const query = 'SELECT * FROM datasets ORDER BY id';
+      const { rows } = await db.query(query);
+      return rows.length > 0 ? rows : db.mockDatasets;
+    } catch (error) {
+      console.warn('Database error, using mock data:', error.message);
+      return db.mockDatasets;
+    }
   }
 
   // Get dataset by ID
   static async getById(id) {
-    const query = 'SELECT * FROM datasets WHERE id = $1';
-    const { rows } = await db.query(query, [id]);
-    return rows[0];
+    try {
+      if (process.env.HF_SPACES === 'true') {
+        return db.mockDatasets.find(dataset => dataset.id == id);
+      }
+      
+      const query = 'SELECT * FROM datasets WHERE id = $1';
+      const { rows } = await db.query(query, [id]);
+      return rows[0] || db.mockDatasets.find(dataset => dataset.id == id);
+    } catch (error) {
+      console.warn('Database error, using mock data:', error.message);
+      return db.mockDatasets.find(dataset => dataset.id == id);
+    }
   }
 
   // Create new dataset

server/models/News.js

@@ -4,16 +4,34 @@ const db = require('../config/db');
 class News {
   // Get all news
   static async getAll() {
-    const query = 'SELECT * FROM news ORDER BY date DESC';
-    const { rows } = await db.query(query);
-    return rows;
+    try {
+      if (process.env.HF_SPACES === 'true') {
+        return db.mockNews;
+      }
+      
+      const query = 'SELECT * FROM news ORDER BY date DESC';
+      const { rows } = await db.query(query);
+      return rows.length > 0 ? rows : db.mockNews;
+    } catch (error) {
+      console.warn('Database error, using mock data:', error.message);
+      return db.mockNews;
+    }
   }
 
   // Get news by ID
   static async getById(id) {
-    const query = 'SELECT * FROM news WHERE id = $1';
-    const { rows } = await db.query(query, [id]);
-    return rows[0];
+    try {
+      if (process.env.HF_SPACES === 'true') {
+        return db.mockNews.find(news => news.id == id);
+      }
+      
+      const query = 'SELECT * FROM news WHERE id = $1';
+      const { rows } = await db.query(query, [id]);
+      return rows[0] || db.mockNews.find(news => news.id == id);
+    } catch (error) {
+      console.warn('Database error, using mock data:', error.message);
+      return db.mockNews.find(news => news.id == id);
+    }
   }
 
   // Create new news item