ssboost commited on
Commit
4767cdb
ยท
verified ยท
1 Parent(s): ef35a5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  import google.generativeai as genai
3
  import os
4
 
5
- class SimpleBanchanChatbot:
6
  def __init__(self):
7
  # ํ—ˆ๊น…ํŽ˜์ด์Šค ์ŠคํŽ˜์ด์Šค์—์„œ๋Š” secrets์—์„œ API ํ‚ค ๊ฐ€์ ธ์˜ค๊ธฐ
8
  api_key = os.getenv("GEMINI_API_KEY")
@@ -13,9 +13,9 @@ class SimpleBanchanChatbot:
13
  genai.configure(api_key=api_key)
14
  self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
15
 
16
- # ์—‘์…€์—์„œ ์ถ”์ถœํ•œ ๋ชจ๋“  FAQ ๋ฐ์ดํ„ฐ
17
  self.faq_data = """
18
- ๋ฐ˜์ฐฌ๊ฐ€๊ฒŒ ๊ณ ๊ฐ์„ผํ„ฐ FAQ:
19
 
20
  == ๊ธฐ๋ณธ ์ •๋ณด ==
21
  โ€ข ๊ณ ๊ฐ์„ผํ„ฐ: 010-8082-0047
@@ -103,7 +103,7 @@ class SimpleBanchanChatbot:
103
 
104
  def chat_stream(self, message, history):
105
  system_prompt = f"""
106
- ๋‹น์‹ ์€ ๋ง›์žˆ๋Š”๋ฐ˜์ฐฌ๊ฐ€๊ฒŒ์˜ ์นœ์ ˆํ•œ ์ƒ๋‹ด์›์ž…๋‹ˆ๋‹ค.
107
 
108
  {self.faq_data}
109
 
@@ -112,7 +112,7 @@ class SimpleBanchanChatbot:
112
  - FAQ์— ์—†๋Š” ๋‚ด์šฉ์€ "์ž์„ธํ•œ ์‚ฌํ•ญ์€ ๊ณ ๊ฐ์„ผํ„ฐ(010-8082-0047)๋กœ ๋ฌธ์˜ํ•ด์ฃผ์„ธ์š”"๋ผ๊ณ  ์•ˆ๋‚ดํ•ด์ฃผ์„ธ์š”
113
  - ๋ฐ˜๋ง๋ณด๋‹ค๋Š” ์กด๋Œ“๋ง์„ ์‚ฌ์šฉํ•ด์ฃผ์„ธ์š”
114
  - ๊ฐ„๊ฒฐํ•˜๊ณ  ์ดํ•ดํ•˜๊ธฐ ์‰ฝ๊ฒŒ ๋‹ต๋ณ€ํ•ด์ฃผ์„ธ์š”
115
- - ์ธ์‚ฌ๋ง๋กœ ์‹œ์ž‘ํ•˜๊ณ  ๊ฐ์‚ฌ์ธ์‚ฌ๋กœ ๋งˆ๋ฌด๋ฆฌํ•ด์ฃผ์„ธ์š”
116
  """
117
 
118
  print(f"์‚ฌ์šฉ์ž ์งˆ๋ฌธ: {message}")
@@ -138,13 +138,13 @@ class SimpleBanchanChatbot:
138
 
139
  # ์ฑ—๋ด‡ ์ธ์Šคํ„ด์Šค ์ƒ์„ฑ
140
  try:
141
- chatbot = SimpleBanchanChatbot()
142
 
143
  # ๊ทธ๋ผ๋””์˜ค ์ธํ„ฐํŽ˜์ด์Šค
144
  demo = gr.ChatInterface(
145
  fn=chatbot.chat_stream,
146
- title="๐Ÿฅข ๋ง›์žˆ๋Š”๋ฐ˜์ฐฌ๊ฐ€๊ฒŒ AI ์ƒ๋‹ด๋ด‡",
147
- description="์ œ๋ฏธ๋‚˜์ด 2.0 ๊ธฐ๋ฐ˜ ๋ฐ˜์ฐฌ๊ฐ€๊ฒŒ ๊ณ ๊ฐ ์ƒ๋‹ด๋ด‡์ž…๋‹ˆ๋‹ค. ๊ถ๊ธˆํ•œ ๊ฒƒ์„ ๋ฌผ์–ด๋ณด์„ธ์š”!",
148
  examples=[
149
  "์ฃผ๋ฌธ์€ ์–ด๋–ป๊ฒŒ ํ•˜๋‚˜์š”?",
150
  "ํฌ์ธํŠธ๋Š” ์–ด๋–ป๊ฒŒ ์‚ฌ์šฉํ•˜๋‚˜์š”?",
 
2
  import google.generativeai as genai
3
  import os
4
 
5
+ class KyungAhNeChatbot:
6
  def __init__(self):
7
  # ํ—ˆ๊น…ํŽ˜์ด์Šค ์ŠคํŽ˜์ด์Šค์—์„œ๋Š” secrets์—์„œ API ํ‚ค ๊ฐ€์ ธ์˜ค๊ธฐ
8
  api_key = os.getenv("GEMINI_API_KEY")
 
13
  genai.configure(api_key=api_key)
14
  self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
15
 
16
+ # ๊ฒฝ์•„๋„ค ๋ฐ˜์ฐฌ๊ฐ€๊ฒŒ FAQ ๋ฐ์ดํ„ฐ
17
  self.faq_data = """
