Update app.py

app.py

@@ -385,9 +385,7 @@ class DocumentRAG:
 
 
 
-
-
-    def generate_story_node(self, state: StoryState):
+    def generate_story_node(self, state: StoryState, language="English"):
         context = "\n\n".join(state.reranked_docs)
         topic = state.story_topic
 
@@ -395,6 +393,7 @@ class DocumentRAG:
         Suppose you're a brilliant science storyteller.
         You write stories that help middle schoolers understand complex science topics with fun and clarity.
         Add subtle humor and make it engaging.
+        Write the story in {language}.
         """
         prompt = f"""
         Use the following context to write a fun and simple story explaining **{topic}** to a middle schooler:\n
@@ -404,7 +403,6 @@ class DocumentRAG:
 
         msg = self.llm.invoke([SystemMessage(system_message), HumanMessage(prompt)])
         return {"stories": msg}
-