ssboost commited on
Commit
ef35a5f
ยท
verified ยท
1 Parent(s): 37cd1bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -29
app.py CHANGED
@@ -115,40 +115,26 @@ class SimpleBanchanChatbot:
115
  - ์ธ์‚ฌ๋ง๋กœ ์‹œ์ž‘ํ•˜๊ณ  ๊ฐ์‚ฌ์ธ์‚ฌ๋กœ ๋งˆ๋ฌด๋ฆฌํ•ด์ฃผ์„ธ์š”
116
  """
117
 
 
 
 
118
  try:
119
  response = self.model.generate_content(
120
- contents=f"{system_prompt}\n\n์‚ฌ์šฉ์ž ์งˆ๋ฌธ: {message}",
121
- generation_config=genai.GenerationConfig(
122
- temperature=0.1,
123
- max_output_tokens=1000,
124
- ),
125
- stream=True
126
  )
127
 
128
- partial_message = ""
129
- for chunk in response:
130
- try:
131
- if chunk.text:
132
- partial_message += chunk.text
133
- yield partial_message
134
- except:
135
- continue
136
-
137
  except Exception as e:
138
- print(f"์—๋Ÿฌ ๋ฐœ์ƒ: {str(e)}")
139
- try:
140
- response = self.model.generate_content(
141
- contents=f"{system_prompt}\n\n์‚ฌ์šฉ์ž ์งˆ๋ฌธ: {message}",
142
- generation_config=genai.GenerationConfig(
143
- temperature=0.1,
144
- max_output_tokens=1000,
145
- ),
146
- stream=False
147
- )
148
- yield response.text
149
- except Exception as fallback_error:
150
- print(f"Fallback ์—๋Ÿฌ: {str(fallback_error)}")
151
- yield f"์ฃ„์†กํ•ฉ๋‹ˆ๋‹ค. ์‹œ์Šคํ…œ ์˜ค๋ฅ˜๊ฐ€ ๋ฐœ์ƒํ–ˆ์Šต๋‹ˆ๋‹ค. ๊ณ ๊ฐ์„ผํ„ฐ(010-8082-0047)๋กœ ์—ฐ๋ฝํ•ด์ฃผ์„ธ์š”."
152
 
153
  # ์ฑ—๋ด‡ ์ธ์Šคํ„ด์Šค ์ƒ์„ฑ
154
  try:
 
115
  - ์ธ์‚ฌ๋ง๋กœ ์‹œ์ž‘ํ•˜๊ณ  ๊ฐ์‚ฌ์ธ์‚ฌ๋กœ ๋งˆ๋ฌด๋ฆฌํ•ด์ฃผ์„ธ์š”
116
  """
117
 
118
+ print(f"์‚ฌ์šฉ์ž ์งˆ๋ฌธ: {message}")
119
+ print(f"API ํ‚ค ์ƒํƒœ: {'์„ค์ •๋จ' if os.getenv('GEMINI_API_KEY') else '์—†์Œ'}")
120
+
121
  try:
122
  response = self.model.generate_content(
123
+ contents=f"{system_prompt}\n\n์‚ฌ์šฉ์ž ์งˆ๋ฌธ: {message}"
 
 
 
 
 
124
  )
125
 
126
+ print(f"์‘๋‹ต ๋ฐ›์Œ: {response.text[:100]}...")
127
+ yield response.text
128
+
 
 
 
 
 
 
129
  except Exception as e:
130
+ error_msg = str(e)
131
+ print(f"์—๋Ÿฌ ๋ฐœ์ƒ: {error_msg}")
132
+
133
+ yield f"""์ฃ„์†กํ•ฉ๋‹ˆ๋‹ค. ๊ธฐ์ˆ ์  ๋ฌธ์ œ๊ฐ€ ๋ฐœ์ƒํ–ˆ์Šต๋‹ˆ๋‹ค.
134
+
135
+ ์—๋Ÿฌ ์ •๋ณด: {error_msg}
136
+
137
+ ๊ณ ๊ฐ์„ผํ„ฐ(010-8082-0047)๋กœ ์—ฐ๋ฝํ•ด์ฃผ์„ธ์š”."""
 
 
 
 
 
 
138
 
139
  # ์ฑ—๋ด‡ ์ธ์Šคํ„ด์Šค ์ƒ์„ฑ
140
  try: