Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
seawolf2357
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,6 @@ import os
|
|
5 |
from datetime import datetime, timedelta
|
6 |
from huggingface_hub import InferenceClient
|
7 |
|
8 |
-
from googletrans import Translator
|
9 |
|
10 |
|
11 |
API_KEY = os.getenv("SERPHOUSE_API_KEY")
|
@@ -120,29 +119,40 @@ COUNTRY_LOCATIONS = {
|
|
120 |
MAJOR_COUNTRIES = list(COUNTRY_LOCATIONS.keys())
|
121 |
|
122 |
|
|
|
|
|
123 |
def translate_query(query, country):
|
124 |
try:
|
125 |
-
# ๋ฒ์ญ๊ธฐ ์ด๊ธฐํ
|
126 |
-
translator = Translator()
|
127 |
-
|
128 |
# ์์ด ์
๋ ฅ์ ๋ชจ๋ ๊ตญ๊ฐ์ ๋ํด ๊ทธ๋๋ก ์ฌ์ฉ
|
129 |
if is_english(query):
|
130 |
print(f"์์ด ๊ฒ์์ด ๊ฐ์ง - ์๋ณธ ์ฌ์ฉ: {query}")
|
131 |
return query
|
132 |
|
|
|
|
|
|
|
|
|
|
|
133 |
# ์ ํ๋ ๊ตญ๊ฐ๊ฐ ๋ฒ์ญ ์ง์ ๊ตญ๊ฐ์ธ ๊ฒฝ์ฐ
|
134 |
if country in COUNTRY_LANGUAGES:
|
135 |
target_lang = COUNTRY_LANGUAGES[country]
|
136 |
|
137 |
-
#
|
138 |
-
|
139 |
-
|
140 |
-
|
141 |
-
|
142 |
-
|
143 |
-
|
144 |
-
|
145 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
146 |
|
147 |
return query
|
148 |
|
|
|
5 |
from datetime import datetime, timedelta
|
6 |
from huggingface_hub import InferenceClient
|
7 |
|
|
|
8 |
|
9 |
|
10 |
API_KEY = os.getenv("SERPHOUSE_API_KEY")
|
|
|
119 |
MAJOR_COUNTRIES = list(COUNTRY_LOCATIONS.keys())
|
120 |
|
121 |
|
122 |
+
|
123 |
+
|
124 |
def translate_query(query, country):
|
125 |
try:
|
|
|
|
|
|
|
126 |
# ์์ด ์
๋ ฅ์ ๋ชจ๋ ๊ตญ๊ฐ์ ๋ํด ๊ทธ๋๋ก ์ฌ์ฉ
|
127 |
if is_english(query):
|
128 |
print(f"์์ด ๊ฒ์์ด ๊ฐ์ง - ์๋ณธ ์ฌ์ฉ: {query}")
|
129 |
return query
|
130 |
|
131 |
+
# South Korea ์ ํ์ ํ๊ธ ์
๋ ฅ์ ๊ทธ๋๋ก ์ฌ์ฉ
|
132 |
+
if country == "South Korea" and is_korean(query):
|
133 |
+
print(f"ํ๊ตญ์ด ๊ฒ์์ด - ์๋ณธ ์ฌ์ฉ: {query}")
|
134 |
+
return query
|
135 |
+
|
136 |
# ์ ํ๋ ๊ตญ๊ฐ๊ฐ ๋ฒ์ญ ์ง์ ๊ตญ๊ฐ์ธ ๊ฒฝ์ฐ
|
137 |
if country in COUNTRY_LANGUAGES:
|
138 |
target_lang = COUNTRY_LANGUAGES[country]
|
139 |
|
140 |
+
# Google Translate API URL
|
141 |
+
url = f"https://translate.googleapis.com/translate_a/single"
|
142 |
+
|
143 |
+
params = {
|
144 |
+
"client": "gtx",
|
145 |
+
"sl": "ko", # ์์ค ์ธ์ด: ํ๊ตญ์ด
|
146 |
+
"tl": target_lang, # ๋์ ์ธ์ด
|
147 |
+
"dt": "t",
|
148 |
+
"q": query
|
149 |
+
}
|
150 |
+
|
151 |
+
response = requests.get(url, params=params)
|
152 |
+
translated_text = response.json()[0][0][0]
|
153 |
+
|
154 |
+
print(f"๋ฒ์ญ ๊ฒฐ๊ณผ: {query} -> {translated_text} ({country})")
|
155 |
+
return translated_text
|
156 |
|
157 |
return query
|
158 |
|