Update app.py

app.py

@@ -30,4 +30,23 @@ if button:
     })
 
     # Print the chatbot's response
-    st.write(output)
+    st.write(output)
+
+    # Create a text input for the chatbot's response
+    chatbot_response = st.text_input("Chatbot's response:")
+
+    # When the user enters a response, call the query function again with the user's response as the payload
+    if chatbot_response:
+        output = query({
+            "inputs": {
+                "source_sentence": chatbot_response,
+                "sentences": [
+                    "That is a happy dog",
+                    "That is a very happy person",
+                    "Today is a sunny day"
+                ]
+            },
+        })
+
+        # Print the chatbot's response
+        st.write(output)