Update app.py

app.py

@@ -18,7 +18,7 @@ import pandas as pd
 import PyPDF2                       # For handling PDF files
 from collections import Counter
 
-from openai import OpenAI, APIError, APITimeoutError   # ⬅️ OpenAI 오류형 포함
+from openai import OpenAI, APIError, APITimeoutError
 from gradio_client import Client
 from kaggle.api.kaggle_api_extended import KaggleApi
 import tempfile
@@ -30,10 +30,32 @@ import pyarrow.parquet as pq
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 
-# ─── NEW : 네트워크 안정화용 라이브러리 ──────────────────────────────
+# ─── 네트워크 안정화용 라이브러리 ──────────────────────────────────────
 import httpx
 from httpx import RemoteProtocolError
-import backoff                       # ⬅️ 자동 재시도 데코레이터
+
+# ▸ backoff 모듈이 없으면 즉석에서 대체 구현
+try:
+    import backoff
+except ImportError:                              # ← 최초 실행 환경에서 발생
+    logging.warning("`backoff` 모듈이 없어 간단 대체 데코레이터를 사용합니다.")
+    def _simple_backoff_on_exception(exc_tuple, max_tries=3, base=2):
+        def decorator(fn):
+            def wrapper(*args, **kwargs):
+                for attempt in range(1, max_tries + 1):
+                    try:
+                        return fn(*args, **kwargs)
+                    except exc_tuple as e:
+                        if attempt == max_tries:
+                            raise
+                        sleep = base ** attempt
+                        logging.info(f"Retry {attempt}/{max_tries} after {sleep}s ({e})")
+                        time.sleep(sleep)
+            return wrapper
+        return decorator
+    class _DummyBackoff:
+        on_exception = _simple_backoff_on_exception
+    backoff = _DummyBackoff()                    # 동일 API 제공
 
 # ─────────────────────────────── Environment Variables / Constants ─────────────────────────
 
@@ -49,12 +71,9 @@ if not (KAGGLE_USERNAME and KAGGLE_KEY):
 os.environ["KAGGLE_USERNAME"] = KAGGLE_USERNAME
 os.environ["KAGGLE_KEY"]      = KAGGLE_KEY
 
-# ▶ 일부 프록시에서 HTTP/2 → 1.1 강제 전환이 필요할 때 활성화
-# os.environ["HTTPX_FORCE_HTTP1"] = "true"
-
 BRAVE_ENDPOINT   = "https://api.search.brave.com/res/v1/web/search"
 IMAGE_API_URL    = "http://211.233.58.201:7896"  # 예시 이미지 생성용 API
-MAX_TOKENS       = 7999       
+MAX_TOKENS       = 4096        # 안전한 토큰 한도
 
 # ─────────────────────────────── Logging ───────────────────────────────
 logging.basicConfig(
@@ -2038,21 +2057,22 @@ def process_example(topic):
 def process_input(prompt: str, uploaded_files):
     """
     메인 채팅 입력을 받아 디자인/발명 아이디어를 생성한다.
-    스트리밍 중단(HTTPX RemoteProtocolError 등) 시 자동 재시도,
-    최종 실패 시 non-stream 호출로 폴백하도록 개선.
+    스트리밍 실패(RemoteProtocolError 등) 시 backoff 재시도 후
+    최종적으로 non-stream 호출로 폴백.
     """
+    # ─── 대화 기록 중복 방지 ──────────────────────────────
     if not any(m["role"] == "user" and m["content"] == prompt for m in st.session_state.messages):
         st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    # 동일 프롬프트-응답 중복 방지
     for i in range(len(st.session_state.messages) - 1):
         if (st.session_state.messages[i]["role"] == "user"
             and st.session_state.messages[i]["content"] == prompt
             and st.session_state.messages[i + 1]["role"] == "assistant"):
             return
 
+    # ─── 결과 생성 ───────────────────────────────────────
     with st.chat_message("assistant"):
         status = st.status("Preparing to generate invention ideas…")
         stream_placeholder = st.empty()
@@ -2064,8 +2084,6 @@ def process_input(prompt: str, uploaded_files):
 
             selected_cat        = st.session_state.get("category_focus", None)
             selected_frameworks = st.session_state.get("selected_frameworks", [])
-
-            # 목적이 "디자인/발명"이므로, system prompt 변경
             sys_prompt = get_idea_system_prompt(
                 selected_category=selected_cat,
                 selected_frameworks=selected_frameworks
@@ -2080,9 +2098,9 @@ def process_input(prompt: str, uploaded_files):
             use_kaggle     = st.session_state.kaggle_enabled
             has_uploaded   = bool(uploaded_files)
 
-            search_content = kaggle_content = file_content = mil_content = None  # 초기화
+            search_content = kaggle_content = file_content = mil_content = None
 
-            # ① 웹검색
+            # ① 웹 검색
             if use_web_search:
                 status.update(label="Searching the web…")
                 with st.spinner("Searching…"):
@@ -2117,7 +2135,7 @@ def process_input(prompt: str, uploaded_files):
                 with st.spinner("Processing files…"):
                     file_content = process_uploaded_files(uploaded_files)
 
-            # ④ 군사 전술 데이터 (필요 시)
+            # ④ 군사 전술 데이터
             if is_military_query(prompt):
                 status.update(label="Searching military tactics dataset…")
                 with st.spinner("Loading military insights…"):
@@ -2132,13 +2150,13 @@ def process_input(prompt: str, uploaded_files):
                             f"**Defense Reasoning:** {row['defense_reasoning']}\n\n---\n"
                         )
 
-            # 사용자 프롬프트 + 참조자료 결합
+            # ─── 유저 콘텐츠 구성 ──────────────────────────
             user_content = prompt
-            for _extra in (search_content, kaggle_content, file_content, mil_content):
-                if _extra:
-                    user_content += "\n\n" + _extra
+            for extra in (search_content, kaggle_content, file_content, mil_content):
+                if extra:
+                    user_content += "\n\n" + extra
 
-            # 내부 분석
+            # ─── 내부 분석 ───────────────────────────────
             status.update(label="분석 중…")
             decision_purpose = identify_decision_purpose(prompt)
             relevance_scores = compute_relevance_scores(prompt, PHYS_CATEGORIES)
@@ -2173,7 +2191,7 @@ def process_input(prompt: str, uploaded_files):
                 for c, s in decision_purpose['constraints']:
                     purpose_info += f"- **{c}** (관련성: {s})\n"
 
-            # 기타 프레임워크
+            # ─── 프레임워크 분석 (옵션) ────────────────────
             framework_contents = []
             for fw in selected_frameworks:
                 if fw == "swot":
@@ -2188,7 +2206,6 @@ def process_input(prompt: str, uploaded_files):
                     framework_contents.append(
                         format_business_framework_analysis("bcg", analyze_with_bcg(prompt))
                     )
-                # sunzi 등 필요 시 추가
 
             if framework_contents:
                 user_content += "\n\n## (Optional) 기타 프레임워크 분석\n\n" + "\n\n".join(framework_contents)
@@ -2203,15 +2220,11 @@ def process_input(prompt: str, uploaded_files):
                 {"role": "user",   "content": user_content},
             ]
 
