Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import requests
 import logging
 import json
+import re
 from datetime import datetime, timedelta
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
@@ -50,27 +51,15 @@ with st.sidebar:
     system_message = st.text_area(
         "System Message",
         value=(
-            "You are a deep thinking AI, you may use extremely long chains of thought to deeply consider the problem and "
-            "deliberate with yourself via systematic reasoning processes to help come to a correct solution prior to answering. "
+            "You are a deep thinking AI, you may use extremely long chains of thought to deeply consider the problem and deliberate with yourself via systematic reasoning processes to help come to a correct solution prior to answering. "
             "You should enclose your thoughts and internal monologue inside tags, and then provide your solution or response to the problem."
         ),
         height=100
     )
 
-    max_tokens = st.slider(
-        "Max Tokens",
-        10, 4000, 1000
-    )
-
-    temperature = st.slider(
-        "Temperature",
-        0.1, 4.0, 0.3
-    )
-
-    top_p = st.slider(
-        "Top-p",
-        0.1, 1.0, 0.6
-    )
+    max_tokens = st.slider("Max Tokens", 10, 4000, 1000)
+    temperature = st.slider("Temperature", 0.1, 4.0, 0.3)
+    top_p = st.slider("Top-p", 0.1, 1.0, 0.6)
 
 # Function to query the Hugging Face API
 def query(payload, api_url):
@@ -103,7 +92,7 @@ def search_web(query, country="United States", page=1, num_result=10):
             "loc": COUNTRY_LOCATIONS.get(country, "United States"),
             "lang": COUNTRY_LANGUAGES.get(country, "en"),
             "device": "desktop",
-            "serp_type": "web",  # 검색 유형을 "web"으로 설정 (원하는 경우 "news" 등으로 변경 가능)
+            "serp_type": "web",  # 원하는 경우 "news" 등으로 변경 가능
             "page": str(page),
             "num": str(num_result),
             "date_range": date_range,
@@ -111,7 +100,6 @@ def search_web(query, country="United States", page=1, num_result=10):
         }
     }
     
-    # st.secrets에 SERPHOUSE_API_TOKEN이 저장되어 있어야 합니다.
     api_key = st.secrets.get("SERPHOUSE_API_TOKEN")
     if not api_key:
         logger.error("SERPHOUSE_API_TOKEN not found in st.secrets")
@@ -124,7 +112,6 @@ def search_web(query, country="United States", page=1, num_result=10):
     }
     
     try:
-        # 세션과 재시도 설정
         session = requests.Session()
         retries = Retry(
             total=5,
@@ -136,12 +123,11 @@ def search_web(query, country="United States", page=1, num_result=10):
         session.mount('http://', adapter)
         session.mount('https://', adapter)
         
-        # 연결 및 읽기 타임아웃 30초씩 설정
         response = session.post(
             url,
             json=payload,
             headers=headers,
-            timeout=(30, 30)
+            timeout=(30, 30)  # 연결 및 읽기 타임아웃 30초씩
         )
         response.raise_for_status()
         return {"results": response.json(), "translated_query": translated_query}
@@ -163,6 +149,11 @@ def search_web(query, country="United States", page=1, num_result=10):
         }
 # --- 끝 ---
 
+# 내부 체인 오브 쏘(생각 과정) 제거 함수
+def remove_chain_of_thought(text):
+    cleaned_text = re.sub(r'<think>.*?</think>', '', text, flags=re.DOTALL)
+    return cleaned_text.strip()
+
 # Chat interface
 st.title("🤖 DeepSeek Chatbot")
 st.caption("Powered by Hugging Face Inference API - Configure in sidebar")
@@ -184,11 +175,12 @@ if prompt := st.chat_input("Type your message..."):
             # 업데이트된 웹서치 기능을 사용하여 검색 수행
             search_results = search_web(prompt, country="United States", page=1, num_result=10)
             
-            # 검색 결과 처리 (API 응답 구조에 따라 적절히 수정 필요)
+            # 검색 결과 처리
             if search_results and "results" in search_results:
                 if 'organic' in search_results["results"]:
-                    search_content = "\n".join([f"**{item['title']}**: {item['snippet']}" 
-                                                for item in search_results["results"]["organic"]])
+                    search_content = "\n".join(
+                        [f"**{item['title']}**: {item['snippet']}" for item in search_results["results"]["organic"]]
+                    )
                     search_content = f"Here are some search results related to your question:\n\n{search_content}\n\n"
                 else:
                     search_content = "Sorry, no relevant search results found.\n\n"
@@ -214,14 +206,12 @@ if prompt := st.chat_input("Type your message..."):
             # Hugging Face API에 쿼리 전송
             output = query(payload, api_url)
     
-            # 응답 처리
+            # API 응답 처리
             if output is not None and isinstance(output, list) and len(output) > 0:
                 if 'generated_text' in output[0]:
                     assistant_response = output[0]['generated_text'].strip()
-    
-                    # 중복 응답 제거 (내부 체인 중 일부 제거)
-                    responses = assistant_response.split("\n</think>\n")
-                    unique_response = responses[0].strip()
+                    # 내부 체인 오브 쏘 제거
+                    unique_response = remove_chain_of_thought(assistant_response)
     
                     logger.info(f"Generated response: {unique_response}")