Update app.py

app.py

@@ -563,7 +563,7 @@ class NovelWritingSystem:
         return full_content
 
     def call_llm_streaming(self, messages: List[Dict[str, str]], role: str, language: str) -> Generator[str, None, None]:
-        """LLM 스트리밍 호출 (반복 방지 파라미터 추가)"""
+        """LLM 스트리밍 호출 (안전한 에러 처리)"""
         try:
             system_prompts = self.get_system_prompts(language)
             full_messages = [{"role": "system", "content": system_prompts.get(role, "You are a helpful assistant.")}, *messages]
@@ -578,32 +578,82 @@ class NovelWritingSystem:
                 "stream": True,
                 "stream_options": {"include_usage": True}
             }
+            
+            logger.info(f"API 호출 시작 - 역할: {role}")
             response = requests.post(self.api_url, headers=self.create_headers(), json=payload, stream=True, timeout=180)
             response.raise_for_status()
+            
             buffer = ""
+            chunk_count = 0
+            
             for line in response.iter_lines():
                 if line:
                     line_str = line.decode('utf-8')
                     if line_str.startswith("data: "):
                         data = line_str[6:]
-                        if data == "[DONE]": break
+                        if data == "[DONE]": 
+                            logger.info(f"스트리밍 완료 - 총 {chunk_count}개 청크 처리")
+                            break
+                        
                         try:
                             chunk = json.loads(data)
-                            content = chunk.get("choices", [{}])[0].get("delta", {}).get("content", "")
-                            if content:
-                                buffer += content
-                                if "\n" in buffer or len(buffer) > 50:
-                                    yield buffer
-                                    buffer = ""
-                        except json.JSONDecodeError:
+                            
+                            # 디버깅을 위해 첫 몇 개 청크 로깅
+                            if chunk_count < 3:
+                                logger.debug(f"청크 {chunk_count}: {json.dumps(chunk, ensure_ascii=False)[:200]}")
+                            
+                            # 안전한 content 추출
+                            choices = chunk.get("choices", [])
+                            if choices and len(choices) > 0:
+                                delta = choices[0].get("delta", {})
+                                content = delta.get("content", "")
+                                
+                                if content:
+                                    buffer += content
+                                    chunk_count += 1
+                                    
+                                    # 버퍼가 충분히 차거나 줄바꿈이 있을 때 yield
+                                    if len(buffer) > 100 or "\n" in buffer:
+                                        yield buffer
+                                        buffer = ""
+                            
+                            # 에러 메시지 체크
+                            elif chunk.get("error"):
+                                error_msg = chunk.get("error", {}).get("message", "Unknown error")
+                                logger.error(f"API 에러 응답: {error_msg}")
+                                yield f"❌ API 오류: {error_msg}"
+                                return
+                                
+                        except json.JSONDecodeError as e:
+                            logger.warning(f"JSON 파싱 오류: {e}, 데이터: {data[:100]}")
                             continue
-            if buffer: yield buffer
+                        except Exception as e:
+                            logger.error(f"청크 처리 중 예외: {e}")
+                            continue
+            
+            # 남은 버퍼 내용 출력
+            if buffer: 
+                yield buffer
+                
+            # 청크가 하나도 처리되지 않았다면
+            if chunk_count == 0:
+                logger.error("응답에서 콘텐츠를 찾을 수 없습니다.")
+                yield "❌ API 응답에 콘텐츠가 없습니다."
+                
+        except requests.exceptions.HTTPError as e:
+            logger.error(f"HTTP 오류: {e}")
+            logger.error(f"응답 상태 코드: {e.response.status_code if e.response else 'N/A'}")
+            logger.error(f"응답 내용: {e.response.text[:500] if e.response else 'N/A'}")
+            yield f"❌ HTTP 오류: {e}"
         except requests.exceptions.RequestException as e:
-            logger.error(f"API 스트리밍 중 오류: {e}")
+            logger.error(f"API 요청 오류: {e}")
             yield f"❌ API 오류: {e}"
         except Exception as e:
-            logger.error(f"스트리밍 처리 중 예외: {e}", exc_info=True)
+            logger.error(f"예기치 않은 오류: {e}", exc_info=True)
             yield f"❌ 처리 오류: {e}"
+    
+    
+
 
     def get_system_prompts(self, language: str) -> Dict[str, str]:
         """역할별 시스템 프롬프트 생성"""