...

cmd/heroagent/main.go

@@ -7,11 +7,13 @@ import (
 	"os"
 
 	"git.ourworld.tf/herocode/heroagent/pkg/heroagent"
+	"git.ourworld.tf/herocode/heroagent/pkg/servers/ui" // Import the new UI package
 )
 
 func main() {
 	// Parse command-line flags
-	portFlag := flag.String("port", "", "Port to run the server on")
+	portFlag := flag.String("port", "", "Port to run the HeroLauncher on")
+	uiPortFlag := flag.String("uiport", "3000", "Port to run the UI server on") // New flag for UI port
 	flag.Parse()
 
 	// Initialize HeroLauncher with default configuration
@@ -30,7 +32,20 @@ func main() {
 	// Create HeroLauncher instance
 	launcher := heroagent.New(config)
 
-	// Start the server
+	// Initialize and start the UI server in a new goroutine
+	go func() {
+		uiApp := ui.NewApp(ui.AppConfig{}) // Assuming default AppConfig is fine
+		uiPort := *uiPortFlag
+		if envUiPort := os.Getenv("UIPORT"); envUiPort != "" {
+			uiPort = envUiPort
+		}
+		fmt.Printf("Starting UI server on port %s...\n", uiPort)
+		if err := uiApp.Listen(":" + uiPort); err != nil {
+			log.Printf("Failed to start UI server: %v", err) // Use Printf to not exit main app
+		}
+	}()
+
+	// Start the main HeroLauncher server
 	fmt.Printf("Starting HeroLauncher on port %s...\n", config.Port)
 	if err := launcher.Start(); err != nil {
 		log.Fatalf("Failed to start HeroLauncher: %v", err)