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 59ee91c..031b923 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("", "", "","") + main("replace your own api key", + "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文", + "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/test.xlsx", + "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/test.txt") diff --git a/file_paper_summary.py b/file_paper_summary.py index d19343e..3a46909 100644 --- a/file_paper_summary.py +++ b/file_paper_summary.py @@ -1,7 +1,7 @@ import os from zhipuai import ZhipuAI # 假设你已经安装并配置好ZhipuAI import argparse -import dotenv +# import dotenv import file_paper_analysis_improve @@ -21,6 +21,7 @@ def generate_summary_from_files(api_key, file_content): except Exception as e: print(f"文件比较出错时出错: {e}") return {} + def main(api_key, file_path_or_folder,output_analysis_excel,output_analysis_question,output_summary_question,output_summary_result): print("===========================开始处理综述部分===========================") @@ -35,6 +36,7 @@ def main(api_key, file_path_or_folder,output_analysis_excel,output_analysis_ques os.makedirs(directory) print(f"目录 {directory} 已创建") + # 获取绝对路径 file_path_or_folder = os.path.abspath(file_path_or_folder) output_summary_question = os.path.abspath(output_summary_question) @@ -68,4 +70,9 @@ def main(api_key, file_path_or_folder,output_analysis_excel,output_analysis_ques if __name__ == "__main__": # API Key, 待解析文件路径, 输出结果文件路径(excel),输出问题路径(txt),输出综述问题路径(txt),输出综述结果路径(txt) - main("", "","","","","") + main("replace your own api key", "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文", + "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/test.xlsx", + "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/test.txt", + "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/question-summary.txt", + "/Users/lee/Desktop/documents/华师/teaching/人工智能通识课-大模型/论文/summary.txt") +