Fix: Use Hugging Face persistent /data directory for file storage

- Updated Dockerfile to create directories in /data (persistent in HF Spaces)
- Modified API routes to use /data in production, fallback to local in dev
- Fixed sessionManager to use persistent storage
- Added MCP server icon and description
- This fixes the 'file not found' error after container restarts

Dockerfile

@@ -38,9 +38,10 @@ COPY --from=builder /app/public ./public
 COPY --from=builder --chown=nextjs:nodejs /app/.next/standalone ./
 COPY --from=builder --chown=nextjs:nodejs /app/.next/static ./.next/static
 
-# Create data directory for file uploads
-RUN mkdir -p /app/data/documents /app/data/public /app/data/exams
-RUN chown -R nextjs:nodejs /app/data
+# Create data directory for file uploads (Hugging Face persistent storage)
+# Note: /data is persistent in Hugging Face Spaces
+RUN mkdir -p /data/documents /data/public /data/exams /data/files
+RUN chown -R nextjs:nodejs /data
 
 # Install Python for MCP backend
 RUN apk add --no-cache python3 py3-pip

app/api/download/route.ts

@@ -2,7 +2,10 @@ import { NextRequest, NextResponse } from 'next/server'
 import fs from 'fs'
 import path from 'path'
 
-const DATA_DIR = path.join(process.cwd(), 'data')
+// Use /data for Hugging Face persistent storage, fallback to local for development
+const DATA_DIR = process.env.NODE_ENV === 'production' && fs.existsSync('/data')
+  ? '/data'
+  : path.join(process.cwd(), 'data')
 const DOCS_DIR = path.join(DATA_DIR, 'documents')
 
 export async function GET(request: NextRequest) {

app/api/files/route.ts

@@ -2,7 +2,10 @@ import { NextRequest, NextResponse } from 'next/server'
 import fs from 'fs'
 import path from 'path'
 
-const DATA_DIR = path.join(process.cwd(), 'data')
+// Use /data for Hugging Face persistent storage, fallback to local for development
+const DATA_DIR = process.env.NODE_ENV === 'production' && fs.existsSync('/data')
+  ? '/data'
+  : path.join(process.cwd(), 'data')
 const DOCS_DIR = path.join(DATA_DIR, 'documents')
 
 // Ensure directories exist

app/api/upload/route.ts

@@ -2,7 +2,10 @@ import { NextRequest, NextResponse } from 'next/server'
 import fs from 'fs'
 import path from 'path'
 
-const DATA_DIR = path.join(process.cwd(), 'data')
+// Use /data for Hugging Face persistent storage, fallback to local for development
+const DATA_DIR = process.env.NODE_ENV === 'production' && fs.existsSync('/data')
+  ? '/data'
+  : path.join(process.cwd(), 'data')
 const DOCS_DIR = path.join(DATA_DIR, 'documents')
 const MAX_FILE_SIZE = 50 * 1024 * 1024 // 50MB
 

lib/sessionManager.ts

@@ -1,5 +1,6 @@
 import crypto from 'crypto';
 import fs from 'fs/promises';
+import fsSync from 'fs';
 import path from 'path';
 
 export interface Session {
@@ -17,8 +18,13 @@ export class SessionManager {
   private publicDir: string;
 
   private constructor() {
-    this.sessionDir = path.join(process.cwd(), 'data', 'files');
-    this.publicDir = path.join(process.cwd(), 'data', 'public');
+    // Use /data for Hugging Face persistent storage, fallback to local for development
+    const baseDataDir = process.env.NODE_ENV === 'production' && fsSync.existsSync('/data')
+      ? '/data'
+      : path.join(process.cwd(), 'data');
+
+    this.sessionDir = path.join(baseDataDir, 'files');
+    this.publicDir = path.join(baseDataDir, 'public');
     this.initializeDirectories();
   }
 

mcp-server.js

@@ -17,6 +17,8 @@ class ReubenOSMCPServer {
       {
         name: 'reubenos-file-manager',
         version: '1.0.0',
+        description: 'AI-powered desktop environment with file management and document generation',
+        icon: '🖥️',
       },
       {
         capabilities: {