Update src/pages/api/sql.ts

src/pages/api/sql.ts

@@ -23,7 +23,7 @@ export default async function handler(
   }
 
   try {
-    const apiKey = process.env.API_TOKEN;
+    const apiKey = process.env.HUGGINGFACE_API_KEY;
     if (!apiKey) {
       return res.status(500).json({ 
         message: "Server configuration error", 
@@ -76,7 +76,7 @@ Text: ${userPrompt}`;
     try {
       const content = response?.choices?.[0]?.message?.content?.trim() || '';
       
-      const jsonMatch = content.match(/\{.*\}/s);
+      const jsonMatch = content.match(/\{[\s\S]*\}/);
       if (jsonMatch) {
         const parsedResponse = JSON.parse(jsonMatch[0]);
         sqlQuery = parsedResponse.query?.trim();
@@ -120,14 +120,14 @@ Text: ${userPrompt}`;
         
         const dataAnalysis = {
           totalColumns: columns.length,
-          numericColumns: columns.filter(col => 
-            typeof firstRow[col] === 'number' && 
+          numericColumns: columns.filter((col: string) => 
+            typeof (firstRow as any)[col] === 'number' && 
             !col.toLowerCase().includes('id') && 
             !col.toLowerCase().includes('_id')
           ),
-          dateColumns: columns.filter(col => firstRow[col] instanceof Date),
-          stringColumns: columns.filter(col => 
-            typeof firstRow[col] === 'string' ||
+          dateColumns: columns.filter((col: string) => (firstRow as any)[col] instanceof Date),
+          stringColumns: columns.filter((col: string) => 
+            typeof (firstRow as any)[col] === 'string' ||
             col.toLowerCase().includes('name') ||
             col.toLowerCase().includes('title')
           ),
@@ -155,13 +155,13 @@ Text: ${userPrompt}`;
                 visualization = {
                   type: requestedChartType,
                   config: {
-                    labels: results.map(row => 
+                    labels: results.map((row: any) => 
                       dataAnalysis.stringColumns[0] 
                         ? String(row[dataAnalysis.stringColumns[0]]) 
                         : `Row ${results.indexOf(row) + 1}`
                     ),
                     datasets: [{
-                      data: results.map(row => row[dataAnalysis.numericColumns[0]]),
+                      data: results.map((row: any) => row[dataAnalysis.numericColumns[0]]),
                       backgroundColor: results.map(() => 
                         `hsla(${Math.random() * 360}, 70%, 50%, 0.6)`
                       )
@@ -177,11 +177,11 @@ Text: ${userPrompt}`;
                   type: 'line',
                   config: {
                     labels: dataAnalysis.dateColumns.length > 0 
-                      ? results.map(row => new Date(row[dataAnalysis.dateColumns[0]]).toLocaleDateString())
+                      ? results.map((row: any) => new Date(row[dataAnalysis.dateColumns[0]]).toLocaleDateString())
                       : results.map((_, idx) => `Point ${idx + 1}`),
-                    datasets: dataAnalysis.numericColumns.map(col => ({
+                    datasets: dataAnalysis.numericColumns.map((col: string) => ({
                       label: col,
-                      data: results.map(row => row[col]),
+                      data: results.map((row: any) => row[col]),
                       borderColor: `hsl(${Math.random() * 360}, 70%, 50%)`,
                       tension: 0.1
                     }))
@@ -196,11 +196,11 @@ Text: ${userPrompt}`;
                 type: 'bar',
                 config: {
                   labels: dataAnalysis.stringColumns.length > 0
-                    ? results.map(row => String(row[dataAnalysis.stringColumns[0]]))
+                    ? results.map((row: any) => String(row[dataAnalysis.stringColumns[0]]))
                     : results.map((_, idx) => `Row ${idx + 1}`),
-                  datasets: dataAnalysis.numericColumns.map(col => ({
+                  datasets: dataAnalysis.numericColumns.map((col: string) => ({
                     label: col,
-                    data: results.map(row => row[col]),
+                    data: results.map((row: any) => row[col]),
                     backgroundColor: `hsla(${Math.random() * 360}, 70%, 50%, 0.6)`,
                     borderColor: `hsl(${Math.random() * 360}, 70%, 50%)`,
                     borderWidth: 1