Update app.js

app.js

@@ -219,42 +219,96 @@ app.post('/feedback', async (req, res) => {
     }
 });
 
-/**
- * FEEDBACK FORWARDER (DASHBOARD)
- * Authenticates via Firebase ID Token (Header).
- * Endpoint: /feedback2
- */
+
+// ---------------------------------------------------------
+// NEW: DELETE ENDPOINT
+// ---------------------------------------------------------
+app.post('/project/delete', verifyFirebaseUser, async (req, res) => {
+    const { projectId } = req.body;
+    const userId = req.user.uid;
+
+    if (!projectId) return res.status(400).json({ error: "Missing Project ID" });
+
+    console.log(`🗑️ Deleting Project: ${projectId} requested by ${userId}`);
+
+    try {
+        // 1. Verify Ownership (Check Firestore or RTDB)
+        const projectRef = db.ref(`projects/${projectId}/info`);
+        const snapshot = await projectRef.once('value');
+        if (snapshot.exists()) {
+            const data = snapshot.val();
+            if (data.userId !== userId) {
+                return res.status(403).json({ error: "Unauthorized" });
+            }
+        }
+
+        const promises = [];
+
+        // 2. Delete from Realtime Database
+        promises.push(db.ref(`projects/${projectId}`).remove());
+        promises.push(db.ref(`plugin_oauth/${userId}/${projectId}`).remove());
+
+        // 3. Delete from Firestore
+        if (firestore) {
+            promises.push(firestore.collection('projects').doc(projectId).delete());
+        }
+
+        // 4. Delete from Storage (Recursive)
+        if (storage) {
+            const bucket = storage.bucket();
+            // Delete all files with the projectId prefix
+            promises.push(bucket.deleteFiles({ prefix: `${projectId}/` }));
+        }
+
+        // 5. Clear from Memory
+        StateManager.cleanupProject(projectId); // You might need to add this method to StateManager or just map delete
+        // Direct map delete if StateManager logic isn't exposed
+        // activeSessions.delete(`${userId}:${projectId}`); 
+        
+        await Promise.all(promises);
+
+        console.log(`✅ Project ${projectId} deleted successfully.`);
+        res.json({ success: true });
+
+    } catch (err) {
+        console.error("Delete Error:", err);
+        res.status(500).json({ error: "Failed to delete project resources" });
+    }
+});
+
+
+// ---------------------------------------------------------
+// UPDATED FEEDBACK ENDPOINT (Handling Images)
+// ---------------------------------------------------------
 app.post('/feedback2', verifyFirebaseUser, async (req, res) => {
-    const { projectId, prompt, ...otherPayload } = req.body;
-    const userId = req.user.uid; // Verified from Firebase Token
+    const { projectId, prompt, images, ...otherPayload } = req.body; // Extract images
+    const userId = req.user.uid;
 
     if (!projectId || !prompt) {
         return res.status(400).json({ error: 'Missing projectId or prompt' });
     }
 
+    // Log if image is received
+    if (images && images.length > 0) {
+        console.log(`📸 Received ${images.length} image(s) from Dashboard.`);
+    }
+
     const externalBase = process.env.EXTERNAL_SERVER_URL || 'http://localhost:7860';
     const targetUrl = externalBase.replace(/\/$/, '') + '/project/feedback';
 
-    console.log(`📨 Forwarding DASHBOARD feedback for ${projectId} (User: ${userId})`);
-
     try {
-        // We inject the authenticated userId here so the Core server knows 
-        // it was verified by the Auth Proxy.
         const response = await axios.post(targetUrl, {
             userId: userId,
             projectId: projectId,
             prompt: prompt,
-            ...otherPayload // images, etc.
+            images: images || [], // Ensure it is passed to Core AI
+            ...otherPayload
         });
 
         return res.json({ success: true, externalResponse: response.data });
-
     } catch (err) {
-        console.error("Dashboard Feedback Forward Error:", err.message);
-        if (err.response) {
-            return res.status(err.response.status).json(err.response.data);
-        }
-        return res.status(502).json({ error: 'Failed to forward feedback to Main AI server' });
+        console.error("Forward Error:", err.message);
+        return res.status(502).json({ error: 'Failed to forward' });
     }
 });