fix: json extractor code

Author: batsidev

src/analysisWorker.js

@@ -200,10 +200,10 @@ class AnalysisWorker {
     });
   }
 
-  extractJsonFromMessage(message) {
+  extractJsonFromMessage(data) {
+    const message = data.message;
     if (!message) return null;
 
-    // 1. Extract code-fenced JSON inside ```json ... ```
     const fenceRegex = /```json\s*([\s\S]*?)```/i;
     const fencedMatch = message.match(fenceRegex);
 
@@ -212,7 +212,6 @@ class AnalysisWorker {
     if (fencedMatch && fencedMatch[1]) {
       jsonString = fencedMatch[1];
     } else {
-      // fallback: try to extract first JSON-like structure from message
       const looseJsonRegex = /\{[\s\S]*\}/;
       const looseMatch = message.match(looseJsonRegex);
       if (looseMatch) jsonString = looseMatch[0];
@@ -223,15 +222,19 @@ class AnalysisWorker {
       return null;
     }
 
-    try {
-      // Clean invisible characters
-      jsonString = jsonString.trim();
+    let cleaned = jsonString
+      .split("\n")
+      .filter((line) => !line.trim().startsWith("//"))
+      .join("\n");
+
+    cleaned = cleaned.replace(/,\s*([}\]])/g, "$1");
+    cleaned = cleaned.trim();
 
-      // Parse JSON safely
-      return JSON.parse(jsonString);
+    try {
+      return JSON.parse(cleaned);
     } catch (err) {
       console.error("❌ Failed to parse JSON:", err);
-      console.error("Raw JSON string:", jsonString);
+      console.error("Cleaned JSON was:\n", cleaned);
       return null;
     }
   }

src/index.js

@@ -162,7 +162,7 @@ app.post('/analyze', async (req, res) => {
 
     const repoName = repo;
     const branchName = branch || 'main';
-    const analysisQuestion = question || 'Analyze and explain the repository architecture, return the full set of snippet for the full project architecture knowledge, analyze every node and provide a rich "analysis_response" result';
+    const analysisQuestion = question || 'Analyze and explain the repository architecture, return the full set of snippet for the full project architecture knowledge, analyze every node and provide a rich and complete "analysis_response" result';
 
     console.log(`Starting analysis for repository: ${repoName}, branch: ${branchName}`);
 

src/potpieClient.js

@@ -52,7 +52,7 @@ class PotpieClient {
     async getParsingStatus(projectId) {
         try {
             const response = await this.client.get(`/api/v2/parsing-status/${projectId}`);
-            console.log(response);
+
             return {
                 success: true,
                 data: response.data