18
+ ๊ฒฝ์•„๋„ค ๋ฐ˜์ฐฌ๊ฐ€๊ฒŒ ๊ณ ๊ฐ์„ผํ„ฐ FAQ:
19
 
20
  == ๊ธฐ๋ณธ ์ •๋ณด ==
21
  โ€ข ๊ณ ๊ฐ์„ผํ„ฐ: 010-8082-0047
 
103
 
104
  def chat_stream(self, message, history):
105
  system_prompt = f"""
106
+ ๋‹น์‹ ์€ ๊ฒฝ์•„๋„ค ๋ฐ˜์ฐฌ๊ฐ€๊ฒŒ์˜ ์นœ์ ˆํ•œ ์ƒ๋‹ด์›์ž…๋‹ˆ๋‹ค.
107
 
108
  {self.faq_data}
109
 
 
112
  - FAQ์— ์—†๋Š” ๋‚ด์šฉ์€ "์ž์„ธํ•œ ์‚ฌํ•ญ์€ ๊ณ ๊ฐ์„ผํ„ฐ(010-8082-0047)๋กœ ๋ฌธ์˜ํ•ด์ฃผ์„ธ์š”"๋ผ๊ณ  ์•ˆ๋‚ดํ•ด์ฃผ์„ธ์š”
113
  - ๋ฐ˜๋ง๋ณด๋‹ค๋Š” ์กด๋Œ“๋ง์„ ์‚ฌ์šฉํ•ด์ฃผ์„ธ์š”
114
  - ๊ฐ„๊ฒฐํ•˜๊ณ  ์ดํ•ดํ•˜๊ธฐ ์‰ฝ๊ฒŒ ๋‹ต๋ณ€ํ•ด์ฃผ์„ธ์š”
115
+ - "์•ˆ๋…•ํ•˜์„ธ์š” ๊ฒฝ์•„๋„ค์ž…๋‹ˆ๋‹ค"๋กœ ์‹œ์ž‘ํ•˜๊ณ  "๊ฐ์‚ฌํ•ฉ๋‹ˆ๋‹ค"๋กœ ๋งˆ๋ฌด๋ฆฌํ•ด์ฃผ์„ธ์š”
116
  """
117
 
118
  print(f"์‚ฌ์šฉ์ž ์งˆ๋ฌธ: {message}")
 
138
 
139
  # ์ฑ—๋ด‡ ์ธ์Šคํ„ด์Šค ์ƒ์„ฑ
140
  try:
141
+ chatbot = KyungAhNeChatbot()
142
 
143
  # ๊ทธ๋ผ๋””์˜ค ์ธํ„ฐํŽ˜์ด์Šค
144
  demo = gr.ChatInterface(
145
  fn=chatbot.chat_stream,
146
+ title="๐Ÿฅข ๊ฒฝ์•„๋„ค AI ์ƒ๋‹ด๋ด‡",
147
+ description="์ œ๋ฏธ๋‚˜์ด 2.0 ๊ธฐ๋ฐ˜ ๊ฒฝ์•„๋„ค ๋ฐ˜์ฐฌ๊ฐ€๊ฒŒ ๊ณ ๊ฐ ์ƒ๋‹ด๋ด‡์ž…๋‹ˆ๋‹ค. ๊ถ๊ธˆํ•œ ๊ฒƒ์„ ๋ฌผ์–ด๋ณด์„ธ์š”!",
148
  examples=[
149
  "์ฃผ๋ฌธ์€ ์–ด๋–ป๊ฒŒ ํ•˜๋‚˜์š”?",
150
  "ํฌ์ธํŠธ๋Š” ์–ด๋–ป๊ฒŒ ์‚ฌ์šฉํ•˜๋‚˜์š”?",