Update app.py
Browse files
app.py
CHANGED
@@ -186,36 +186,31 @@ def get_pharmkg_context(query):
|
|
186 |
return ""
|
187 |
|
188 |
|
189 |
-
|
190 |
-
|
191 |
-
return "", history
|
192 |
-
|
193 |
-
try:
|
194 |
-
# PharmKG ์ปจํ
์คํธ ์ถ๊ฐ
|
195 |
-
pharmkg_context = get_pharmkg_context(message)
|
196 |
-
system_message = SYSTEM_PREFIX + system_message + pharmkg_context
|
197 |
-
|
198 |
-
|
199 |
-
SYSTEM_PREFIX = """์ ๋ ์ฝ๋ฆฌํ ์ ๋ฌธ AI ์ด์์คํดํธ 'GiniGEN Pharm'์
๋๋ค. PharmKG ๋ฐ์ดํฐ๋ฒ ์ด์ค๋ฅผ ๊ธฐ๋ฐ์ผ๋ก
|
200 |
๋ค์๊ณผ ๊ฐ์ ์ ๋ฌธ์ฑ์ ๊ฐ์ง๊ณ ์ํตํ๊ฒ ์ต๋๋ค:
|
201 |
-
|
202 |
1. ๐ ์ฝ๋ฌผ ๊ด๋ จ ์ ๋ฌธ ์ง์ ์ ๊ณต
|
203 |
2. ๐ฌ ์ฝ๋ฌผ ์ํธ์์ฉ ๋ฐ ๋ถ์์ฉ ์ ๋ณด
|
204 |
3. ๐งฌ ์ง๋ณ-์ฝ๋ฌผ ๊ด๊ณ ์ค๋ช
|
205 |
4. ๐ ์์ ๋ฐ์ดํฐ ๊ธฐ๋ฐ ๋ถ์
|
206 |
5. โ๏ธ ์์ฝํ ์์ ์ฑ ์ ๋ณด
|
207 |
-
|
208 |
๋ค์ ์์น์ผ๋ก ์ํตํ๊ฒ ์ต๋๋ค:
|
209 |
1. ๐ค ์ ๋ฌธ์ ์ด๊ณ ์ ๋ขฐํ ์ ์๋ ์ ๋ณด ์ ๊ณต
|
210 |
2. ๐ก ์ดํดํ๊ธฐ ์ฌ์ด ์ค๋ช
๊ณผ ํจ๊ป ๊ณผํ์ ๊ทผ๊ฑฐ ์ ์
|
211 |
3. ๐ฏ ์ง๋ฌธ์ ์๋๋ฅผ ์ ํํ ํ์
ํ์ฌ ๋ง์ถคํ ๋ต๋ณ
|
212 |
4. โ ๏ธ ์๋ฃ์ ์กฐ์ธ์ด ์๋ ์ ๋ณด ์ ๊ณต ๋ชฉ์ ์์ ๋ช
์
|
213 |
5. โจ ์ถ๊ฐ ์ฐธ๊ณ ์๋ฃ ๋ฐ ์ฐ๊ตฌ ๋ฐ์ดํฐ ์ ์
|
214 |
-
|
215 |
์ํ์ ๊ฒฐ์ ์ ๋ฐ๋์ ์ ๋ฌธ ์๋ฃ์ง๊ณผ ์๋ดํ์๊ธฐ ๋ฐ๋๋๋ค."""
|
216 |
|
|
|
|
|
|
|
217 |
|
218 |
try:
|
|
|
|
|
|
|
|
|
219 |
# ํ์ผ ์
๋ก๋ ์ฒ๋ฆฌ
|
220 |
if uploaded_file:
|
221 |
content, file_type = read_uploaded_file(uploaded_file)
|
@@ -241,7 +236,7 @@ def chat(message, history, uploaded_file, system_message="", max_tokens=4000, te
|
|
241 |
5. ๐ฌ ์ถ๊ฐ ์ง๋ฌธ์ด๋ ํ์ํ ์ค๋ช
"""
|
242 |
|
243 |
# ๋ฉ์์ง ์ฒ๋ฆฌ
|
244 |
-
messages = [{"role": "system", "content":
|
245 |
|
246 |
# ์ด์ ๋ํ ํ์คํ ๋ฆฌ ์ถ๊ฐ
|
247 |
if history:
|
|
|
186 |
return ""
|
187 |
|
188 |
|
189 |
+
# SYSTEM_PREFIX๋ฅผ ์ ์ญ ๋ณ์๋ก ์ด๋
|
190 |
+
SYSTEM_PREFIX = """์ ๋ ์ฝ๋ฆฌํ ์ ๋ฌธ AI ์ด์์คํดํธ 'GiniGEN Pharm'์
๋๋ค. PharmKG ๋ฐ์ดํฐ๋ฒ ์ด์ค๋ฅผ ๊ธฐ๋ฐ์ผ๋ก
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
191 |
๋ค์๊ณผ ๊ฐ์ ์ ๋ฌธ์ฑ์ ๊ฐ์ง๊ณ ์ํตํ๊ฒ ์ต๋๋ค:
|
|
|
192 |
1. ๐ ์ฝ๋ฌผ ๊ด๋ จ ์ ๋ฌธ ์ง์ ์ ๊ณต
|
193 |
2. ๐ฌ ์ฝ๋ฌผ ์ํธ์์ฉ ๋ฐ ๋ถ์์ฉ ์ ๋ณด
|
194 |
3. ๐งฌ ์ง๋ณ-์ฝ๋ฌผ ๊ด๊ณ ์ค๋ช
|
195 |
4. ๐ ์์ ๋ฐ์ดํฐ ๊ธฐ๋ฐ ๋ถ์
|
196 |
5. โ๏ธ ์์ฝํ ์์ ์ฑ ์ ๋ณด
|
|
|
197 |
๋ค์ ์์น์ผ๋ก ์ํตํ๊ฒ ์ต๋๋ค:
|
198 |
1. ๐ค ์ ๋ฌธ์ ์ด๊ณ ์ ๋ขฐํ ์ ์๋ ์ ๋ณด ์ ๊ณต
|
199 |
2. ๐ก ์ดํดํ๊ธฐ ์ฌ์ด ์ค๋ช
๊ณผ ํจ๊ป ๊ณผํ์ ๊ทผ๊ฑฐ ์ ์
|
200 |
3. ๐ฏ ์ง๋ฌธ์ ์๋๋ฅผ ์ ํํ ํ์
ํ์ฌ ๋ง์ถคํ ๋ต๋ณ
|
201 |
4. โ ๏ธ ์๋ฃ์ ์กฐ์ธ์ด ์๋ ์ ๋ณด ์ ๊ณต ๋ชฉ์ ์์ ๋ช
์
|
202 |
5. โจ ์ถ๊ฐ ์ฐธ๊ณ ์๋ฃ ๋ฐ ์ฐ๊ตฌ ๋ฐ์ดํฐ ์ ์
|
|
|
203 |
์ํ์ ๊ฒฐ์ ์ ๋ฐ๋์ ์ ๋ฌธ ์๋ฃ์ง๊ณผ ์๋ดํ์๊ธฐ ๋ฐ๋๋๋ค."""
|
204 |
|
205 |
+
def chat(message, history, uploaded_file, system_message="", max_tokens=4000, temperature=0.7, top_p=0.9):
|
206 |
+
if not message:
|
207 |
+
return "", history
|
208 |
|
209 |
try:
|
210 |
+
# PharmKG ์ปจํ
์คํธ ์ถ๊ฐ
|
211 |
+
pharmkg_context = get_pharmkg_context(message)
|
212 |
+
system_message = SYSTEM_PREFIX + system_message + pharmkg_context
|
213 |
+
|
214 |
# ํ์ผ ์
๋ก๋ ์ฒ๋ฆฌ
|
215 |
if uploaded_file:
|
216 |
content, file_type = read_uploaded_file(uploaded_file)
|
|
|
236 |
5. ๐ฌ ์ถ๊ฐ ์ง๋ฌธ์ด๋ ํ์ํ ์ค๋ช
"""
|
237 |
|
238 |
# ๋ฉ์์ง ์ฒ๋ฆฌ
|
239 |
+
messages = [{"role": "system", "content": system_message}]
|
240 |
|
241 |
# ์ด์ ๋ํ ํ์คํ ๋ฆฌ ์ถ๊ฐ
|
242 |
if history:
|