all output

app.py

@@ -132,16 +132,16 @@ from langgraph.graph import END, StateGraph, START
 # add a chain to the node to analyze the ACF plot?
 workflow = StateGraph(AgentState)
 workflow.add_node("EDA", eda_node)
-# workflow.add_node("PlotAnalysis", plot_node)
-# workflow.add_node("Difference", diff_node)
-# workflow.add_node("ARIMA", arima_node)
+workflow.add_node("PlotAnalysis", plot_node)
+workflow.add_node("Difference", diff_node)
+workflow.add_node("ARIMA", arima_node)
 
 # conditional_edge to refit and the loop refit with resid?
 workflow.add_edge(START, "EDA")
 workflow.add_edge("EDA", END)
-# workflow.add_conditional_edges("PlotAnalysis", router)
-# workflow.add_edge("Difference", "EDA")
-# workflow.add_edge("ARIMA", END)
+workflow.add_conditional_edges("PlotAnalysis", router)
+workflow.add_edge("Difference", "EDA")
+workflow.add_edge("ARIMA", END)
 
 graph = workflow.compile()
 
@@ -149,7 +149,7 @@ graph = workflow.compile()
 def gradio_interface(input_text):
     # input_data = {"messages": [HumanMessage(content="Run the analysis")]}
     resp = graph.invoke({"messages": [HumanMessage(content="Run the analysis")]}) # debug=True
-    return resp['messages'][-1].content
+    return '\n'.join([msg.content for msg in resp['messages'][1:]])
 
 dropdown = gr.Dropdown(
     choices=["Option 1", "Option 2", "Option 3"],