Update app.py

app.py

@@ -18,14 +18,65 @@ def create_deepseek_interface():
     if not serphouse_api_key:
         print("경고: SERPHOUSE_API_KEY 환경 변수가 설정되지 않았습니다.")
     
+    # 키워드 추출 함수 (LLM 기반)
+    def extract_keywords_with_llm(query):
+        if not api_key:
+            return "LLM 키워드 추출을 위한 FW_API_KEY가 설정되지 않았습니다.", query
+        
+        # LLM을 사용하여 키워드 추출 (DeepSeek 모델 사용)
+        url = "https://api.fireworks.ai/inference/v1/chat/completions"
+        payload = {
+            "model": "accounts/fireworks/models/deepseek-v3-0324",
+            "max_tokens": 200,
+            "temperature": 0.1,  # 일관된 결과를 위해 낮은 온도 사용
+            "messages": [
+                {
+                    "role": "system",
+                    "content": "사용자의 질문에서 웹 검색에 효과적인 핵심 키워드 3-5개를 추출하세요. 키워드만 쉼표로 구분하여 출력하고 다른 설명이나 부가 정보는 제공하지 마세요."
+                },
+                {
+                    "role": "user",
+                    "content": query
+                }
+            ]
+        }
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {api_key}"
+        }
+        
+        try:
+            response = requests.post(url, headers=headers, json=payload)
+            response.raise_for_status()
+            result = response.json()
+            
+            # 응답에서 키워드 추출
+            keywords = result["choices"][0]["message"]["content"].strip()
+            
+            # 키워드가 너무 길거나 형식이 잘못된 경우 원본 쿼리 사용
+            if len(keywords) > 100 or "," not in keywords:
+                return f"추출된 키워드: {keywords}", query
+                
+            return f"추출된 키워드: {keywords}", keywords
+            
+        except Exception as e:
+            print(f"키워드 추출 중 오류 발생: {str(e)}")
+            return f"키워드 추출 중 오류 발생: {str(e)}", query
+    
     # SerpHouse API를 사용하여 검색 수행 함수
     def search_with_serphouse(query):
         if not serphouse_api_key:
             return "SERPHOUSE_API_KEY가 설정되지 않았습니다."
         
+        # 키워드 추출
+        extraction_result, search_query = extract_keywords_with_llm(query)
+        print(f"원본 쿼리: {query}")
+        print(extraction_result)
+        
         url = "https://api.serphouse.com/serp/live"
         payload = {
-            "q": query,
+            "q": search_query,
             "domain": "google.com",
             "loc": "us",
             "lang": "en",
@@ -47,6 +98,8 @@ def create_deepseek_interface():
             
             # 검색 결과 파싱 및 포맷팅
             formatted_results = []
+            formatted_results.append(f"검색어: {search_query}\n\n")
+            
             if "organic" in search_results and len(search_results["organic"]) > 0:
                 for result in search_results["organic"][:5]:  # 상위 5개 결과만 사용
                     title = result.get("title", "제목 없음")
@@ -56,7 +109,7 @@ def create_deepseek_interface():
                 
                 return "".join(formatted_results)
             else:
-                return "검색 결과가 없습니다."
+                return f"검색어 '{search_query}'에 대한 검색 결과가 없습니다."
                 
         except Exception as e:
             return f"검색 중 오류 발생: {str(e)}"
@@ -68,11 +121,23 @@ def create_deepseek_interface():
             return
         
         search_context = ""
+        search_info = ""
         if use_deep_research:
-            # 검색 수행
+            # 검색 수행 (첫 메시지 전달)
+            yield history + [(message, "🔍 최적의 키워드 추출 및 웹 검색 중...")], ""
+            
+            # 검색 실행
             search_results = search_with_serphouse(message)
             if not search_results.startswith("검색 중 오류 발생") and not search_results.startswith("SERPHOUSE_API_KEY"):
-                search_context = f"다음은 사용자 질문과 관련된 최신 검색 결과입니다. 이 정보를 참고하여 정확하고 최신 정보가 반영된 응답을 제공하세요:\n\n{search_results}\n\n사용자 질문에 답변하세요.\n\n"
+                search_context = f"""
+다음은 사용자 질문과 관련된 최신 검색 결과입니다. 이 정보를 참고하여 정확하고 최신 정보가 반영된 응답을 제공하세요:
+
+{search_results}
+
+위 검색 결과를 기반으로 사용자의 다음 질문에 답변하세요. 검색 결과에서 명확한 답변을 찾을 수 없는 경우, 당신의 지식을 활용하여 최선의 답변을 제공하세요.
+검색 결과를 인용할 때는 출처를 명시하고, 답변이 최신 ���보를 반영하도록 하세요.
+"""
+                search_info = f"🔍 Deep Research 기능 활성화: 관련 웹 검색 결과를 기반으로 응답 생성 중..."
         
         # API 요청을 위한 대화 기록 준비
         messages = []
@@ -112,12 +177,15 @@ def create_deepseek_interface():
             response = requests.request("POST", url, headers=headers, data=json.dumps(payload), stream=True)
             response.raise_for_status()  # HTTP 오류 발생 시 예외 발생
             
-            # 메시지를 추가하고 초기 빈 응답으로 시작
+            # 메시지를 추가하고 초기 응답으로 시작
             new_history = history.copy()
-            new_history.append((message, ""))
+            
+            # search_info가 있으면 시작 메시지에 포함
+            start_msg = search_info if search_info else ""
+            new_history.append((message, start_msg))
             
             # 응답 전체 텍스트
-            full_response = ""
+            full_response = start_msg
             
             # 스트리밍 응답 처리
             for line in response.iter_lines():
@@ -179,7 +247,7 @@ def create_deepseek_interface():
                 with gr.Row():
                     use_deep_research = gr.Checkbox(
                         label="Deep Research 활성화",
-                        info="최신 웹 검색 결과를 활용한 응답 생성"
+                        info="최적의 키워드 추출 및 웹 검색을 통한 최신 정보 활용"
                     )
                 
                 # 입력 영역