From 7d720b54474d4fbe2fa1068db992c617520fe8bc Mon Sep 17 00:00:00 2001
From: Aymeric Gaurat-Apelli <aymeric@gaurat.net>
Date: Wed, 2 Oct 2024 15:52:08 +1000
Subject: [PATCH] Fix JSON?

---
 src/main.ts | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/main.ts b/src/main.ts
index ef4d584..63859ba 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -140,7 +140,15 @@ async function getAIResponse(prompt: string): Promise<Array<{
     });
 
     const res = response.choices[0].message?.content?.trim() || "{}";
-    return JSON.parse(res).reviews;
+    // Remove lines ```json and ``` from the response
+    const jsonStart = res.indexOf("{");
+    const jsonEnd = res.lastIndexOf("}");
+    if (jsonStart === -1 || jsonEnd === -1) {
+      console.error("Error: JSON not found in response: " + res);
+      return null;
+    }
+    const jsonRes = res.substring(jsonStart, jsonEnd + 1);
+    return JSON.parse(jsonRes).reviews;
   } catch (error) {
     console.error("Error:", error);
     return null;