Update app.py

app.py

@@ -644,7 +644,11 @@ if st.button("Run Story Graph"):
         st.warning("Please process documents first before running the story graph.")
     else:
         with st.spinner("Generating subtopics and stories..."):
-            result = st.session_state.rag_system.run_multiagent_storygraph(topic=story_topic, context=story_context)
+            result = st.session_state.rag_system.run_multiagent_storygraph(
+                topic=story_topic,
+                context=story_context,
+                language=story_language  # Pass selected language to story generator
+            )
 
             subtopics = result.get("sub_topic_list", [])
             st.markdown("### 🧠 Extracted Subtopics")
@@ -655,16 +659,40 @@ if st.button("Run Story Graph"):
             if stories:
                 st.markdown("### 📚 Generated Stories")
 
-                # Present stories in tabs
                 tabs = st.tabs([f"Story {i+1}" for i in range(len(stories))])
-                for tab, story in zip(tabs, stories):
+                for i, (tab, story) in enumerate(zip(tabs, stories)):
                     with tab:
+                        st.markdown("#### 🎨 Concept Visual")
+
+                        # Prompt for the diagram using story language
+                        visual_prompt = f"""
+                        A colorful hand-drawn educational illustration in {story_language}, labeled and clear.
+                        Topic: {story_topic}
+                        Subtopic: {subtopics[i] if i < len(subtopics) else 'Concept'}
+                        Style: for middle school students, simple icons, visually intuitive.
+                        """
+
+                        try:
+                            image, _ = infer(
+                                prompt=visual_prompt,
+                                seed=42,
+                                randomize_seed=True,
+                                width=1024,
+                                height=1024,
+                                num_inference_steps=4
+                            )
+                            st.image(image, caption="Concept Visual", use_column_width=True)
+                        except Exception as e:
+                            st.error(f"Diagram generation failed: {e}")
+
+                        st.markdown("#### ✍️ Story")
                         st.markdown(story.content)
             else:
                 st.warning("No stories were generated.")
 
 
 
+
 # Step 5: Generate Podcast
 st.subheader("Step 4: Generate Podcast")
 st.write("Select Podcast Language:")