Update app.py

app.py

@@ -11,21 +11,80 @@ load_dotenv()
 def create_deepseek_interface():
     # 환경 변수에서 API 키 가져오기
     api_key = os.getenv("FW_API_KEY")
+    serphouse_api_key = os.getenv("SERPHOUSE_API_KEY")
+    
     if not api_key:
         print("경고: FW_API_KEY 환경 변수가 설정되지 않았습니다.")
+    if not serphouse_api_key:
+        print("경고: SERPHOUSE_API_KEY 환경 변수가 설정되지 않았습니다.")
+    
+    # SerpHouse API를 사용하여 검색 수행 함수
+    def search_with_serphouse(query):
+        if not serphouse_api_key:
+            return "SERPHOUSE_API_KEY가 설정되지 않았습니다."
+        
+        url = "https://api.serphouse.com/serp/live"
+        payload = {
+            "q": query,
+            "domain": "google.com",
+            "loc": "us",
+            "lang": "en",
+            "device": "desktop",
+            "serp_type": "web",
+            "page": 1,
+            "num": 5  # 상위 5개 결과만 가져오기
+        }
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {serphouse_api_key}"
+        }
+        
+        try:
+            response = requests.post(url, headers=headers, json=payload)
+            response.raise_for_status()
+            
+            search_results = response.json()
+            
+            # 검색 결과 파싱 및 포맷팅
+            formatted_results = []
+            if "organic" in search_results and len(search_results["organic"]) > 0:
+                for result in search_results["organic"][:5]:  # 상위 5개 결과만 사용
+                    title = result.get("title", "제목 없음")
+                    snippet = result.get("snippet", "내용 없음")
+                    link = result.get("link", "#")
+                    formatted_results.append(f"제목: {title}\n내용: {snippet}\n링크: {link}\n\n")
+                
+                return "".join(formatted_results)
+            else:
+                return "검색 결과가 없습니다."
+                
+        except Exception as e:
+            return f"검색 중 오류 발생: {str(e)}"
     
     # 스트리밍 방식으로 DeepSeek API 호출 함수
-    def query_deepseek_streaming(message, history):
+    def query_deepseek_streaming(message, history, use_deep_research):
         if not api_key:
             yield history, "환경 변수 FW_API_KEY가 설정되지 않았습니다. 서버에서 환경 변수를 확인해주세요."
             return
         
+        search_context = ""
+        if use_deep_research:
+            # 검색 수행
+            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"
+        
         # API 요청을 위한 대화 기록 준비
         messages = []
         for user, assistant in history:
             messages.append({"role": "user", "content": user})
             messages.append({"role": "assistant", "content": assistant})
         
+        # 검색 컨텍스트가 있으면 시스템 메시지 추가
+        if search_context:
+            # DeepSeek 모델은 시스템 메시지를 지원합니다
+            messages.insert(0, {"role": "system", "content": search_context})
+        
         # 새 사용자 메시지 추가
         messages.append({"role": "user", "content": message})
         
@@ -105,35 +164,24 @@ def create_deepseek_interface():
             """
         )
         
-        # 메인 레이아웃 (두 개의 열)
+        # 메인 레이아웃
         with gr.Row():
-            # 사이드바 - 모델 정보
-            with gr.Column(scale=1):
-                gr.Markdown(
-                    """
-                    ## 📊 모델 세부 정보
-                    - **모델**: DeepSeek-V3-0324
-                    - **제공자**: Fireworks AI
-                    - **최대 토큰**: 20,480
-                    - **온도**: 0.6
-                    - **기능**: 고급 언어 이해와 실시간 스트리밍 응답
-                    
-                    > **참고**: API 키는 환경 변수(FW_API_KEY)에서 자동으로 로드됩니다.
-                    """
-                )
-                
-                # 오류 메시지 표시
-                error_box = gr.Markdown("")
-            
             # 메인 콘텐츠 영역
-            with gr.Column(scale=2):
-                # 채팅 인터페이스 - 호환성 문제 해결을 위해 매개변수 제거
+            with gr.Column():
+                # 채팅 인터페이스
                 chatbot = gr.Chatbot(
                     height=500,
                     show_label=False,
                     container=True
                 )
                 
+                # Deep Research 토글 추가
+                with gr.Row():
+                    use_deep_research = gr.Checkbox(
+                        label="Deep Research 활성화",
+                        info="최신 웹 검색 결과를 활용한 응답 생성"
+                    )
+                
                 # 입력 영역
                 with gr.Row():
                     msg = gr.Textbox(
@@ -157,11 +205,14 @@ def create_deepseek_interface():
                     ],
                     inputs=msg
                 )
+                
+                # 오류 메시지 표시
+                error_box = gr.Markdown("")
         
         # 버튼과 기능 연결
         submit.click(
             query_deepseek_streaming,
-            inputs=[msg, chatbot],
+            inputs=[msg, chatbot, use_deep_research],
             outputs=[chatbot, error_box]
         ).then(
             lambda: "",
@@ -172,7 +223,7 @@ def create_deepseek_interface():
         # Enter 키 제출 허용
         msg.submit(
             query_deepseek_streaming,
-            inputs=[msg, chatbot],
+            inputs=[msg, chatbot, use_deep_research],
             outputs=[chatbot, error_box]
         ).then(
             lambda: "",