Update app.py

app.py

@@ -38,42 +38,48 @@ def hf_chat(model_name, message):
     return bot_response
 
 # Define the Gradio interface for chatbot
-api_key_input = gr.inputs.Textbox(label="OpenAI API Key", default=None)
+api_key_input = gr.inputs.Textbox(label="OpenAI API Key", default=None, block="sidebar")
 model_input = gr.inputs.Dropdown(
     label="Select OpenAI model",
     choices=["davinci", "curie", "babbage"],
     default="davinci",
+    block="sidebar"
 )
 hf_model_input = gr.inputs.Dropdown(
     label="Select Hugging Face model",
     choices=["microsoft/DialoGPT-large", "Salesforce/codegen-2B-multi", "microsoft/DialoGPT-small"],
     default="microsoft/DialoGPT-large",
+    block="sidebar"
 )
 mode_input = gr.inputs.Dropdown(
     label="Select chatbot mode",
     choices=["OpenAI", "Hugging Face"],
     default="OpenAI",
+    block="sidebar"
 )
-message_input = gr.inputs.Textbox(label="Enter your message here")
-output = gr.outputs.Textbox(label="Bot response")
+message_input = gr.inputs.Textbox(label="Enter your message here", block="input")
+output = gr.outputs.Textbox(label="Bot response", block="output")
 
 # Define the chat window
 chat_window = []
 
-def chatbot(chat_window, message, mode, model, hf_model, api_key):
-    if message == "/clear":
+def chatbot(chat_window, message, mode, model, hf_model, api_key, send_button, clear_button):
+    if clear_button:
         chat_window.clear()
         return "Chat history cleared."
-    if message:
-        if mode == "Hugging Face":
-            bot_response = hf_chat(hf_model, message)
-        else:
-            bot_response = openai_chat(api_key, model, message)
-        chat_window.append(("User", message))
-        chat_window.append(("Bot", bot_response))
+    if send_button:
+        if message:
+            if mode == "Hugging Face":
+                bot_response = hf_chat(hf_model, message)
+            else:
+                bot_response = openai_chat(api_key, model, message)
+            chat_window.append(("User", message))
+            chat_window.append(("Bot", bot_response))
     return "\n".join([f"{name}: {text}" for name, text in chat_window])
 
 # Define the Gradio interface for chatbot
+send_button = gr.inputs.Button(label="Send")
+clear_button = gr.inputs.Button(label="Clear Chat History")
 chat_interface = gr.Interface(
     fn=chatbot,
     inputs=[
@@ -82,8 +88,8 @@ chat_interface = gr.Interface(
         model_input,
         hf_model_input,
         api_key_input,
-        gr.inputs.Button(label="Send"),
-        gr.inputs.Button(label="Clear Chat History")
+        send_button,
+        clear_button
     ],
     outputs=output,
     title="Chatbot",
@@ -92,6 +98,7 @@ chat_interface = gr.Interface(
     allow_flagging=False,
     allow_screenshot=False,
     allow_share=False,
+    layout="vertical"
 )
 
 # Launch the page