give description, constraints, and examples to gpt for better gpt analysis

Author: zubyj

README.md

@@ -1,7 +1,7 @@
 # Leetcode Explained 
 ![Downloads](https://img.shields.io/chrome-web-store/users/cofoinjfjcpgcjiinjhcpomcjoalijbe)
-![Version](https://img.shields.io/chrome-web-store/v/cofoinjfjcpgcjiinjhcpomcjoalijbe) 
 ![Rating](https://img.shields.io/chrome-web-store/rating/cofoinjfjcpgcjiinjhcpomcjoalijbe)
+![Version](https://img.shields.io/chrome-web-store/v/cofoinjfjcpgcjiinjhcpomcjoalijbe) 
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 
 A free, open source browser extension that adds video solutions, company tags, and GPT code analysis into each Leetcode problem
@@ -52,10 +52,8 @@ We welcome contributions! Please see the [CONTRIBUTING.md](docs/CONTRIBUTING.md)
 ### Screenshots
 
 <img src="src/assets/images/screenshots/add-video.png" alt="Add Video" width="600"/>
-
-
+<img src="src/assets/images/screenshots/get-company-tags.png" alt="Company Tags" width="600"/>
 <img src="src/assets/images/screenshots/get-complexity.png" alt="Code Complexity" width="600"/>
-
 <img src="src/assets/images/screenshots/fix-code.png" alt="Fix Code" width="600"/>
 
 

src/content-script/get-user-code.ts

@@ -7,13 +7,40 @@ chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
     }
 });
 
-// Read the users code from the Leetcode problem page
+// Read the users code and examples from the Leetcode problem page
 function getCode() {
     const viewLines = document.getElementsByClassName('view-line');
-    const textArray = [];
+    const textArray = ['Heres the code'];
 
     for (const viewLine of viewLines) {
         textArray.push(viewLine.textContent);
     }
+
+    textArray.push('Heres the problem description, test cases, and constraints\n');
+
+    // Add the test cases & examples to the 
+    const descriptionContainer = document.querySelector('div._1l1MA') as Element;
+    if (!descriptionContainer) {
+        return;
+    }
+
+    // Add all the text from description container to the text array
+    for (const child of descriptionContainer.children) {
+        textArray.push(child.textContent);
+    }
+
+    const examples = descriptionContainer.getElementsByClassName('example');
+    if (examples && examples.length > 0) {
+        const parent = examples[0].parentNode as Element;
+        if (!parent) {
+            return;
+        }
+        const startIndex = Array.from(descriptionContainer.children).indexOf(parent);
+        for (let i = startIndex; i < descriptionContainer.children.length; i++) {
+            const child = descriptionContainer.children[i] as HTMLElement;
+            console.log(child);
+            child.style.display = showExamples ? 'block' : 'none';
+        }
+    }
     return textArray;
 }

src/popup/popup.ts

@@ -137,13 +137,14 @@ function processCode(
         As a coding expert, I require your aid with a specific LeetCode problem
         called ${problemTitle}. Generate the best possible, optimized solution 
         for this problem. Only the function definition and code should be returned
-        in plain text format with no usage of code blocks. Do not return any code comments.
-        Anything other than the code text is not permitted.`;
-        if (infoMessage) infoMessage.textContent = 'Creating the solution ...';
+        in plain text format with no usage of code blocks. Code comments are optional.
+        Anything other than the code text is not permitted. I should be able to copy
+        and paste the code into the LeetCode editor and run it successfully.`;
+        if (infoMessage) infoMessage.textContent = 'Generating solution code ...';
         analyzeCodeResponse && analyzeCodeResponse.classList.add('hidden');
         fixCodeContainer && fixCodeContainer.classList.remove('hidden');
     }
-    prompt += 'Copy the same function definition. Ignore my code\n' + codeText;
+    prompt += 'Heres the problem description, examples, constraints and my code. Note that code might not exist and may be incorrect.\n' + codeText;
 
     let response = '';
     Promise.race([