diff --git a/file_chat.py b/file_chat.py index 88945df..278f78c 100644 --- a/file_chat.py +++ b/file_chat.py @@ -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"]) diff --git a/file_paper_analysis_improve.py b/file_paper_analysis_improve.py index 323d96d..c63f66f 100644 --- a/file_paper_analysis_improve.py +++ b/file_paper_analysis_improve.py @@ -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")