Update app.py

app.py

@@ -4,7 +4,14 @@ import os
 
 class SimpleBanchanChatbot:
     def __init__(self):
-        self.genai_client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+        # 허깅페이스 스페이스에서는 secrets에서 API 키 가져오기
+        api_key = os.getenv("GEMINI_API_KEY")
+        if not api_key:
+            raise ValueError("GEMINI_API_KEY 환경변수가 설정되지 않았습니다!")
+            
+        # 제미나이 클라이언트 설정
+        genai.configure(api_key=api_key)
+        self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
         
         # 엑셀에서 추출한 모든 FAQ 데이터
         self.faq_data = """
@@ -105,16 +112,18 @@ class SimpleBanchanChatbot:
 - FAQ에 없는 내용은 "자세한 사항은 고객센터(010-8082-0047)로 문의해주세요"라고 안내해주세요
 - 반말보다는 존댓말을 사용해주세요
 - 간결하고 이해하기 쉽게 답변해주세요
+- 인사말로 시작하고 감사인사로 마무리해주세요
         """
         
         try:
-            response = self.genai_client.models.generate_content_stream(
-                model="gemini-2.0-flash-exp",
-                contents=message,
-                config=genai.types.GenerateContentConfig(
-                    system_instruction=system_prompt,
-                    temperature=0.1  # 일관된 답변을 위해 낮게 설정
-                )
+            response = self.model.generate_content(
+                message,
+                generation_config=genai.types.GenerationConfig(
+                    temperature=0.1,
+                    max_output_tokens=1000,
+                ),
+                system_instruction=system_prompt,
+                stream=True
             )
             
             partial_message = ""
@@ -124,35 +133,46 @@ class SimpleBanchanChatbot:
                     yield partial_message
                     
         except Exception as e:
-            yield f"죄송합니다. 시스템 오류가 발생했습니다. 고객센터(010-8082-0047)로 연락해주세요. 오류: {str(e)}"
+            yield f"죄송합니다. 시스템 오류가 발생했습니다. 고객센터(010-8082-0047)로 연락해주세요."
 
 # 챗봇 인스턴스 생성
-chatbot = SimpleBanchanChatbot()
-
-# 그라디오 인터페이스
-demo = gr.ChatInterface(
-    fn=chatbot.chat_stream,
-    title="🥢 맛있는반찬가게 AI 상담봇",
-    description="간단하고 빠른 반찬가게 상담봇 (제미나이 2.0 + FAQ 데이터)",
-    examples=[
-        "주문은 어떻게 하나요?",
-        "포인트는 어떻게 사용하나요?",
-        "배송비가 얼마인가요?",
-        "환불 방법을 알려주세요",
-        "최소 주문금액이 있나요?"
-    ],
-    theme=gr.themes.Soft(),
-    css="""
-    .gradio-container {
-        max-width: 800px !important;
-        margin: auto !important;
-    }
-    """
-)
+try:
+    chatbot = SimpleBanchanChatbot()
+    
+    # 그라디오 인터페이스
+    demo = gr.ChatInterface(
+        fn=chatbot.chat_stream,
+        title="🥢 맛있는반찬가게 AI 상담봇",
+        description="제미나이 2.0 기반 반찬가게 고객 상담봇입니다. 궁금한 것을 물어보세요!",
+        examples=[
+            "주문은 어떻게 하나요?",
+            "포인트는 어떻게 사용하나요?",
+            "배송비가 얼마인가요?",
+            "환불 방법을 알려주세요",
+            "정기식 구성이 어떻게 되나요?",
+            "후기 쓰면 혜택이 있나요?",
+            "반찬이 상한 것 같아요",
+            "원산지를 알고 싶어요"
+        ],
+        theme=gr.themes.Soft(),
+        css="""
+        .gradio-container {
+            max-width: 800px !important;
+            margin: auto !important;
+        }
+        """
+    )
+    
+except Exception as e:
+    # API 키가 없을 때 에러 메시지 표시
+    def error_message(message, history):
+        yield "❌ GEMINI_API_KEY가 설정되지 않았습니다. 관리자에게 문의하세요."
+    
+    demo = gr.ChatInterface(
+        fn=error_message,
+        title="🥢 설정 오류",
+        description="API 키 설정이 필요합니다."
+    )
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+    demo.launch()