package main

import (
	"fmt"
	"log"
	"wasm-go/internal/api"
)

func main() {
	fmt.Println("=== 企业行业分类示例 ===")

	// 创建LLM客户端 - 这里使用Ollama作为示例
	// 如果Ollama不可用，也可以使用OpenAI
	var llmClient *api.LLMClient

	// 尝试连接到Ollama
	fmt.Println("尝试连接到Ollama服务...")
	llmClient = api.NewOllamaClient("http://localhost:11434", "qwen3.5:9b")

	// 如果需要使用OpenAI，取消下面的注释并替换API KEY
	/*
		llmClient = api.NewOpenAIClient("your-openai-api-key", "gpt-3.5-turbo")
	*/

	// 创建企业行业分类器
	classifier := api.NewCompanyIndustryClassifier(llmClient)

	// 示例公司名称 - 可以从命令行参数获取
	companyName := "深圳市致远速联信息技术有限公司"

	fmt.Printf("正在分析公司: %s\n", companyName)

	// 分析企业行业
	result, err := classifier.ClassifyCompany(companyName)
	if err != nil {
		log.Printf("分析失败: %v", err)
		return
	}

	// 输出结果
	fmt.Printf("\n✅ 企业行业分析完成!\n")
	fmt.Printf("公司名称: %s\n", result.CompanyName)
	fmt.Printf("主要行业: %s - %s\n", result.Primary.Code, result.Primary.Name)
	fmt.Printf("判断依据: %s\n", result.Primary.Reason)

	if len(result.Secondary) > 0 {
		fmt.Printf("次要行业:\n")
		for _, sec := range result.Secondary {
			fmt.Printf("  %s - %s\n", sec.Code, sec.Name)
			fmt.Printf("  判断依据: %s\n", sec.Reason)
		}
	} else {
		fmt.Printf("次要行业: 无\n")
	}
}