#21 Api /llama/{prompt} returns ollama text

expiriment/main.go

 package main
 
 import (
+	"fmt"
 	"log"
+	"net/http"
 	"os/exec"
+	"regexp"
 	"runtime"
 )
 
 const prompt = "Hello!"
 
 func main() {
+	runHandler()
+}
 
-	var cmd = getOllamaCmd(prompt)
+func removeANSI(s string) string {
+	re := regexp.MustCompile(`\x1b\[[0-?]*[ -/]*[@-~]`)
+	return re.ReplaceAllString(s, "")
+}
+
+func promptHandler(w http.ResponseWriter, r *http.Request) {
+	// get the value of the "prompt" query parameter
+	prompt := r.PathValue("prompt")
+	log.Println("Prompt: ", prompt)
 
+	// Run ollama prompt
+	var cmd = getOllamaCmd(prompt)
 	output, err := cmd.CombinedOutput()
+
 	log.Printf("Output: %s, Error: %v", output, err)
+
+	finalOutput := removeANSI(string(output))
+	finalOutput = fmt.Sprintf("%s", finalOutput)
+	//outputString := string(output)
+	//_, err = fmt.Fprintln(w, output)
+	_, err = w.Write([]byte(finalOutput))
+	if err != nil {
+		log.Println("Error writing response: ", err)
+		return
+	}
+}
+
+func runHandler() {
+	http.HandleFunc("/llama/{prompt}", promptHandler)
+
+	log.Println("Starting server on :8080")
+
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		_, err := w.Write([]byte("To ask the llama a question, make a GET request to /llama/{prompt}"))
+		if err != nil {
+			log.Println("Error writing response: ", err)
+			return
+		}
+	})
+
+	log.Fatal(http.ListenAndServe(":8080", nil))
 }
 
 func getCommand() *exec.Cmd {