psicologo.go

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"strings"
	"text/tabwriter"
	"time"
)

type Message struct {
	Content   string `json:"content"`
	ID        string `json:"id"`
	Role      string `json:"role"`
	CreatedAt string `json:"createdAt"`
}

type RequestBody struct {
	Messages          []Message              `json:"messages"`
	PreviewToken      interface{}            `json:"previewToken"`
	CodeModelMode     bool                   `json:"codeModelMode"`
	AgentMode         AgentMode              `json:"agentMode"`
	TrendingAgentMode map[string]interface{} `json:"trendingAgentMode"`
	IsMicMode         bool                   `json:"isMicMode"`
	MaxTokens         int                    `json:"maxTokens"`
}

type AgentMode struct {
	Mode bool   `json:"mode"`
	ID   string `json:"id"`
}

func getID() string {
	return fmt.Sprintf("%d", time.Now().UnixNano())
}

func getMessage(content string, role string) Message {
	return Message{
		Content:   content,
		ID:        getID(),
		Role:      role,
		CreatedAt: time.Now().Format(time.RFC3339),
	}
}

func typingAnimation(text string, delay time.Duration) {
	for _, char := range text {
		fmt.Print(string(char))
		time.Sleep(delay)
	}
	fmt.Println()
}

func wrapText(text string, width int) string {
	var buf bytes.Buffer
	w := tabwriter.NewWriter(&buf, 0, 0, 1, ' ', 0)
	fmt.Fprintln(w, text)
	w.Flush()
	return buf.String()
}

func main() {
	if len(os.Args) < 2 {
		fmt.Println("Por favor, ingresa una pregunta.")
		return
	}

	pregunta := os.Args[1]
	url := "https://www.blackbox.ai/api/chat"

	messages := []Message{
		getMessage("Habla en español, y no uses para los textos * o #", "user"),
		getMessage("Claro, puedo hablar en español. ¿En qué te puedo ayudar hoy?", "assistant"),
		getMessage(pregunta, "user"),
	}

	requestBody := RequestBody{
		Messages:      messages,
		PreviewToken:  nil,
		CodeModelMode: true,
		AgentMode: AgentMode{
			Mode: true,
			ID:   "PsicologocGVjQgS",
		},
		TrendingAgentMode: map[string]interface{}{},
		IsMicMode:         false,
		MaxTokens:         512,
	}

	jsonData, err := json.Marshal(requestBody)
	if err != nil {
		fmt.Println("Error al convertir a JSON:", err)
		return
	}

	resp, err := http.Post(url, "application/json", bytes.NewBuffer(jsonData))
	if err != nil {
		fmt.Println("Error al realizar la solicitud:", err)
		return
	}
	defer resp.Body.Close()

	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println("Error al leer la respuesta:", err)
		return
	}

	responseStr := string(body)

	parts := strings.Split(responseStr, "$@$")
	if len(parts) > 2 {
		responseStr = parts[2]
	} else {
		responseStr = responseStr
	}

	typingAnimation(wrapText(responseStr, 20), 13*time.Millisecond)
}