updated app.py

app.py

@@ -91,7 +91,10 @@ async def generate_story(story_request: StoryRequest):
     #     return {"story": response}
     # except Exception as e:
     #     raise HTTPException(status_code=500, detail=str(e))
-    response = chain.invoke(final_prompt)
+    # response = chain.invoke(final_prompt)
+    # async for s in chain.astream(final_prompt):
+    #     print(s.content, end="", flush=True)
+    response = await chain.ainvoke(final_prompt)
     
     if not response:
         raise HTTPException(status_code=500, detail="Failed to generate the story")