RYDE-WORK/FileChat
1
0
Fork 0
mirror of https://github.com/primedigitaltech/FileChat.git synced 2026-01-19 13:03:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

minor edits

Browse Source
This commit is contained in:
18688905432 2024-09-15 12:06:23 +08:00
parent 842d6e5709
commit 1097c8a99d
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
file_chat.py
View File

@ -40,7 +40,8 @@ def main():
file_object = client.files.create(file=uploaded_file, purpose="file-extract")
file_content = json.loads(client.files.content(file_id=file_object.id).content)["content"]
client.files.delete(file_id=file_object.id)
message_content = f"请对\n{file_content}\n的内容进行分析，并撰写一份论文摘要。"
# message_content = f"请对\n{file_content}\n的内容进行分析，并撰写一份论文摘要。"
message_content = f"你是人工智能领域的专家，请对\n{file_content}\n的内容进行分析,提取论文的摘要原文,摘要在Abstract之后,Introduction之前。"
response = client.chat.completions.create(
model="glm-4-long",
messages=[
@ -48,8 +49,8 @@ def main():
],
)
msg = response.choices[0].message.content
st.session_state["messages"] = [{"role": "assistant", "content": msg}] # 每次提问时不带上全文信息
# st.session_state["messages"] = [{"role": "user", "content": message_content}, {"role": "assistant", "content": msg}] # 每次提问时带上全文信息
# st.session_state["messages"] = [{"role": "assistant", "content": msg}] # 每次提问时不带上全文信息
st.session_state["messages"] = [{"role": "user", "content": message_content}, {"role": "assistant", "content": msg}] # 每次提问时带上全文信息
for msg in st.session_state.get("messages", []):
st.chat_message(msg["role"]).write(msg["content"])

5
file_paper_analysis_improve.py
View File

@ -165,4 +165,7 @@ def main(api_key, file_path_or_folder, output_excel,output_question):
if __name__ == "__main__":
# API Key, 待解析文件路径, 输出结果文件路径（excel）,输出问题路径（txt）
main("f3489a829804ad775c645151fa17495a.AzE83g1jBq4PqSyY", "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文", "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/test.xlsx","/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/test.txt")
main("f3489a829804ad775c645151fa17495a.AzE83g1jBq4PqSyY",
"/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文",
"/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/test.xlsx",
"/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/test.txt")