updated prompt to generate summary and takeaway

app.py

@@ -55,10 +55,29 @@ def generate(
     seed = seed + 1
 
     arxiv_key = prompt.strip().split(".")
-    if len(arxiv_key) != 2:
+    if len(arxiv_key) <= 2:
         output = "Please enter a valid arXiv ID."
         return output
     
+
+    arxiv_id = prompt
+    docs = ArxivLoader(query="1605.08386", load_max_docs=2).load()
+    doc = docs[0]
+
+    title = doc["Title"]
+    abstract = doc["Summary"]
+    authors = doc["Authors"]
+    published = doc["Published"]
+
+    prompt = "You are a RESEARCH Assistant, who is tasked with Summarizing a scholarly research article paper. You are given the following information:\n"
+    prompt += f"Title: {title}\n"
+    prompt += f"Authors: {authors}\n"
+    prompt += f"Published: {published}\n"
+    prompt += f"Abstract: {abstract}\n"
+    prompt += f"Your task is to summarize the paper in 3-4 bullet points, generating key takeaways \n"
+    prompt += f"Generate a FUTURE IDEA for the paper, that can be used to extend the paper. \n"
+
+    
     formatted_prompt = format_prompt(prompt, history, system_prompt)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)