From 44b6c2915f4d89420237d6903784bae3d4862650 Mon Sep 17 00:00:00 2001
From: Vincent Yang <missuo@pm.me>
Date: Tue, 3 Dec 2024 11:23:56 -0500
Subject: [PATCH] fix: while sourceLang is auto

---
 go.mod                 | 1 +
 go.sum                 | 2 ++
 translate/translate.go | 8 +++++---
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/go.mod b/go.mod
index 32ba855d..d2066593 100644
--- a/go.mod
+++ b/go.mod
@@ -3,6 +3,7 @@ module github.com/OwO-Network/DeepLX
 go 1.23.3
 
 require (
+	github.com/abadojack/whatlanggo v1.0.1
 	github.com/andybalholm/brotli v1.1.0
 	github.com/gin-contrib/cors v1.6.0
 	github.com/gin-gonic/gin v1.9.1
diff --git a/go.sum b/go.sum
index 8bf90ac8..a856c218 100644
--- a/go.sum
+++ b/go.sum
@@ -1,3 +1,5 @@
+github.com/abadojack/whatlanggo v1.0.1 h1:19N6YogDnf71CTHm3Mp2qhYfkRdyvbgwWdd2EPxJRG4=
+github.com/abadojack/whatlanggo v1.0.1/go.mod h1:66WiQbSbJBIlOZMsvbKe5m6pzQovxCH9B/K8tQB2uoc=
 github.com/andybalholm/brotli v1.1.0 h1:eLKJA0d02Lf0mVpIDgYnqXcUn0GqVmEFny3VuID1U3M=
 github.com/andybalholm/brotli v1.1.0/go.mod h1:sms7XGricyQI9K10gOSf56VKKWS4oLer58Q+mhRPtnY=
 github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
diff --git a/translate/translate.go b/translate/translate.go
index 1e7a8d5d..94ef6024 100644
--- a/translate/translate.go
+++ b/translate/translate.go
@@ -2,7 +2,7 @@
  * @Author: Vincent Young
  * @Date: 2024-09-16 11:59:24
  * @LastEditors: Vincent Yang
- * @LastEditTime: 2024-11-01 23:19:11
+ * @LastEditTime: 2024-12-03 11:23:23
  * @FilePath: /DeepLX/translate/translate.go
  * @Telegram: https://t.me/missuo
  * @GitHub: https://github.com/missuo
@@ -15,12 +15,14 @@ package translate
 import (
 	"bytes"
 	"fmt"
-	"github.com/imroc/req/v3"
 	"io"
 	"net/http"
 	"net/url"
 	"strings"
 
+	"github.com/abadojack/whatlanggo"
+	"github.com/imroc/req/v3"
+
 	"github.com/andybalholm/brotli"
 	"github.com/tidwall/gjson"
 )
@@ -135,7 +137,7 @@ func TranslateByDeepLX(sourceLang, targetLang, text string, tagHandling string,
 
 	// Get detected language if source language is auto
 	if sourceLang == "auto" || sourceLang == "" {
-		sourceLang = strings.ToLower(splitResult.Get("result.lang.detected").String())
+		sourceLang = strings.ToUpper(whatlanggo.DetectLang(text).Iso6391())
 	}
 
 	// Prepare jobs from split result