Update app.py

app.py

@@ -173,32 +173,48 @@ class DocumentRAG:
             return f"Error processing documents: {str(e)}"
 
     def generate_summary(self, text, language):
-        """Generate a summary of the provided text focusing on specific sections in the specified language."""
+        """Generate a structured summary from all chunks of the document."""
         if not self.api_key:
             return "API Key not set. Please set it in the environment variables."
+
         try:
             client = OpenAI(api_key=self.api_key)
-            response = client.chat.completions.create(
-                model="gpt-4",
-                messages=[
-                    {"role": "system", "content": f"""
-                    Summarize the following document focusing mainly on these sections:
-                    1. Abstract
-                    2. In the Introduction, specifically focus 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 7-9 key points for discussion in a structured format."""},
-                    {"role": "user", "content": text[:4000]}
-                ],
-                temperature=0.3
-            )
-            return response.choices[0].message.content
+
+            # Split into chunks
+            chunks = [text[i:i + 3000] for i in range(0, len(text), 3000)]
+            summaries = []
+
+            for i, chunk in enumerate(chunks):
+                response = client.chat.completions.create(
+                    model="gpt-4",
+                    messages=[
+                        {"role": "system", "content": f"""
+                            You are a scientific summarization assistant. 
+                            Summarize the input below in {language} in a structured format, covering:
+                            - Abstract (if present)
+                            - Key Contributions
+                            - Results/Findings
+                            - Conclusion
+                            - Limitations
+                            - Future Work
+                            
+                            If any section is missing, just skip it. Keep the language clear and concise.
+                        """},
+                        {"role": "user", "content": chunk}
+                    ],
+                    temperature=0.4
+                )
+
+                content = response.choices[0].message.content.strip()
+                summaries.append(f"### Part {i+1}\n{content}")
+
+            full_summary = "\n\n".join(summaries)
+            return full_summary
+
         except Exception as e:
             return f"Error generating summary: {str(e)}"
 
+
     def create_podcast(self, language):
         """Generate a podcast script and audio based on doc summary in the specified language."""
         if not self.document_summary: