Update app.py

app.py

@@ -92,11 +92,11 @@ chain2 = LLMChain(llm=llm2, prompt=PromptTemplate(input_variables=["query"], tem
 
 # Define the transition functions
 def transition_s1(query):
-    output = chain1.invoke(query=query)
+    output = chain1.invoke(input=query)
     return agent1(output), s2
 
 def transition_s2(query):
-    output = chain2.invoke(query=query)
+    output = chain2.invoke(input=query)
     return agent2(output), s1
 
 # Define the respond function