Update app.py

app.py

@@ -116,9 +116,8 @@ hf_chat_button = gr.Interface(
 )
 
 chat_button = gr.Interface(
-    inputs="text",
-    outputs=["text", "text"],
-    layout="vertical",
+    inputs=[message_input, model_input, hf_model_input, api_key_input],
+    outputs=output,
     title="Chatbot",
     description="Enter your message below to chat with an AI",
     theme="compact",
@@ -126,34 +125,11 @@ chat_button = gr.Interface(
     allow_screenshot=False,
     allow_share=False,
     examples=[
-        ["Hello, how are you?", ""],
-        ["What's the weather like today?", ""],
-        ["Can you help me with some Python code?", "```python\nfor i in range(10):\n    print(i)\n```"],
+        ["Hello, how are you?", "", "", None],
+        ["What's the weather like today?", "", "", None],
+        ["Can you help me with some Python code?", "```python\nfor i in range(10):\n    print(i)\n```", "", None],
     ],
+    live=False,
 )
 
-chat_button.set_config(gr.InterfaceConfig(
-    dependencies=["transformers"],
-    inputs=[gr.InterfaceComponent(model_input, label="OpenAI Model", visible=False)],
-    outputs=[gr.InterfaceComponent(output, label="Bot Response"), gr.InterfaceComponent(api_key_input, label="OpenAI API Key", visible=False)],
-    layout="vertical",
-    title="OpenAI Chatbot (API Key Required)",
-))
-
-chat_button.set_config(gr.InterfaceConfig(
-    dependencies=["transformers"],
-    inputs=[gr.InterfaceComponent(hf_model_input, label="Hugging Face Model", visible=False)],
-    outputs=[gr.InterfaceComponent(output, label="Bot Response")],
-    layout="vertical",
-    title="Hugging Face Chatbot",
-))
-
-chat_button.set_config(gr.InterfaceConfig(
-    dependencies=["transformers"],
-    inputs=[gr.InterfaceComponent(model_input, label="OpenAI Model"), gr.InterfaceComponent(hf_model_input, label="Hugging Face Model", visible=False)],
-    outputs=[gr.InterfaceComponent(output, label="Bot Response"), gr.InterfaceComponent(api_key_input, label="OpenAI API Key")],
-    layout="vertical",
-    title="Chatbot",
-))
-
 chat_button.test_launch()  # Launch the Gradio interface