thatll do

main.go

@@ -1,19 +1,124 @@
 package main
 
 import (
+	"bytes"
 	"context"
+	"encoding/json"
+	"fmt"
+	"html/template"
+	"os"
 	"os/signal"
+	"strings"
 	"syscall"
+
+	"google.golang.org/genai"
 )
 
 func main() {
 	ctx, can := signal.NotifyContext(context.Background(), syscall.SIGINT)
 	defer can()
-	if err := run(ctx); err != nil && ctx.Err() == nil {
+	if err := run(ctx, os.Args[1]); err != nil && ctx.Err() == nil {
 		panic(err)
 	}
 }
 
-func run(ctx context.Context) error {
+func run(ctx context.Context, city string) error {
+	city = strings.Title(city)
+
+	client, err := genai.NewClient(ctx, nil)
+	if err != nil {
+		return err
+	}
+
+	descriptors := []string{
+		`Literally poison in the air that politicians are adamant against doing anything to mitigate`,
+		`Drought worst-case`,
+		`Wildfire risk`,
+		`Hurricane`,
+		`Earthquake`,
+		`Tornado`,
+		`Crime`,
+		`Consistent Air Pollution`,
+		`Transient Air Pollution (from wildfires, etc)`,
+		`Colorado river distribution conflict / reduced capacity`,
+		`PFAS`,
+		`Homelessness`,
+		`Neighbors disappointed in church attendance`,
+		`Hot`,
+		`Personal gripes`,
+		`Alcohol universal`,
+		`Few traffic destination hubs`,
+		`Popularity of booze`,
+		`Tourist destination (eg Branson-like)`,
+		`Weed universal`,
+		`Greater cost of living than Utah`,
+		`Traffic >= Bend at 3k/sqmi`,
+		`Lots of schools`,
+		`Coffee shops`,
+		`Yummy resturaunts`,
+		`Board game stores`,
+		`Home for < 400k`,
+		`20min drive to fun`,
+		`Would we get out and be active? Bike++`,
+		`Close airport`,
+		`DnD Players`,
+		`Trees!`,
+		`Population is college educated`,
+		`Home for < 600k`,
+		`Software developers per capita`,
+		`Fiber-speed internet`,
+		`Organic Groceries`,
+		`Close hospital for emergency situations`,
+		`Home for < 800`,
+		`Hematologist reasonably close`,
+		`Does it have the vibes?`,
+		`Concrete Jungle`,
+	}
+
+	tmpl, err := template.New("tmpl").Parse(`
+For the city {{.City}}, answer the following descriptors '1' for true or '0' for false. Output only a JSON object mapping descriptors to answers.
+
+{{ range .Descriptors -}}
+* {{ . }}
+{{ end }}
+   `)
+	if err != nil {
+		return err
+	}
+	w := bytes.NewBuffer(nil)
+	if err := tmpl.Execute(w, map[string]any{
+		"City":        city,
+		"Descriptors": descriptors,
+	}); err != nil {
+		return err
+	}
+	//panic(w.String())
+
+	result, err := client.Models.GenerateContent(
+		ctx,
+		"gemini-2.5-flash",
+		genai.Text(w.String()),
+		nil,
+	)
+	if err != nil {
+		return err
+	}
+	text := "{" + strings.Split(strings.Split(result.Text(), "{")[1], "}")[0] + "}"
+
+	var m map[string]any
+	if err := json.Unmarshal([]byte(text), &m); err != nil {
+		return err
+	}
+	for _, d := range descriptors {
+		if _, ok := m[d]; !ok {
+			return fmt.Errorf("desc %q missing", d)
+		}
+	}
+
+	fmt.Println(text)
+	for _, d := range descriptors {
+		fmt.Println(m[d])
+	}
+
 	return ctx.Err()
 }