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

Merge pull request #10 from real-clh-pdt/main

Browse Source 
Minor edits
This commit is contained in:
real-clh-pdt 2024-09-15 12:31:09 +08:00 committed by GitHub
commit a85a3500c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 17 additions and 6 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_object = client.files.create(file=uploaded_file, purpose="file-extract")
file_content = json.loads(client.files.content(file_id=file_object.id).content)["content"] file_content = json.loads(client.files.content(file_id=file_object.id).content)["content"]
client.files.delete(file_id=file_object.id) 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( response = client.chat.completions.create(
model="glm-4-long", model="glm-4-long",
messages=[ messages=[
@ -48,8 +49,8 @@ def main():
], ],
) )
msg = response.choices[0].message.content msg = response.choices[0].message.content
st.session_state["messages"] = [{"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}] # 每次提问时带上全文信息 st.session_state["messages"] = [{"role": "user", "content": message_content}, {"role": "assistant", "content": msg}] # 每次提问时带上全文信息
for msg in st.session_state.get("messages", []): for msg in st.session_state.get("messages", []):
st.chat_message(msg["role"]).write(msg["content"]) 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__": if __name__ == "__main__":
# API Key, 待解析文件路径, 输出结果文件路径（excel）,输出问题路径（txt） # 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")

11
file_paper_summary.py
View File

@ -1,7 +1,7 @@
import os import os
from zhipuai import ZhipuAI # 假设你已经安装并配置好ZhipuAI from zhipuai import ZhipuAI # 假设你已经安装并配置好ZhipuAI
import argparse import argparse
import dotenv # import dotenv
import file_paper_analysis_improve import file_paper_analysis_improve
@ -22,6 +22,7 @@ def generate_summary_from_files(api_key, file_content):
print(f"文件比较出错时出错: {e}") print(f"文件比较出错时出错: {e}")
return {} return {}
def main(api_key, file_path_or_folder,output_analysis_excel,output_analysis_question,output_summary_question,output_summary_result): def main(api_key, file_path_or_folder,output_analysis_excel,output_analysis_question,output_summary_question,output_summary_result):
print("===========================开始处理综述部分===========================") print("===========================开始处理综述部分===========================")
@ -35,6 +36,7 @@ def main(api_key, file_path_or_folder,output_analysis_excel,output_analysis_ques
os.makedirs(directory) os.makedirs(directory)
print(f"目录 {directory} 已创建") print(f"目录 {directory} 已创建")
# 获取绝对路径 # 获取绝对路径
file_path_or_folder = os.path.abspath(file_path_or_folder) file_path_or_folder = os.path.abspath(file_path_or_folder)
output_summary_question = os.path.abspath(output_summary_question) 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__": if __name__ == "__main__":
# API Key, 待解析文件路径, 输出结果文件路径（excel）,输出问题路径（txt）,输出综述问题路径（txt）,输出综述结果路径（txt） # 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")