| from __future__ import annotations |
|
|
| import asyncio |
| import os |
| from pathlib import Path |
|
|
| from ppt_agent.client import PptAgentEnv |
| from ppt_agent.models import PptAgentAction |
|
|
|
|
| def _print_step(label: str, result) -> None: |
| observation = result.observation |
| print(label) |
| print(f" task: {observation.task_name}") |
| print(f" slides: {observation.slide_count}") |
| print(f" reward: {result.reward}") |
| print(f" score: {observation.score}") |
| print(f" done: {result.done}") |
| print(f" error: {observation.last_action_error}") |
| print(f" action_result: {observation.last_action_result}") |
| print(f" termination_reason: {observation.termination_reason}") |
| print() |
|
|
|
|
| def _slide_one_payload() -> dict: |
| return { |
| "background_color": "<bg>", |
| "shapes": [ |
| {"type": "accent_bar", "name": "accent", "color_hex": "<accent>"}, |
| { |
| "type": "text", |
| "name": "title", |
| "text": "Northstar Growth Plan 2026", |
| "x": 0.8, |
| "y": 0.9, |
| "w": 8.0, |
| "h": 0.7, |
| "style": { |
| "font_name": "<font>", |
| "font_size_pt": "<title_size>", |
| "color_hex": "<primary>", |
| "bold": True, |
| }, |
| }, |
| ], |
| } |
|
|
|
|
| def _slide_one_update_payload(title_shape_id: int) -> dict: |
| return { |
| "update_shapes": [ |
| { |
| "shape_id": title_shape_id, |
| "type": "text", |
| "name": "title", |
| "text": "Northstar Growth Plan 2026 Updated", |
| "style": { |
| "font_name": "<font>", |
| "font_size_pt": "<title_size>", |
| "color_hex": "<primary>", |
| "bold": True, |
| }, |
| } |
| ], |
| "add_shapes": [ |
| { |
| "type": "text", |
| "name": "summary", |
| "text": "Professional deck generated through structured actions.", |
| "x": 0.9, |
| "y": 1.8, |
| "w": 7.0, |
| "h": 0.8, |
| "style": { |
| "font_name": "<font>", |
| "font_size_pt": "<body_size>", |
| "color_hex": "<secondary>", |
| }, |
| } |
| ], |
| } |
|
|
|
|
| def _slide_two_payload() -> dict: |
| return { |
| "background_color": "<surface>", |
| "shapes": [ |
| {"type": "accent_bar", "name": "accent", "color_hex": "<accent>"}, |
| { |
| "type": "text", |
| "name": "title", |
| "text": "Results", |
| "x": 0.8, |
| "y": 0.8, |
| "w": 3.0, |
| "h": 0.6, |
| "style": { |
| "font_name": "<font>", |
| "font_size_pt": 24, |
| "color_hex": "<primary>", |
| "bold": True, |
| }, |
| }, |
| { |
| "type": "text", |
| "name": "body", |
| "text": "Retention improved from 88% to 93%.\nOnboarding time reduced by 35%.", |
| "x": 0.9, |
| "y": 1.7, |
| "w": 6.5, |
| "h": 1.8, |
| "style": { |
| "font_name": "<font>", |
| "font_size_pt": "<body_size>", |
| "color_hex": "<secondary>", |
| }, |
| }, |
| ], |
| } |
|
|
|
|
| def _slide_three_payload() -> dict: |
| return { |
| "background_color": "<surface>", |
| "shapes": [ |
| {"type": "accent_bar", "name": "accent", "color_hex": "<accent>"}, |
| { |
| "type": "text", |
| "name": "title", |
| "text": "Revenue Targets", |
| "x": 0.8, |
| "y": 0.8, |
| "w": 4.0, |
| "h": 0.6, |
| "style": { |
| "font_name": "<font>", |
| "font_size_pt": 24, |
| "color_hex": "<primary>", |
| "bold": True, |
| }, |
| }, |
| { |
| "type": "chart", |
| "name": "chart", |
| "chart_type": "column_clustered", |
| "chart_data": { |
| "categories": ["Q1", "Q2", "Q3", "Q4"], |
| "series": [{"name": "Target", "values": [18, 24, 28, 32]}], |
| }, |
| "x": 0.9, |
| "y": 1.6, |
| "w": 6.8, |
| "h": 3.8, |
| "style": { |
| "title": "Quarterly revenue targets", |
| "series_colors": ["<accent>"], |
| }, |
| }, |
| ], |
| } |
|
|
|
|
| async def main() -> None: |
| base_url = os.getenv("ENV_BASE_URL", "http://localhost:7860") |
| env = PptAgentEnv(base_url=base_url, message_timeout_s=180.0) |
| await env.connect() |
|
|
| try: |
| result = await env.reset() |
| print("reset") |
| print(f" task: {result.observation.task_name}") |
| print(f" prompt: {result.observation.prompt_summary}") |
| print() |
|
|
| result = await env.step( |
| PptAgentAction(action_type="create_slide", payload=_slide_one_payload()) |
| ) |
| _print_step("step 1: create slide 1", result) |
|
|
| action_result = result.observation.last_action_result or {} |
| tool_result = action_result.get("tool_result") or {} |
| title_shape_id = tool_result.get("named_shapes", {}).get("title") |
| if title_shape_id is not None: |
| result = await env.step( |
| PptAgentAction( |
| action_type="update_slide", |
| slide_index=1, |
| payload=_slide_one_update_payload(title_shape_id), |
| ) |
| ) |
| _print_step("step 2: update slide 1", result) |
|
|
| result = await env.step( |
| PptAgentAction(action_type="create_slide", payload=_slide_two_payload()) |
| ) |
| _print_step("step 3: create slide 2", result) |
|
|
| result = await env.step( |
| PptAgentAction(action_type="create_slide", payload=_slide_three_payload()) |
| ) |
| _print_step("step 4: create slide 3", result) |
|
|
| save_path = Path("outputs") / f"{result.observation.task_name}_client_test.pptx" |
| if not result.done: |
| result = await env.step( |
| PptAgentAction( |
| action_type="save_presentation", |
| payload={"path": str(save_path)}, |
| ) |
| ) |
| _print_step("step 5: save presentation", result) |
|
|
| if os.getenv("DEMO_DELETE_SLIDE") == "1" and not result.done: |
| result = await env.step( |
| PptAgentAction(action_type="delete_slide", slide_index=1) |
| ) |
| _print_step("step 5: delete slide 1", result) |
| finally: |
| await env.close() |
|
|
|
|
| if __name__ == "__main__": |
| asyncio.run(main()) |
|
|
