Added more editable parameters

backend/api/templates/index.html

@@ -49,7 +49,6 @@
 
     <!-- Sidebar on the right -->
     <div class="w-1/4 bg-white shadow p-6">
-        <h2 class="text-xl font-bold mb-4">Options</h2>
         <div class="flex flex-col space-y-4">
             <label class="block">
                 <span class="text-gray-700">Top K:</span>
@@ -61,6 +60,15 @@
                     class="mt-1 p-2 w-full border rounded"
                 />
             </label>
+            <label class="flex items-center space-x-2">
+                <input
+                    type="checkbox"
+                    id="add_to_dataset"
+                    name="add_to_dataset"
+                    class="form-checkbox text-green-600 focus:ring-green-500"
+                />
+                <span>Add to Dataset</span>
+            </label>
             <label class="flex items-center space-x-2">
                 <input
                     type="checkbox"
@@ -89,6 +97,39 @@
                 <span>Induce Hallucination</span>
             </label>
         </div>
+        <div class="my-4 border-t border-gray-300"></div>
+        <div class="flex flex-col space-y-4">
+            <label class="block">
+                <span class="text-gray-700">Project:</span>
+                <input
+                    type="text"
+                    id="project_name"
+                    name="project_name"
+                    value="{{ default_project_name }}"
+                    class="mt-1 p-2 w-full border rounded"
+                />
+            </label>
+            <label class="block">
+                <span class="text-gray-700">Logstream:</span>
+                <input
+                    type="text"
+                    id="logstream_name"
+                    name="logstream_name"
+                    value="{{ default_logstream_name }}"
+                    class="mt-1 p-2 w-full border rounded"
+                />
+            </label>
+            <label class="block">
+                <span class="text-gray-700">Dataset:</span>
+                <input
+                    type="text"
+                    id="dataset_name"
+                    name="dataset_name"
+                    value="{{ default_dataset_name }}"
+                    class="mt-1 p-2 w-full border rounded"
+                />
+            </label>
+        </div>
     </div>
 
     <script>
@@ -102,6 +143,15 @@
             if (urlParams.has('top_k')) {
                 $('#top_k').val(urlParams.get('top_k'));
             }
+            if (urlParams.has('dataset_name')) {
+                $('#dataset_name').val(urlParams.get('dataset_name'));
+            }
+            if (urlParams.has('logstream_name')) {
+                $('#logstream_name').val(urlParams.get('logstream_name'));
+            }
+            if (urlParams.has('project_name')) {
+                $('#project_name').val(urlParams.get('project_name'));
+            }
             if (urlParams.has('protection')) {
                 $('#protection').prop('checked', urlParams.get('protection') === 'true');
             }
@@ -117,10 +167,13 @@
 
                 const query = $('#query').val();
                 const top_k = $('#top_k').val();
+                const add_to_dataset = $('#add_to_dataset').is(':checked');
                 const protection = $('#protection').is(':checked');
                 const hallucination_detection = $('#hallucination_detection').is(':checked');
                 const induce_hallucination = $('#induce_hallucination').is(':checked');
-
+                const project_name = $('#project_name').val();
+                const dataset_name = $('#dataset_name').val();
+                const logstream_name = $('#logstream_name').val();
                 // Show loading spinner
                 $('#loadingSpinner').removeClass('hidden');
 
@@ -134,9 +187,13 @@
                     data: {
                         query: query,
                         top_k: top_k,
+                        add_to_dataset: add_to_dataset,
                         protection: protection,
                         hallucination_detection: hallucination_detection,
-                        induce_hallucination: induce_hallucination
+                        induce_hallucination: induce_hallucination,
+                        project_name: project_name,
+                        logstream_name: logstream_name,
+                        dataset_name: dataset_name,
                     },
                     success: function (response) {
                         // Hide loading spinner
@@ -205,6 +262,7 @@
                                                 <style>
                                                     pii {
                                                         font-weight: bold;
+                                                        text-decoration: line-through;
                                                         background-color: yellow;
                                                     }
                                                 </style>
@@ -340,19 +398,24 @@
             $(document).on('click', '#retry_button', function() {
                 const query = $('#query').val();
                 const retry_top_k = $('#retry_top_k').val();
+                const project_name = $('#project_name').val();
+                const dataset_name = $('#dataset_name').val();
+                const logstream_name = $('#logstream_name').val();
+                const add_to_dataset = $('#add_to_dataset').is(':checked');
                 const protection = $('#protection').is(':checked');
                 const hallucination_detection = $('#hallucination_detection').is(':checked');
                 const induce_hallucination = $('#induce_hallucination').is(':checked');
-                
                 // Create URL parameters to reload with form pre-filled
                 const params = new URLSearchParams();
                 params.set('query', query);
                 params.set('top_k', retry_top_k);
                 params.set('retry', 'true'); // Flag to indicate this is a retry attempt
+                if (project_name) params.set('project_name', project_name);
+                if (logstream_name) params.set('logstream_name', logstream_name);
+                if (add_to_dataset) params.set('add_to_dataset', 'true');
                 if (protection) params.set('protection', 'true');
                 if (hallucination_detection) params.set('hallucination_detection', 'true');
                 if (induce_hallucination) params.set('induce_hallucination', 'true');
-                
                 // Reload the page with parameters
                 window.location.href = window.location.pathname + '?' + params.toString();
             });
@@ -365,4 +428,4 @@
     </script>
 
 </body>
-</html>
+</html>