Update presentation_api.py

presentation_api.py

@@ -21,38 +21,22 @@ router = APIRouter(
     tags=["presentation"]
 )
 
+import json
 
-from typing import Dict, List
-from pydantic import BaseModel, Field
-
-class Slide(BaseModel):
-    slide_number: int
-    title: str
-    description: str
-
-# Define the slides container
-class SlidesInput(BaseModel):
-    slides: List[Slide]
-
-@tool(parse_docstring=True)
-def plan(input: SlidesInput) -> str:
-    """Input presentation plan with numbered slides and their descriptions. 
-    Returns a confirmation message indicating that the plan has been created.
-
-    Args:
-        input: PlanInput object containing presentation details
-    """
-    return f"Plan created with {len(input)} slides"
-
-@tool(parse_docstring=True)
-def create_slide(slideno: int, content: str) -> str:
-    """Create a single presentation slide. Returns a confirmation message indicating that the slide has been created.
-
-    Args:
-        slideno: The slide number to create.
-        content: The content for the slide.
-    """
-    return f"slide {slideno} created"
+@tool
+def plan(slides_json: str) -> str:
+    """Create a presentation plan from a JSON string of slides (keys=slide numbers, values=content)."""
+    try:
+        slides = json.loads(slides_json)
+        return f"Plan created with {len(slides)} slides: {', '.join(slides.keys())}."
+    except json.JSONDecodeError as e:
+        return f"Invalid JSON format. Please provide a valid JSON string {str(e)[:100]}."
+
+@tool
+def create_slide(slide_number: int, content: str) -> str:
+    """Create a slide with the given number and content."""
+    # Integration with slide creation API or template would go here
+    return f"Slide {slide_number} created: {content}"
 
 @tool(parse_docstring=True)
 def execute_python(expression: str) -> str: