File size: 2,425 Bytes
9e075cc ef651bf 9e075cc ef651bf 9e075cc ef651bf 9e075cc ef651bf 9e075cc ef651bf 9e075cc e9b663f ef651bf |
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 |
from TempSummary import generate_temp_summary
import time
def generate_summary(llm, file):
print("Generating temporary summary...")
temp_summary, length_of_research_paper = generate_temp_summary(file)
print("Temporary summary generated successfully")
print("Generating final summary...")
prompt = f"As a text script expert, please help me to write a short text script with the research paper \"{temp_summary}\".You have three tasks, which are:\\n 1.to summarize the text I provided into a Summary . Please answer within 200-300 characters.\\n 2.to summarize the text I provided, using up to seven concise Highlights. Choose appropriate emoji for each Highlight.\\n 3.to summarize the text I provided, using up to seven Key Insights. Each insight should include a brief in-depth analysis. Choose appropriate emoji for each key insights. Using the following template strictly, provide the results for the three tasks:\\n ### Summary\\n ### Highlights -[emoji]\\n ### key Insights -[emoji] .\\n Importantly your output must use language \"English\"\""
response = llm.create_chat_completion(
messages = [
{'role':'system',
'content': 'You are a helpful research assistant for generating well-formatted summaries from scientific research papers.'},
{'role':'user',
'content': prompt}
],
temperature=0.5,
top_k=200,
top_p=3.0,
)
summary = response['choices'][0]['message']['content']
return summary, length_of_research_paper
def summarize(llm, file):
import time
start_time = time.time()
response, length_of_research_paper = generate_summary(llm, file)
if "**" in response:
response = response.replace("- **", "### ")
response = response.replace("**", "")
response = response.replace("\n\n", "\n")
response = response.replace("\\n\\n", "\\n")
summary = ""
for line in response.splitlines():
if line.startswith("###"):
summary += "\n\n" + line
else:
summary += "\n" + line
end_time = time.time()
total_time_taken = end_time - start_time
total_time_taken_minutes = round(total_time_taken / 60, 3)
info = f"The research paper of {length_of_research_paper} characters long was summarized in {total_time_taken_minutes} minutes."
print(info)
return summary.strip(), info
|