hoduyquocbao commited on
Commit
5121e98
1 Parent(s): b09c08f

fix new features

Browse files
Files changed (1) hide show
  1. app.py +14 -5
app.py CHANGED
@@ -227,12 +227,21 @@ def handle_functions(function_call: Dict[str, Any], prompt: str, chat_history: L
227
  yield "📄 **Kết quả tìm kiếm:**\n" + web_summary
228
 
229
  elif function_name == "summarize_query":
230
- prompt_text = arguments["prompt"]
 
 
 
 
 
 
 
 
 
 
 
 
231
  yield "📝 Đang tóm tắt thông tin..."
232
- # Trích xuất thông tin cần tóm tắt từ prompt (giả định phần tóm tắt nằm sau từ khóa)
233
- # Ví dụ: "tóm tắt nội dung về AI"
234
- # Bạn có thể cải thiện logic này tùy theo cách bạn muốn trích xuất nội dung để tóm tắt
235
- summary = summarize_text(prompt_text)
236
  yield "📄 **Tóm tắt:**\n" + summary
237
 
238
  elif function_name == "sentiment_analysis":
 
227
  yield "📄 **Kết quả tìm kiếm:**\n" + web_summary
228
 
229
  elif function_name == "summarize_query":
230
+ # Khi người dùng yêu cầu tóm tắt, hệ thống sẽ thực hiện tìm kiếm và sau đó tóm tắt kết quả
231
+ query = arguments["prompt"]
232
+ yield "🔍 Đang thực hiện tìm kiếm để tóm tắt..."
233
+ web_results = search(query)
234
+ if not web_results:
235
+ yield "⚠️ Không tìm thấy kết quả để tóm tắt."
236
+ return
237
+ # Lấy nội dung từ kết quả tìm kiếm để tóm tắt
238
+ combined_text = ' '.join([res['text'] for res in web_results if res['text'] != "Không thể lấy nội dung."])
239
+ if not combined_text:
240
+ yield "⚠️ Không có nội dung để tóm tắt."
241
+ return
242
+ # Tóm tắt nội dung đã lấy
243
  yield "📝 Đang tóm tắt thông tin..."
244
+ summary = summarize_text(combined_text)
 
 
 
245
  yield "📄 **Tóm tắt:**\n" + summary
246
 
247
  elif function_name == "sentiment_analysis":