Update app.py

app.py

@@ -37,52 +37,29 @@ def hf_chat(model_name, message):
     
     return bot_response
 
-# Define the Gradio interface for OpenAI chatbot
+# Define the Gradio interface for chatbot
 api_key_input = gr.inputs.Textbox(label="OpenAI API Key", default=None)
 model_input = gr.inputs.Dropdown(
     label="Select OpenAI model",
     choices=["davinci", "curie", "babbage"],
     default="davinci",
 )
-message_input = gr.inputs.Textbox(label="Enter your message here")
-output = gr.outputs.Textbox(label="Bot response")
-
-openai_chat_button = gr.Interface(
-    fn=openai_chat,
-    inputs=[api_key_input, model_input, message_input],
-    outputs=output,
-    title="OpenAI Chatbot",
-    description="Enter your message below to chat with an OpenAI AI",
-    theme="compact",
-    layout="vertical",
-    allow_flagging=False,
-    allow_screenshot=False,
-    allow_share=False,
-)
-
-# Define the Gradio interface for Hugging Face chatbot
 hf_model_input = gr.inputs.Dropdown(
     label="Select Hugging Face model",
     choices=["microsoft/DialoGPT-large", "microsoft/DialoGPT-medium", "microsoft/DialoGPT-small"],
     default="microsoft/DialoGPT-large",
 )
-
-hf_chat_button = gr.Interface(
-    fn=hf_chat,
-    inputs=[hf_model_input, message_input],
-    outputs=output,
-    title="Hugging Face Chatbot",
-    description="Enter your message below to chat with a Hugging Face AI",
-    theme="compact",
-    layout="vertical",
-    allow_flagging=False,
-    allow_screenshot=False,
-    allow_share=False,
+mode_input = gr.inputs.Dropdown(
+    label="Select chatbot mode",
+    choices=["OpenAI", "Hugging Face"],
+    default="OpenAI",
 )
+message_input = gr.inputs.Textbox(label="Enter your message here")
+output = gr.outputs.Textbox(label="Bot response")
 
 chat_button = gr.Interface(
-    fn=lambda message, model, hf_model, api_key: hf_chat(hf_model, message) if hf_model else openai_chat(api_key, model, message),
-    inputs=[message_input, model_input, hf_model_input, api_key_input],
+    fn=lambda message, model, hf_model, api_key, mode: hf_chat(hf_model, message) if mode == "Hugging Face" else openai_chat(api_key, model, message),
+    inputs=[message_input, model_input, hf_model_input, api_key_input, mode_input],
     outputs=output,
     title="Chatbot",
     description="Enter your message below to chat with an AI",
@@ -91,11 +68,10 @@ chat_button = gr.Interface(
     allow_screenshot=False,
     allow_share=False,
     examples=[
-        ["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],
+        ["Hello, how are you?", "", "", None, "OpenAI"],
+        ["What's the weather like today?", "", "", None, "Hugging Face"],
+        ["Can you help me with some Python code?", "```python\nfor i in range(10):\n    print(i)\n```", "", None, "OpenAI"],
     ],
-    live=False,
 )
 
-chat_button.launch() # Launch the Gradio interface
+gradio.launch(chat_button)