Update app.py

app.py

@@ -13,8 +13,8 @@ import re
 import traceback
 import requests # API 호출을 위해 필요
 
-# --- Together AI SDK (제거됨) ---
-# from together import Together
+# --- Together AI SDK ---
+from together import Together
 
 # --- eventlet monkey patch (Gunicorn + SocketIO 필수!) ---
 import eventlet
@@ -67,6 +67,14 @@ if not TOGETHER_API_KEY:
     raise EnvironmentError("TOGETHER_API_KEY가 설정되지 않았습니다. Hugging Face Secrets에 추가하세요.")
 # client = Together(api_key=TOGETHER_API_KEY) # <--- Together SDK 클라이언트 제거
 
+try:
+    # TOGETHER_API_KEY를 사용해 클라이언트 초기화 (TOGETHER_API_KEY가 코드 내에 정의되어 있다고 가정)
+    client = Together(api_key=TOGETHER_API_KEY) 
+except NameError:
+    # TOGETHER_API_KEY가 정의되지 않은 경우 환경 변수 사용을 시도
+    client = Together()
+
+
 # --- RAG 로딩 ---
 def load_rag_objects():
     global region_rag_objects
@@ -187,26 +195,30 @@ def Gemma3_AI_analysis(query_txt, content_txt):
     query_txt = str(query_txt)
     prompt = lexi_prompts.use_prompt(lexi_prompts.AI_system_prompt, query_txt=query_txt, content_txt=content_txt)
 
-    headers = {
-        "Authorization": f"Bearer {TOGETHER_API_KEY}",
-        "Content-Type": "application/json"
-    }
-    payload = {
-        "model": "meta-llama/Llama-3.3-70B-Instruct-Turbo",
-        "messages": [{"role": "user", "content": prompt}],
-        "max_tokens": 1024,
-        "temperature": 0.7
-    }
-
     try:
-        response = requests.post("https://api.together.xyz/v1/chat/completions", headers=headers, json=payload, timeout=120)
-        response.raise_for_status() # HTTP 오류가 발생하면 예외 발생
-        
-        data = response.json()
-        return data["choices"][0]["message"]["content"]
-    except requests.exceptions.RequestException as e:
+        response = client.chat.completions.create(
+            model="meta-llama/Llama-3.3-70B-Instruct-Turbo",
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            # 기존 payload의 나머지 설정들을 추가합니다.
+            max_tokens=1024,
+            temperature=0.7,
+            # Together SDK는 기본적으로 요청 실패 시 예외를 발생시킵니다.
+        )
+    
+        # 응답에서 결과 텍스트를 추출
+        AI_Result = response.choices[0].message.content
+        return AI_Result
+    
+    except Exception as e:
+        # Together SDK의 오류는 requests.exceptions.RequestException이 아닌 다른 종류의 예외로 발생합니다.
+        # 따라서 일반적인 Exception으로 처리하는 것이 안전합니다.
         logger.info(f"Together AI 분석 API 호출 실패: {e}")
-        traceback.logger.info_exc()
+        traceback.print_exc() # traceback.logger.info_exc() 대신 일반 print_exc()를 사용하거나, logging 모듈 설정을 확인하세요.
         return f"AI 분석 중 오류가 발생했습니다: {e}"
 
 # --- Together AI 번역 (SDK -> requests 직접 호출로 변경) ---
@@ -214,27 +226,36 @@ def Gemma3_AI_Translate(query_txt):
     query_txt = str(query_txt)
     prompt = lexi_prompts.use_prompt(lexi_prompts.query_translator, query_txt=query_txt)
 
-    headers = {
-        "Authorization": f"Bearer {TOGETHER_API_KEY}",
-        "Content-Type": "application/json"
-    }
-    payload = {
-        "model": "meta-llama/Llama-3.2-3B-Instruct-Turbo",
-        "messages": [{"role": "user", "content": prompt}],
-        "max_tokens": 512,
-        "temperature": 0.3
-    }
-    
     try:
-        response = requests.post("https://api.together.xyz/v1/chat/completions", headers=headers, json=payload, timeout=60)
-        response.raise_for_status() # HTTP 오류가 발생하면 예외 발생
-
-        data = response.json()
-        return data["choices"][0]["message"]["content"]
-    except requests.exceptions.RequestException as e:
+        response = client.chat.completions.create(
+            model="meta-llama/Llama-3.3-70B-Instruct-Turbo",
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            # 기존 payload의 나머지 설정들을 추가합니다.
+            max_tokens=1024,
+            temperature=0.7,
+            # Together SDK는 기본적으로 요청 실패 시 예외를 발생시킵니다.
+        )
+    
+        # 응답에서 결과 텍스트를 추출
+        AI_Result = response.choices[0].message.content
+        return AI_Result
+    
+    except Exception as e:
+        # API 호출 실패 시 처리 (SDK 사용 시 일반 Exception으로 처리)
         logger.info(f"Together AI 번역 API 호출 실패: {e}")
-        traceback.logger.info_exc()
-        return query_txt # 번역 실패 시 원래 쿼리를 사용 (최소한의 기능 유지)
+        
+        # traceback.logger.info_exc() 대신 traceback.print_exc() 사용 (권장)
+        # 만약 기존 로깅 시스템에서 해당 함수를 정의해 사용하고 있다면 그대로 두셔도 됩니다.
+        # 여기서는 표준 traceback 모듈을 사용합니다.
+        traceback.print_exc() 
+        
+        # 번역 실패 시 query_txt 변수를 반환 (기존 코드 로직 반영)
+        return query_txt
 
 # --- 검색 ---
 def search_DB_from_multiple_regions(query, selected_regions, region_rag_objects):