Update app.py

app.py

@@ -556,29 +556,51 @@ if st.button("Process Documents"):
         st.warning("No files uploaded.")
 
 # Step 2: Generate Summary
-st.subheader("Step 2: Generate Summary")
-st.write("Select Summary Language:")
-summary_language_options = ["English", "Hindi", "Spanish", "French", "German", "Chinese", "Japanese"]
-summary_language = st.radio(
-    "", 
-    summary_language_options, 
-    horizontal=True, 
-    key="summary_language"
-)
-
 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, please wait..."):
-            summary = st.session_state.rag_system.generate_summary(st.session_state.rag_system.document_text, summary_language)
-        if summary:
-            st.session_state.rag_system.document_summary = summary
-            st.text_area("Document Summary", summary, height=200)
-            st.success("Summary generated successfully!")
-        else:
-            st.error("Failed to generate summary.")
+        with st.spinner("Generating summary and visual..."):
+            summary = st.session_state.rag_system.generate_summary(
+                st.session_state.rag_system.document_text,
+                summary_language
+            )
+
+            if summary:
+                st.session_state.rag_system.document_summary = summary
+                st.text_area("Document Summary", summary, height=250)
+                st.success("Summary generated successfully!")
+
+                # Automatically generate Mind Map after summary
+                st.markdown("### 🧠 Mind Map of the Summary")
+
+                mindmap_prompt = f"""
+                A hand-drawn colorful mind map diagram, educational style, clear hierarchy.
+                Language: {summary_language}
+                Topic: Research Summary
+                ├── ABSTRACT
+                ├── CONTRIBUTIONS
+                ├── RESULTS
+                ├── CONCLUSION
+                ├── LIMITATIONS
+                └── FUTURE WORK
+                """
+
+                mindmap_image, _ = infer(
+                    prompt=mindmap_prompt,
+                    seed=42,
+                    randomize_seed=True,
+                    width=1024,
+                    height=1024,
+                    num_inference_steps=4
+                )
+
+                st.image(mindmap_image, caption="Mind Map Diagram", use_column_width=True)
+
+            else:
+                st.error("Failed to generate summary.")
     else:
         st.info("Please process documents first to generate summary.")
 
+
 # Step 3: Ask Questions
 st.subheader("Step 3: Ask Questions")
 st.write("Select Q&A Language:")