Update app.py

app.py

@@ -105,22 +105,26 @@ MAJOR_COUNTRIES = list(COUNTRY_LOCATIONS.keys())
 def translate_query(query, country):
     try:
         if country in COUNTRY_LANGUAGES:
+            # 입력 검색어 길이 제한
+            query = query[:100]  # 입력 검색어 길이 제한
+            
             target_lang = COUNTRY_LANGUAGES[country]
             prompt = f"Translate only this text to {target_lang} language without any explanation. Output only translated text: {query}"
             
             translated = hf_client.text_generation(
                 prompt,
-                max_new_tokens=100,
+                max_new_tokens=50,
                 temperature=0.1
             )
-            translated = translated.strip()
+            # 번역된 텍스트 정제 및 길이 제한
+            translated = translated.strip()[:255]  # API 제한에 맞춰 255자로 제한
             print(f"Original query: {query}")
             print(f"Translated query: {translated}")
             return translated
-        return query
+        return query[:255]  # 번역하지 않는 경우에도 길이 제한
     except Exception as e:
         print(f"Translation error: {str(e)}")
-        return query
+        return query[:255]
 
 def search_serphouse(query, country, page=1, num_result=10):
     url = "https://api.serphouse.com/serp/live"
@@ -135,19 +139,14 @@ def search_serphouse(query, country, page=1, num_result=10):
             "q": translated_query,
             "domain": "google.com",
             "loc": COUNTRY_LOCATIONS.get(country, "United States"),
-            "lang": "en",
+            "lang": COUNTRY_LANGUAGES.get(country, "en"),
             "device": "desktop",
-            "serp_type": "web",  # web으로 변경
+            "serp_type": "news",
             "page": "1",
-            "num_result": "10",  # num_result로 변경
-            "verbatim": "0",
-            "gfilter": "0"
+            "num": "10"  # num으로 변경
         }
     }
 
-    # API 요청 디버깅을 위한 출력
-    print("Full payload:", json.dumps(payload, indent=2, ensure_ascii=False))
-
     headers = {
         "accept": "application/json",
         "content-type": "application/json",
@@ -156,8 +155,8 @@ def search_serphouse(query, country, page=1, num_result=10):
 
     try:
         response = requests.post(url, json=payload, headers=headers)
+        print("Request payload:", json.dumps(payload, indent=2, ensure_ascii=False))
         print("Response status:", response.status_code)
-        print("Response content:", response.text[:500])  # 응답 내용 확인
         
         response.raise_for_status()
         return {"results": response.json(), "translated_query": translated_query}