Update app.py

app.py

@@ -48,7 +48,8 @@ class ChatWrapper:
         return history, history
 
 chat = ChatWrapper()
-            
+state = gr.State()
+agent_state = gr.State()        
 def echo(name, request: gr.Request):
     if request:
         print("Request headers dictionary:", request.headers)
@@ -61,7 +62,7 @@ def my_inference_function(name):
 
 gradio_interface = gr.Interface(
   fn = chat,
-  inputs = "text",
+  inputs = [openai_api_key_textbox, message, state, agent_state],
   outputs = "text"
 )
 gradio_interface.launch()