updated app.py

app.py

@@ -19,6 +19,7 @@ class StoryRequest(BaseModel):
     mood: str
     story_type: str
     theme: str
+    length: int
     num_scenes: int
     txt: str
 
@@ -64,11 +65,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}")])
-
-@app.get("/")
-def read_root():
-    return {"message": "Welcome to the Story Generation API!"}
+prompt_template = ChatPromptTemplate.from_messages([("system", system), ("human", "{text}")])
 
 # FastAPI endpoint to generate the story
 @app.post("/generate_story/")
@@ -79,9 +76,8 @@ 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)
@@ -92,10 +88,7 @@ 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)
-    # async for s in chain.astream(final_prompt):
-    #     print(s.content, end="", flush=True)
-    response = await chain.ainvoke()
+    response = chain.invoke(final_prompt)
     
     if not response:
         raise HTTPException(status_code=500, detail="Failed to generate the story")
@@ -125,4 +118,14 @@ async def generate_story(story_request: StoryRequest):
     return {
         "story": response,
         "images": images
-    }
+    }
+    
+
+
+# image_prompt = (
+#         f"Generate an image for Scene {i+1}. "
+#         f"This image should represent the details described in paragraph {i+1} of the story. "
+#         f"Mood: {mood}, Story Type: {', '.join(story_type)}, Theme: {theme}. "
+#         f"Story: {response} "
+#         f"Focus on the key elements in paragraph {i+1}."
+#     )