INNER CODE UNIT · Python

handle_book_content

victordibia/designing-multiagent-systems · course/samples/book_generator/autogen_core/app.py:120

    async def handle_book_content(self, message: BookSections, ctx: MessageContext) -> None:
        try:
            print(">>> BookGeneratorAgent: Generating story book pdf")
            pdf_path = OUTPUT_DIR / "story_book.pdf"
            generate_pdf_report(
                sections=[s.model_dump() for s in message.sections],
                output_file=str(pdf_path),
                report_title=message.sections[0].title
            )
            print(f"Generated story book at: {pdf_path}")
        except Exception as e:
            print(f"Error in book generation: {str(e)}")

async def main(prompt: str) -> None:
    api_key = os.getenv("GEMINI_API_KEY")
    if not api_key:
        raise ValueError("Please set GEMINI_API_KEY environment variable")
    

View source record →