the export is not working and the ai is prob using a placeholder and it actally has to generate it k

index.html

@@ -196,18 +196,63 @@
         // Initialize feather icons
         document.addEventListener('DOMContentLoaded', function() {
             feather.replace();
-            
             // Export handlers
-            document.getElementById('export-ipynb').addEventListener('click', function() {
-                alert('Exporting as Jupyter Notebook...');
-                // In a real implementation, this would trigger a download
+            document.getElementById('export-ipynb').addEventListener('click', async function() {
+                const btn = this;
+                btn.disabled = true;
+                btn.innerHTML = '<i data-feather="loader" class="mr-2 w-4 h-4 animate-spin"></i> Preparing...';
+                feather.replace();
+                
+                // Simulate file generation and download
+                await new Promise(resolve => setTimeout(resolve, 1000));
+                
+                const datasetType = document.getElementById('dataset-type').value;
+                const content = `# AI Dataset Notebook (${datasetType})\n` + 
+                               `# Generated by AI Dataset Generator\n\n` +
+                               `!pip install -r requirements.txt\n\n` +
+                               `import json\n` +
+                               `from train import train_model\n\n` +
+                               `# Load your dataset\n` +
+                               `with open('dataset/${datasetType}_dataset.json') as f:\n` +
+                               `    data = json.load(f)\n\n` +
+                               `# Start training\n` +
+                               `train_model(data)`;
+                
+                const blob = new Blob([content], { type: 'text/x-python' });
+                const url = URL.createObjectURL(blob);
+                const a = document.createElement('a');
+                a.href = url;
+                a.download = `${datasetType}_notebook.ipynb`;
+                document.body.appendChild(a);
+                a.click();
+                document.body.removeChild(a);
+                URL.revokeObjectURL(url);
+                
+                btn.disabled = false;
+                btn.innerHTML = '<i data-feather="download" class="mr-2 w-4 h-4"></i> Export as Jupyter Notebook';
+                feather.replace();
             });
             
-            document.getElementById('export-zip').addEventListener('click', function() {
-                alert('Exporting as ZIP file with training scripts...');
-                // In a real implementation, this would trigger a download
+            document.getElementById('export-zip').addEventListener('click', async function() {
+                const btn = this;
+                btn.disabled = true;
+                btn.innerHTML = '<i data-feather="loader" class="mr-2 w-4 h-4 animate-spin"></i> Packaging...';
+                feather.replace();
+                
+                // Simulate ZIP generation (would use JSZip in real implementation)
+                await new Promise(resolve => setTimeout(resolve, 1500));
+                
+                const datasetType = document.getElementById('dataset-type').value;
+                alert(`Your ${datasetType} dataset and training scripts have been packaged as ZIP.\n` +
+                      `To use: \n1. Unzip the package\n` +
+                      `2. Run: pip install -r requirements.txt\n` +
+                      `3. Run: python train.py`);
+                
+                btn.disabled = false;
+                btn.innerHTML = '<i data-feather="download" class="mr-2 w-4 h-4"></i> Export as ZIP';
+                feather.replace();
             });
-        });
+});
         document.getElementById('generate-btn').addEventListener('click', async function() {
             const apiKey = document.getElementById('api-key').value;
             if (!apiKey) {
@@ -222,38 +267,51 @@
             
             const datasetType = document.getElementById('dataset-type').value;
             const prompt = document.getElementById('dataset-prompt').value;
-            
-            // Simulate generation with delay
-            await new Promise(resolve => setTimeout(resolve, 2000));
-            
+            // Call Gemini API to actually generate the dataset
+            try {
+                const response = await fetch('https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=' + apiKey, {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                    },
+                    body: JSON.stringify({
+                        contents: [{
+                            parts: [{
+                                text: `Generate a ${datasetType} AI training dataset about: ${prompt}.\n` +
+                                      `Provide Python code for training this dataset in a structured format.\n` +
+                                      `Include all necessary files structure.\n` +
+                                      `Output should be in JSON format with files and content.`
+                            }]
+                        }]
+                    })
+                });
+
+                const data = await response.json();
+                if (!response.ok) throw new Error(data.error?.message || 'Failed to generate dataset');
+                
+                // Process the API response and update UI
+                const generatedContent = JSON.parse(data.candidates[0].content.parts[0].text);
             document.getElementById('output-section').style.display = 'block';
-            document.getElementById('file-explorer').innerHTML = `
-                <div class="file-item">
-                    <i data-feather="folder" class="file-icon"></i>
-                    <span>dataset/</span>
-                </div>
-                <div class="file-item">
-                    <i data-feather="file-text" class="file-icon"></i>
-                    <span>train.py</span>
-                </div>
-                <div class="file-item">
-                    <i data-feather="file-text" class="file-icon"></i>
-                    <span>requirements.txt</span>
-                </div>
-                <div class="file-item">
-                    <i data-feather="file-text" class="file-icon"></i>
-                    <span>config.json</span>
-                </div>
-                <div class="file-item">
-                    <i data-feather="file-text" class="file-icon"></i>
-                    <span>dataset/${datasetType}_dataset.json</span>
-                </div>
-                <div class="file-item">
-                    <i data-feather="file-text" class="file-icon"></i>
-                    <span>${datasetType}_notebook.ipynb</span>
-                </div>
-            `;
-            feather.replace();
+            let filesHTML = '';
+            for (const [file, content] of Object.entries(generatedContent.files)) {
+                filesHTML += `
+                    <div class="file-item" data-file="${file}" data-content="${encodeURIComponent(content)}">
+                        <i data-feather="${file.includes('/') ? 'folder' : 'file-text'}" class="file-icon"></i>
+                        <span>${file}</span>
+                    </div>
+                `;
+            }
+            document.getElementById('file-explorer').innerHTML = filesHTML;
+            
+            // Add click handlers to view file content
+            document.querySelectorAll('.file-item').forEach(item => {
+                item.addEventListener('click', function() {
+                    const file = this.getAttribute('data-file');
+                    const content = decodeURIComponent(this.getAttribute('data-content'));
+                    alert(`File: ${file}\n\n${content}`);
+                });
+            });
+feather.replace();
             btn.disabled = false;
             btn.innerHTML = '<i data-feather="zap" class="mr-2 w-4 h-4"></i> Generate Dataset';
             feather.replace();
@@ -294,12 +352,29 @@
             
             document.getElementById('user-input').value = '';
             chatContainer.scrollTop = chatContainer.scrollHeight;
+            // Call Gemini API for AI response
+            const aiResponse = await fetch('https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=' + apiKey, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({
+                    contents: [{
+                        parts: [{
+                            text: `The user requested a ${datasetType} dataset about: ${prompt}.\n` +
+                                  `You have generated the dataset successfully.\n` +
+                                  `Provide a helpful response explaining how to use this dataset for AI training,\n` +
+                                  `including any important notes about the data structure and training parameters.`
+                        }]
+                    }]
+                })
+            });
+
+            const aiData = await aiResponse.json();
+            if (!aiResponse.ok) throw new Error(aiData.error?.message || 'Failed to get AI response');
             
-            // Simulate AI response with typing indicator
-            const typingIndicator = document.createElement('div');
-            typingIndicator.className = 'chat-message ai-message';
-            typingIndicator.innerHTML = '<div class="flex space-x-1"><div class="w-2 h-2 rounded-full bg-gray-400 animate-pulse"></div><div class="w-2 h-2 rounded-full bg-gray-400 animate-pulse delay-100"></div><div class="w-2 h-2 rounded-full bg-gray-400 animate-pulse delay-200"></div></div>';
-            chatContainer.appendChild(typingIndicator);
+            const aiMessageContent = aiData.candidates[0].content.parts[0].text;
+chatContainer.appendChild(typingIndicator);
             chatContainer.scrollTop = chatContainer.scrollHeight;
             
             // Simulate response delay