pwilczewski commited on
Commit
4e6c0ce
·
1 Parent(s): 8d12897

limiting output

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -149,7 +149,7 @@ graph = workflow.compile()
149
  def gradio_interface(input_text):
150
  # input_data = {"messages": [HumanMessage(content="Run the analysis")]}
151
  resp = graph.invoke({"messages": [HumanMessage(content="Run the analysis")]}) # debug=True
152
- return resp['messages']
153
 
154
  dropdown = gr.Dropdown(
155
  choices=["Option 1", "Option 2", "Option 3"],
 
149
  def gradio_interface(input_text):
150
  # input_data = {"messages": [HumanMessage(content="Run the analysis")]}
151
  resp = graph.invoke({"messages": [HumanMessage(content="Run the analysis")]}) # debug=True
152
+ return resp['messages'][-1]['content']
153
 
154
  dropdown = gr.Dropdown(
155
  choices=["Option 1", "Option 2", "Option 3"],