Update app.py
Browse files
app.py
CHANGED
@@ -60,14 +60,14 @@ def extract_key_contents(script, num_contents=10):
|
|
60 |
Trích xuất các ý chính từ script.
|
61 |
"""
|
62 |
try:
|
63 |
-
response = client.completions.create(
|
64 |
model="gpt-3.5-turbo",
|
65 |
messages=[
|
66 |
{"role": "system", "content": f"Bạn là một chuyên gia phân tích nội dung. Hãy trích xuất chính xác {num_contents} ý chính quan trọng nhất từ đoạn văn sau, mỗi ý không quá 20 từ."},
|
67 |
{"role": "user", "content": script}
|
68 |
]
|
69 |
)
|
70 |
-
key_contents = response.choices[0].message
|
71 |
return key_contents[:num_contents]
|
72 |
except Exception as e:
|
73 |
print(f"Lỗi khi trích xuất nội dung: {str(e)}")
|
|
|
60 |
Trích xuất các ý chính từ script.
|
61 |
"""
|
62 |
try:
|
63 |
+
response = client.chat.completions.create(
|
64 |
model="gpt-3.5-turbo",
|
65 |
messages=[
|
66 |
{"role": "system", "content": f"Bạn là một chuyên gia phân tích nội dung. Hãy trích xuất chính xác {num_contents} ý chính quan trọng nhất từ đoạn văn sau, mỗi ý không quá 20 từ."},
|
67 |
{"role": "user", "content": script}
|
68 |
]
|
69 |
)
|
70 |
+
key_contents = response.choices[0].message['content'].split('\n')
|
71 |
return key_contents[:num_contents]
|
72 |
except Exception as e:
|
73 |
print(f"Lỗi khi trích xuất nội dung: {str(e)}")
|