updated app.py

app.py

@@ -64,7 +64,7 @@ Total pages in storie are seven each page have one short paragraph and dont ask
 to create another adventure soon! 
 """
 
-prompt_template = ChatPromptTemplate.from_messages([("system", system), ("human", "{text}")])
+# prompt_template = ChatPromptTemplate.from_messages([("system", system), ("human", "{text}")])
 
 @app.get("/")
 def read_root():
@@ -79,8 +79,9 @@ async def generate_story(story_request: StoryRequest):
     - **Theme:** {story_request.theme}
     - **Details Provided:** {story_request.txt}
     """
+    prompt_template = ChatPromptTemplate.from_messages([("system", system), ("human", story)])
     
-    final_prompt = prompt_template.format(text=story)
+    # final_prompt = prompt_template.format(text=story)
 
     # Create the LLMChain
     # chain = LLMChain(llm=llm, prompt=prompt_template)
@@ -94,7 +95,7 @@ async def generate_story(story_request: StoryRequest):
     # 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)
+    response = await chain.ainvoke()
     
     if not response:
         raise HTTPException(status_code=500, detail="Failed to generate the story")