Update app.py

app.py

@@ -112,8 +112,9 @@ class DocumentRAG:
         except Exception as e:
             return f"Error processing documents: {str(e)}"
 
+
     def generate_summary(self, text, language):
-        """Generate a summary of the provided text in the specified language."""
+        """Generate a summary of the provided text focusing on specific sections in the specified language."""
         if not self.api_key:
             return "API Key not set. Please set it in the environment variables."
         try:
@@ -121,12 +122,22 @@ class DocumentRAG:
             response = client.chat.completions.create(
                 model="gpt-4",
                 messages=[
-                    {"role": "system", "content": f"Summarize the document content concisely in {language}. Provide 5-7 key points for discussion."},
+                    {"role": "system", "content": f"""
+                    Summarize the following document focusing mainly using the following sections:
+                1. Abstract
+                2. In the introduction section, focus specifically on the portion where the key contributions of the research paper are highlighted.
+                3. Conclusion
+                4. Limitations
+                5. Future Work
+                
+                Ensure the summary is concise, logically ordered, and suitable for {language}. 
+                Provide 5-7 key points for discussion in a structured format."""
+                },
                     {"role": "user", "content": text[:4000]}
                 ],
                 temperature=0.3
-            )
-            return response.choices[0].message.content
+                )
+                return response.choices[0].message.content
         except Exception as e:
             return f"Error generating summary: {str(e)}"