Update app.py

app.py

@@ -634,7 +634,7 @@ else:
 
 
 # Step 4: Multi-Agent Story Explorer
-st.subheader("Step 5: Explore Subtopics via Multi-Agent Graph")
+st.subheader("Step 5: Explore Subtopics via Stories")
 st.write("Select Story Language:")
 story_language_options = ["English", "Hindi", "Spanish", "French", "German", "Chinese", "Japanese"]
 story_language = st.radio(
@@ -655,7 +655,7 @@ if st.button("Run Story Graph"):
             result = st.session_state.rag_system.run_multiagent_storygraph(
                 topic=story_topic,
                 context=story_context,
-                language=story_language  # Pass selected language to story generator
+                language=story_language
             )
 
             subtopics = result.get("sub_topic_list", [])
@@ -670,29 +670,6 @@ if st.button("Run Story Graph"):
                 tabs = st.tabs([f"Story {i+1}" for i in range(len(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: