DEMO UI (#2)

* ui-test

* ui-test

* ui-test

* ui-test

* ui-test

* ui-test

* ui-test

* ui-test

* test

* update ui

* ui-test

* Update server.go

* Update queryClient.go

* Update ui.html

* self-provision url

* Update Demo UI

---------

Co-authored-by: Lorenzo Mangani <lmangani@Raspberry-M1.local>
Co-authored-by: lmangani <>

Author: lmangani

server.go

@@ -2,7 +2,7 @@
 package main
 
 import (
-        _ "embed"
+	_ "embed"
 	"encoding/json"
 	"flag"
 	"fmt"
@@ -13,7 +13,6 @@ import (
 	"time"
 )
 
-
 //go:embed ui.html
 var uiContent []byte
 
@@ -51,8 +50,24 @@ type ErrorResponse struct {
 	Error string `json:"error"`
 }
 
+// addCORSHeaders adds CORS headers to the response
+func addCORSHeaders(w http.ResponseWriter) {
+	w.Header().Set("Access-Control-Allow-Origin", "*")
+	w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
+	w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
+}
+
 // handleQuery handles the /query endpoint
 func (s *Server) handleQuery(w http.ResponseWriter, r *http.Request) {
+	// Add CORS headers
+	addCORSHeaders(w)
+
+	// Handle preflight requests
+	if r.Method == http.MethodOptions {
+		w.WriteHeader(http.StatusOK)
+		return
+	}
+
 	// Only allow POST
 	if r.Method != http.MethodPost {
 		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
@@ -140,20 +155,22 @@ func sendErrorResponse(w http.ResponseWriter, message string, statusCode int) {
 
 // Health check endpoint
 func (s *Server) handleHealth(w http.ResponseWriter, r *http.Request) {
+	// Add CORS headers
+	addCORSHeaders(w)
+
+	// Handle preflight requests
+	if r.Method == http.MethodOptions {
+		w.WriteHeader(http.StatusOK)
+		return
+	}
+
 	w.Header().Set("Content-Type", "application/json")
 	json.NewEncoder(w).Encode(map[string]interface{}{
 		"status":    "ok",
 		"timestamp": time.Now().Format(time.RFC3339),
 	})
 }
 
-// addCORSHeaders adds CORS headers to the response
-func addCORSHeaders(w http.ResponseWriter) {
-	w.Header().Set("Access-Control-Allow-Origin", "*")
-	w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
-	w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
-}
-
 // handleUI serves the main UI page
 func (s *Server) handleUI(w http.ResponseWriter, r *http.Request) {
 	// Add CORS headers
@@ -229,12 +246,15 @@ func main() {
 	}
 	defer server.Close()
 
+	// Create a new mux for routing
+	mux := http.NewServeMux()
+
 	// Set up routes
-        http.HandleFunc("/", server.handleUI) 
-	http.HandleFunc("/health", server.handleHealth)
-	http.HandleFunc("/query", server.handleQuery)
+	mux.HandleFunc("/", server.handleUI)  // Serve UI at root path
+	mux.HandleFunc("/health", server.handleHealth)
+	mux.HandleFunc("/query", server.handleQuery)
 
 	// Start server
 	log.Printf("GigAPI server running at http://localhost:%s", port)
-	log.Fatal(http.ListenAndServe(":"+port, nil))
+	log.Fatal(http.ListenAndServe(":"+port, mux))
 }