trying out streaming

app.py

@@ -145,9 +145,26 @@ workflow.add_edge("ARIMA", END)
 
 graph = workflow.compile()
 
-def greet(name):
-    resp = graph.invoke({"messages": [HumanMessage(content="Run the analysis")]}, debug=True)
-    return resp
-
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch(share=True)
+from langgraph_sdk import get_client
+
+# Initialize the LangGraph client
+client = get_client(url="https://huggingface.co/spaces/pwilczewski/gradiobox")
+assistant_id = "graph"
+
+async def stream_response(input_data):
+    thread = await client.threads.create()
+    async for chunk in client.runs.stream(
+        thread["thread_id"],
+        assistant_id,
+        input=input_data,
+        stream_mode="values"
+    ):
+        yield chunk.data  # Yield the data as it is received
+
+def gradio_interface(input_text):
+    # resp = graph.invoke({"messages": [HumanMessage(content="Run the analysis")]}) # debug=True
+    input_data = {"messages": [HumanMessage(content="Run the analysis")]}
+    return stream_response(input_data)
+
+demo = gr.Interface(fn=gradio_interface, inputs="text", outputs="text", live=True)
+demo.launch()