Enha: server update

1 files changed, 2 insertions, 5 deletions

diff --git a/main.go b/main.go
index 7766a60..a95db19 100644
--- a/main.go
+++ b/main.go
@@ -3,7 +3,6 @@ package main
 import (
 	"flag"
 	"fmt"
-	"net/http"
 	"unicode"
 
 	"github.com/rivo/tview"
@@ -30,10 +29,8 @@ func main() {
 	apiPort := flag.Int("port", 0, "port to host api")
 	flag.Parse()
 	if apiPort != nil && *apiPort > 3000 {
-		// start api server
-		http.HandleFunc("POST /completion", completion)
-		http.ListenAndServe(fmt.Sprintf(":%d", *apiPort), nil)
-		// no tui
+		srv := Server{}
+		srv.ListenToRequests(fmt.Sprintf("%d", *apiPort))
 		return
 	}
 	pages.AddPage("main", flex, true, true)