mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-25 00:36:31 +02:00
31 lines
924 B
Python
31 lines
924 B
Python
|
from langgraph.graph import StateGraph
|
||
|
|
|
||
|
|
from .configuration import Configuration
|
||
|
|
from .nodes import (
|
||
|
|
create_presentation_slides,
|
||
|
|
create_slide_audio,
|
||
|
|
generate_slide_scene_codes,
|
||
|
|
)
|
||
|
|
from .state import State
|
||
|
|
|
||
|
|
|
||
|
|
def build_graph():
|
||
|
|
workflow = StateGraph(State, config_schema=Configuration)
|
||
|
|
|
||
|
|
workflow.add_node("create_presentation_slides", create_presentation_slides)
|
||
|
|
workflow.add_node("create_slide_audio", create_slide_audio)
|
||
|
|
workflow.add_node("generate_slide_scene_codes", generate_slide_scene_codes)
|
||
|
|
|
||
|
|
workflow.add_edge("__start__", "create_presentation_slides")
|
||
|
|
workflow.add_edge("create_presentation_slides", "create_slide_audio")
|
||
|
|
workflow.add_edge("create_slide_audio", "generate_slide_scene_codes")
|
||
|
|
workflow.add_edge("generate_slide_scene_codes", "__end__")
|
||
|
|
|
||
|
|
graph = workflow.compile()
|
||
|
|
graph.name = "Surfsense Video Presentation"
|
||
|
|
|
||
|
|
return graph
|
||
|
|
|
||
|
|
|
||
|
|
graph = build_graph()
|