Update app.py

app.py

@@ -167,17 +167,20 @@ class Reviewer:
                     cnt += 1
                 text = retrieval_content + content
                 chat_review_limitations = self.chat_review(text=text)  
+                chat_review_text = self.chat_refine(text=text, limitations=chat_review_limitations)
             else:
                 text = content
                 chat_review_limitations = self.chat_review(text=text)   
                 retrieved_papers = "" 
+                chat_review_text = self.chat_refine(text=text, limitations=chat_review_limitations)
         else:
             text = content
             chat_review_limitations = self.chat_review(text=text)   
             retrieved_papers = ""   
+            chat_review_text = self.chat_refine(text=text, limitations=chat_review_limitations)
 
-        text = f"Paper:\n{paper['content']}\n\n"
-        chat_review_text = self.chat_refine(text=text, limitations=chat_review_limitations)
+        # text = f"Paper:\n{paper['content']}\n\n"
+        # chat_review_text = self.chat_refine(text=text, limitations=chat_review_limitations)
         return chat_review_text, retrieved_papers
 
     def query_gen(self, abstract):
@@ -418,7 +421,7 @@ Organize the result in JSON format as follows:
         messages = []
         for limitation in limitations:
             message=[
-                    {"role": "system", "content": f"Read the following scientific paper and a limitation of the paper. Based on this information, provide a highly specific and actionable suggestion to address the limitation. If suggesting an additional dataset, specify the exact dataset(s) by name. If proposing a methodological change, describe the specific modification. Keep the response within 50 words."} ,
+                    {"role": "system", "content": f"Read the following scientific paper, its limitation, and several relevant papers to gain knowledge of the relevant field. Using insights from the relevant papers, provide a highly specific and actionable suggestion to address the limitation in the paper to review. If suggesting an additional dataset, specify the exact dataset(s) by name. If proposing a methodological change, describe the specific modification. Keep the response within 50 words."} ,
                     {"role": "user", "content": f"{text}\nLimitation: {limitation}"},
                 ]
             messages.append(message)