From 4cca1ee261ac761c73822b2f8b9d33f53d1b84ee Mon Sep 17 00:00:00 2001
From: The Ghost of FOSS' Future
 <163201376+sneedgroup-holder@users.noreply.github.com>
Date: Tue, 17 Sep 2024 07:08:30 -0500
Subject: [PATCH] Update index.html

---
 index.html | 96 ++++++++++++++++++++++++++++++------------------------
 1 file changed, 54 insertions(+), 42 deletions(-)
diff --git a/index.html b/index.html
index 6639ee8..cced366 100644
--- a/index.html
+++ b/index.html
@@ -2,80 +2,92 @@
 <html>
 <head>
   <title>SneedPage</title>
+  <style>
+    #preview {
+      width: 100%;
+      height: 400px; /* Adjust height as needed */
+      border: 1px solid #ccc;
+    }
+  </style>
 </head>
 <body>
   <textarea id="prompt" rows="10" cols="50"></textarea>
-  <button onclick="generateWebsite()">Prompt</button>
-  <div id="preview" width="100%" height="32%"></div>
+  <button onclick="generateWebsite()">Generate</button>
+  <iframe id="preview"></iframe>
   <button onclick="saveHTML()">Save HTML</button>
 
   <script>
-    let currentHTML;
+    let currentHTML = '';
+
     function generateWebsite() {
       const prompt = document.getElementById("prompt").value;
-      // Replace with your Ollama API call and processing logic
       const generatedHTML = generateHTMLFromPrompt(prompt);
 
       // Load the generated HTML into the iframe
       const iframe = document.getElementById("preview");
-      iframe.contentDocument.open();
-      iframe.contentDocument.write(generatedHTML);
-      iframe.contentDocument.close();
+      const iframeDoc = iframe.contentWindow.document;
+      iframeDoc.open();
+      iframeDoc.write(generatedHTML);
+      iframeDoc.close();
     }
 
     function saveHTML() {
-      const generatedHTML = currentHTML;
-      const blob = new Blob([generatedHTML], { type: "text/html" });
+      if (!currentHTML) {
+        alert("No HTML to save. Generate content first.");
+        return;
+      }
+
+      const blob = new Blob([currentHTML], { type: "text/html" });
       const a = document.createElement("a");
       a.href = URL.createObjectURL(blob);
       a.download = "generated_website.html";
       a.click();
     }
 
-    // Replace this with your Ollama API integration and HTML generation logic
-    let answerCount = 0
-    let body;
     function generateHTMLFromPrompt(prompt) {
       const url = "https://ollama-api.nodemixaholic.com/api/generate";
-      if (answercount < 1) {
+      const answerCount = 0; // Assuming you want this to start at 0
+
+      if (answerCount < 1) {
         prompt += `
 Create this in HTML, CSS, and Javascript as a single-page application with dynamically loading content.
-Do NOT use external libraries.`
-    } else {
-      prompt += `
-
+Do NOT use external libraries.`;
+      } else {
+        prompt += `
 See that? Yeah do that to the following website code:
 
 ${currentHTML}
 
 Be sure to use HTML, CSS, and Javascript and make it a single-page application with dynamically loading content.
 Do NOT use external libraries.
-Fix any bugs as well.`
-    }
+Fix any bugs as well.`;
+      }
 
-    body = JSON.stringify({
-      model: "starcoder2:7b",
-      prompt: prompt // User's prompt from the text area
-    });
-    fetch(url, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json"
-      },
-      body: body
-    })
-    .then(response => response.json())
-    .then(data => {
-      // Display the generated website code in a designated area (e.g., pre tag)
-      currentHTML = data.response;
-      document.getElementById("preview").innerHTML = currentHTML;
-      answerCount++
-      return data.response;
-    })
-    .catch(error => {
-      console.error(error);
-      // Handle API errors and display an error message to the user
-    });
+      const body = JSON.stringify({
+        model: "starcoder2:7b",
+        prompt: prompt // User's prompt from the text area
+      });
+
+      return fetch(url, {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json"
+        },
+        body: body
+      })
+      .then(response => response.json())
+      .then(data => {
+        // Update the currentHTML and display it in the iframe
+        currentHTML = data.response;
+        document.getElementById("preview").contentWindow.document.open();
+        document.getElementById("preview").contentWindow.document.write(currentHTML);
+        document.getElementById("preview").contentWindow.document.close();
+        return data.response;
+      })
+      .catch(error => {
+        console.error('Error:', error);
+        // Handle API errors and display an error message to the user
+      });
     }
   </script>
 </body>
