add main and test code container for visualization + model selection for the user

src/modules/extraction/extract.go

@@ -8,7 +8,7 @@ import (
 
 // var GoPrompt = "The code should be in the Go programming language. There should also be 3 robust test cases within the same file, these test cases should use 'testing' module. There should also be a main function inside of which the execution of the implemented function takes place. Please always provide the source code and no further explanation, The format should be ```go <yourcode + testcases> ```"
 
-var GoPrompt = "The code should be in the Go programming language. You should generate two files. One file should contain the main code (main function plus a function that performs the required job) and another file should contain 3 robust test cases. Please always provide the source codes and no further explanation, The format should be ```go main code``` and ``go testcode```"
+var GoPrompt = "The code should be in the Go programming language. You should generate two files. One file should contain the main code (main function plus a function that performs the required job) and another file should contain 3 robust test cases. Please note that only use main package in both files. Please always provide the source codes and no further explanation, The format should be ```go main code``` and ``go testcode```"
 
 var RustPrompt = "The code should be in the Rust programming language. There should also be 3 robust test cases within the same code. There should also be a main function inside of which all the execution takes place. Please only provide the source code and no further explanation, The format should be ```rust <yourcode + testcases> ```"
 

src/templates/index.html

@@ -331,16 +331,16 @@
 .center-wrapper {
   display: flex;
   justify-content: center;
-  align-items: center;
-  width: 100%;
-  height: 100vh; /* Center vertically on the screen */
+  align-items: flex-start; /* Align items at the top to prevent truncation */
   gap: 50px; /* Space between code display and main container */
+  padding: 20px; /* Add padding to prevent clipping near screen edges */
+  overflow-y: auto; /* Enable scrolling if content exceeds window height */
 }
 
 /* Code Display Container with White Background and Blue Border */
 .code-display-container {
   width: 700px;
-  max-height: 80vh; /* Allow it to grow, but with max height */
+  max-height: 100vh; /* Allow it to grow, but with max height */
   overflow-y: auto; /* Enable scrolling if content exceeds */
   background-color: #ffffff;
   border: 8px solid #61dafb;
@@ -387,7 +387,7 @@
   padding: 1em;
   border-radius: 5px;
   font-size: 0.95em;
-  max-height: 250px; /* Increased height */
+  max-height: 400px; /* Increased height */
   overflow-y: auto;
   border: 1px solid #ddd;
   white-space: pre-wrap;