Update app.py

app.py

@@ -538,23 +538,50 @@ if st.button("Process Documents"):
         st.warning("No files uploaded.")
 
 # Step 2: Generate Summary
-if st.button("Generate Summary"):
-    if hasattr(st.session_state.rag_system, "document_text") and st.session_state.rag_system.document_text:
-        with st.spinner("Generating summary and visual..."):
-            summary = st.session_state.rag_system.generate_summary(
-                st.session_state.rag_system.document_text,
-                summary_language
+def generate_summary(self, text, 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)
+
+        # Split into chunks to avoid token limit issues
+        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 multilingual academic summarization assistant.
+
+Your job is to summarize scientific or technical documents in a structured and concise way, **in {language}**.
+
+Follow this structured format if applicable:
+- Abstract
+- Key Contributions
+- Results/Findings
+- Conclusion
+- Limitations
+- Future Work
+
+If a section is not present in the document, skip it without guessing. Keep language formal, clear, and audience-appropriate.
+                    """},
+                    {"role": "user", "content": chunk}
+                ],
+                temperature=0.4
             )
 
-            if summary:
-                st.session_state.rag_system.document_summary = summary
-                st.text_area("Document Summary", summary, height=250)
-                st.success("Summary generated successfully!")
+            summaries.append(f"### Part {i+1}\n{response.choices[0].message.content.strip()}")
+
+        return "\n\n".join(summaries)
+
+    except Exception as e:
+        return f"Error generating summary: {str(e)}"
+
 
-            else:
-                st.error("Failed to generate summary.")
-    else:
-        st.info("Please process documents first to generate summary.")
 
 
 # Step 3: Ask Questions