-            # ─────────────────────────────────────────────────────────────
-            # ⬇️ 1. 안전한 스트리밍 호출 (backoff 재시도)
+            # ─── OpenAI Chat 호출 (backoff 재시도) ─────────────────
             @backoff.on_exception(
-                backoff.expo,
-                (RemoteProtocolError, APITimeoutError, APIError),
-                max_tries=3,
-                jitter=None
+                (RemoteProtocolError, APITimeoutError, APIError), max_tries=3
             )
-            def safe_stream_chat():
+            def safe_stream():
                 return client.chat.completions.create(
                     model="gpt-4.1-mini",
                     messages=api_messages,
@@ -2222,13 +2235,13 @@ def process_input(prompt: str, uploaded_files):
                 )
 
             try:
-                stream = safe_stream_chat()
+                stream = safe_stream()
                 for chunk in stream:
                     if chunk.choices and chunk.choices[0].delta.content:
                         full_response += chunk.choices[0].delta.content
                         stream_placeholder.markdown(full_response + "▌")
             except (RemoteProtocolError, APITimeoutError, APIError) as stream_err:
-                logging.warning(f"💡 스트리밍 중단, non-stream으로 폴백: {stream_err}")
+                logging.warning(f"스트리밍 실패, non-stream 폴백: {stream_err}")
                 resp = client.chat.completions.create(
                     model="gpt-4.1-mini",
                     messages=api_messages,
@@ -2239,11 +2252,10 @@ def process_input(prompt: str, uploaded_files):
                 )
                 full_response = resp.choices[0].message.content
                 stream_placeholder.markdown(full_response)
-            # ─────────────────────────────────────────────────────────────
 
             status.update(label="Invention ideas created!", state="complete")
 
-            # 자동 이미지 생성
+            # ─── 이미지 생성 ────────────────────────────────
             img_data = img_caption = None
             if st.session_state.generate_image and full_response:
                 match = re.search(r"###\s*이미지\s*프롬프트\s*\n+([^\n]+)", full_response, re.I)
@@ -2256,7 +2268,7 @@ def process_input(prompt: str, uploaded_files):
                     if img_data:
                         st.image(img_data, caption=f"Visualized Concept – {img_caption}")
 
-            # 세션 메시지 업데이트
+            # ─── 세션 메시지 저장 ─────────────────────────────
             answer_msg = {"role": "assistant", "content": full_response}
             if img_data:
                 answer_msg["image"]         = img_data
@@ -2264,7 +2276,7 @@ def process_input(prompt: str, uploaded_files):
             st.session_state["_skip_dup_idx"] = len(st.session_state.messages)
             st.session_state.messages.append(answer_msg)
 
-            # 다운로드 버튼
+            # ─── 다운로드 옵션 ──────────────────────────────
             st.subheader("Download This Output")
             col_md, col_html = st.columns(2)
             col_md.download_button(
@@ -2292,6 +2304,7 @@ def process_input(prompt: str, uploaded_files):
                 {"role": "assistant", "content": f"⚠️ 오류: {e}"}
             )
 
+
 def main():
     idea_generator_app()