Update app.py

app.py

@@ -1,42 +1,4 @@
-try:
-            # 2.0 버전용 안정적인 API 호출
-            response = self.model.generate_content(
-                contents=f"{system_prompt}\n\n사용자 질문: {message}",
-                generation_config=genai.GenerationConfig(
-                    temperature=0.1,
-                    max_output_tokens=1000,
-                ),
-                stream=True
-            )
-            
-            partial_message = ""
-            for chunk in response:
-                try:
-                    if chunk.text:
-                        partial_message += chunk.text
-                        yield partial_message
-                except Exception as chunk_error:
-                    # 개별 청크 에러는 무시하고 계속 진행
-                    continue
-                    
-        except Exception as e:
-            # 에러 내용을 더 자세히 보기 위해
-            print(f"에러 발생: {str(e)}")
-            
-            # 스트리밍이 안되면 일반 방식으로 시도
-            try:
-                response = self.model.generate_content(
-                    contents=f"{system_prompt}\n\n사용자 질문: {message}",
-                    generation_config=genai.GenerationConfig(
-                        temperature=0.1,
-                        max_output_tokens=1000,
-                    ),
-                    stream=False
-                )
-                yield response.text
-            except Exception as fallback_error:
-                print(f"Fallback 에러: {str(fallback_error)}")
-                yield f"죄송합니다. 시스템 오류가 발생했습니다. 고객센터(010-8082-0047)로 연락해주세요."import gradio as gr
+import gradio as gr
 import google.generativeai as genai
 import os
 
@@ -49,7 +11,6 @@ class SimpleBanchanChatbot:
             
         # 제미나이 클라이언트 설정
         genai.configure(api_key=api_key)
-        # 2.0 버전 사용
         self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
         
         # 엑셀에서 추출한 모든 FAQ 데이터
@@ -156,23 +117,38 @@ class SimpleBanchanChatbot:
         
         try:
             response = self.model.generate_content(
-                message,
-                generation_config=genai.types.GenerationConfig(
+                contents=f"{system_prompt}\n\n사용자 질문: {message}",
+                generation_config=genai.GenerationConfig(
                     temperature=0.1,
                     max_output_tokens=1000,
                 ),
-                system_instruction=system_prompt,
                 stream=True
             )
             
             partial_message = ""
             for chunk in response:
-                if chunk.text:
-                    partial_message += chunk.text
-                    yield partial_message
+                try:
+                    if chunk.text:
+                        partial_message += chunk.text
+                        yield partial_message
+                except:
+                    continue
                     
         except Exception as e:
-            yield f"죄송합니다. 시스템 오류가 발생했습니다. 고객센터(010-8082-0047)로 연락해주세요."
+            print(f"에러 발생: {str(e)}")
+            try:
+                response = self.model.generate_content(
+                    contents=f"{system_prompt}\n\n사용자 질문: {message}",
+                    generation_config=genai.GenerationConfig(
+                        temperature=0.1,
+                        max_output_tokens=1000,
+                    ),
+                    stream=False
+                )
+                yield response.text
+            except Exception as fallback_error:
+                print(f"Fallback 에러: {str(fallback_error)}")
+                yield f"죄송합니다. 시스템 오류가 발생했습니다. 고객센터(010-8082-0047)로 연락해주세요."
 
 # 챗봇 인스턴스 생성
 try:
@@ -203,7 +179,6 @@ try:
     )
     
 except Exception as e:
-    # API 키가 없을 때 에러 메시지 표시
     def error_message(message, history):
         yield "❌ GEMINI_API_KEY가 설정되지 않았습니다. 관리자에게 문의하세요."