ssboost commited on
Commit
5cb4fa8
ยท
verified ยท
1 Parent(s): d961d8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +40 -1
app.py CHANGED
@@ -1,4 +1,42 @@
1
- import gradio as gr
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  import google.generativeai as genai
3
  import os
4
 
@@ -11,6 +49,7 @@ class SimpleBanchanChatbot:
11
 
12
  # ์ œ๋ฏธ๋‚˜์ด ํด๋ผ์ด์–ธํŠธ ์„ค์ •
13
  genai.configure(api_key=api_key)
 
14
  self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
15
 
16
  # ์—‘์…€์—์„œ ์ถ”์ถœํ•œ ๋ชจ๋“  FAQ ๋ฐ์ดํ„ฐ
 
1
+ try:
2
+ # 2.0 ๋ฒ„์ „์šฉ ์•ˆ์ •์ ์ธ API ํ˜ธ์ถœ
3
+ response = self.model.generate_content(
4
+ contents=f"{system_prompt}\n\n์‚ฌ์šฉ์ž ์งˆ๋ฌธ: {message}",
5
+ generation_config=genai.GenerationConfig(
6
+ temperature=0.1,
7
+ max_output_tokens=1000,
8
+ ),
9
+ stream=True
10
+ )
11
+
12
+ partial_message = ""
13
+ for chunk in response:
14
+ try:
15
+ if chunk.text:
16
+ partial_message += chunk.text
17
+ yield partial_message
18
+ except Exception as chunk_error:
19
+ # ๊ฐœ๋ณ„ ์ฒญํฌ ์—๋Ÿฌ๋Š” ๋ฌด์‹œํ•˜๊ณ  ๊ณ„์† ์ง„ํ–‰
20
+ continue
21
+
22
+ except Exception as e:
23
+ # ์—๋Ÿฌ ๋‚ด์šฉ์„ ๋” ์ž์„ธํžˆ ๋ณด๊ธฐ ์œ„ํ•ด
24
+ print(f"์—๋Ÿฌ ๋ฐœ์ƒ: {str(e)}")
25
+
26
+ # ์ŠคํŠธ๋ฆฌ๋ฐ์ด ์•ˆ๋˜๋ฉด ์ผ๋ฐ˜ ๋ฐฉ์‹์œผ๋กœ ์‹œ๋„
27
+ try:
28
+ response = self.model.generate_content(
29
+ contents=f"{system_prompt}\n\n์‚ฌ์šฉ์ž ์งˆ๋ฌธ: {message}",
30
+ generation_config=genai.GenerationConfig(
31
+ temperature=0.1,
32
+ max_output_tokens=1000,
33
+ ),
34
+ stream=False
35
+ )
36
+ yield response.text
37
+ except Exception as fallback_error:
38
+ print(f"Fallback ์—๋Ÿฌ: {str(fallback_error)}")
39
+ yield f"์ฃ„์†กํ•ฉ๋‹ˆ๋‹ค. ์‹œ์Šคํ…œ ์˜ค๋ฅ˜๊ฐ€ ๋ฐœ์ƒํ–ˆ์Šต๋‹ˆ๋‹ค. ๊ณ ๊ฐ์„ผํ„ฐ(010-8082-0047)๋กœ ์—ฐ๋ฝํ•ด์ฃผ์„ธ์š”."import gradio as gr
40
  import google.generativeai as genai
41
  import os
42
 
 
49
 
50
  # ์ œ๋ฏธ๋‚˜์ด ํด๋ผ์ด์–ธํŠธ ์„ค์ •
51
  genai.configure(api_key=api_key)
52
+ # 2.0 ๋ฒ„์ „ ์‚ฌ์šฉ
53
  self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
54
 
55
  # ์—‘์…€์—์„œ ์ถ”์ถœํ•œ ๋ชจ๋“  FAQ ๋ฐ์ดํ„ฐ