updated prompt to generate summary and takeaway with bug fixes

app.py

@@ -54,30 +54,34 @@ def generate(
     )
     seed = seed + 1
 
-    arxiv_key = prompt.strip().split(".")
-    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"
+    try:
 
+        arxiv_key = prompt.strip().split(".")
+        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"
+
+    except Exception as e:
+        print(Exception + " " + e)
+        prompt = "How can you help me with research."
     formatted_prompt = format_prompt(prompt, history, system_prompt)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)