File size: 4,630 Bytes
2cf1d61 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
import openai
import gradio as gr
from datetime import datetime
import json
import chromadb
import pandas as pd
from sentence_transformers import SentenceTransformer
from tqdm import tqdm
from FlagEmbedding import BGEM3FlagModel
import re
# Đặt API key của bạn ở đây
api_key = "sk-optimi-ai-zR7o9I8TWt6vOwOHJFEpT3BlbkFJlGLNXidFLiV3E10jQv1p"
openai.api_key = api_key
# Khởi tạo mô hình embedding
# Đọc dữ liệu từ JSONL
data_path = "/home/acer/Documents/ABI/Lich_optimi/nhansu_optimi.xlsx"
data = pd.read_excel(data_path)
data['index'] = range(len(data))
# Hàm truy vấn cơ sở dữ liệu để tìm tên đầy đủ của người tham gia
def get_full_name(participant):
list_name = []
for i in range(len(data)):
if participant in data['Họ và tên'][i]:
#print(data['Họ và tên'][i])
list_name.append(data['Họ và tên'][i])
return list_name
def replace_participants(reply, dict_replace):
for key, names in dict_replace.items():
if names:
# Join names with comma for display
names_str = ','.join(names)
# Replace the key with the list of names
reply = reply.replace(key, names_str)
else:
reply = reply.replace(key,'')
parts = reply.split('Người tham gia:')
if len(parts) > 1:
participants = parts[1].strip().split(',')
cleaned_participants = [p.strip() for p in participants if p.strip()]
reply = parts[0] + 'Người tham gia: ' + ', '.join(cleaned_participants)
return reply
# Định nghĩa hệ thống prompt
def create_system_prompt():
localtime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
system_prompt = f"Tách yêu cầu sau thành 5 trường thông tin gồm: Tên cuộc họp (Ngắn gọn dưới 10 từ bao quát về mục đích cuộc họp), Ngày họp (DD/MM/YYYY), Giờ học bắt đầu (HH/MM), Giờ học kết thúc (HH/MM) và Người tham gia (Chỉ trích dẫn tên).\
Chú ý xác định thời điểm hiện tại là: {localtime} từ đó xác định ngày họp cụ thể bao gồm ngày, tháng, năm.\
Ví dụ: Tách yêu cầu 'họp với team AI, anh Hoa và giám đốc Dương design search ngày mai 10h sáng đến 3 giờ chiều.' thành:\
Tên cuộc họp: Design search.\
Ngày họp: 15/06/2024.\
Giờ bắt đầu: 10:00.\
Giờ kết thúc: 15:00.\
Người tham gia: Hoa,Dương.\
Bắt đầu tách yêu cầu sau: "
return system_prompt
# Hàm dự đoán cho Gradio
def predict(message, history):
# Tạo prompt hệ thống
system_prompt = create_system_prompt()
# Chuẩn bị lịch sử hội thoại
history_openai_format = [{"role": "system", "content": system_prompt}]
for human, assistant in history:
history_openai_format.append({"role": "user", "content": human})
history_openai_format.append({"role": "assistant", "content": assistant})
history_openai_format.append({"role": "user", "content": message})
# Gọi API của OpenAI
response = openai.ChatCompletion.create(
model='gpt-3.5-turbo',
messages=history_openai_format,
temperature=1.0
)
# Trích xuất phản hồi từ mô hình
reply = response.choices[0].message['content']
#print(reply)
#print(type(reply))
# Tách các tên người tham gia từ phản hồi
# Tách các tên người tham gia từ phản hồi
start_index = reply.find("Người tham gia:") + len("Người tham gia:")
end_index = reply.find("\n", start_index)
# Trích xuất danh sách người tham gia
participants_str = reply[start_index:end_index].strip()
# Tách danh sách thành từng phần tử
participants = [participant.strip() for participant in participants_str.split(",")]
participants = [participant.rstrip('.') for participant in participants_str.split(",")]
print(participants)
# Thay thế các tên người tham gia bằng tên đầy đủ
replace_dict = {participant: get_full_name(participant) for participant in participants}
print(replace_dict)
# replace_dict = {key: value for key, value in replace_dict.items() if value != []}
# print(replace_dict)
reply = replace_participants(reply,replace_dict)
print(reply)
# Cập nhật lịch sử hội thoại
history.append((message, reply))
return reply
# Khởi tạo giao diện Gradio
iface = gr.ChatInterface(fn=predict, title="Tách Thông Tin Cuộc Họp")
# Chạy Gradio
iface.launch(share=True)
|