Update app.py

app.py

@@ -57,9 +57,26 @@ mode_input = gr.inputs.Dropdown(
 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, 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],
+# Define the chat window
+chat_window = []
+
+def chatbot(chat_window, message, mode, model, hf_model, api_key):
+    if message == "/clear":
+        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))
+    return "\n".join([f"{name}: {text}" for name, text in chat_window])
+
+# Define the Gradio interface for chatbot
+chat_interface = gr.Interface(
+    fn=chatbot,
+    inputs=[message_input, mode_input, model_input, hf_model_input, api_key_input],
     outputs=output,
     title="Chatbot",
     description="Enter your message below to chat with an AI",
@@ -67,11 +84,29 @@ chat_button = gr.Interface(
     allow_flagging=False,
     allow_screenshot=False,
     allow_share=False,
-    examples=[
-        ["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"],
-    ],
 )
 
-chat_button.launch()
+# Add a clear button to the chat window
+clear_button = gr.Interface(
+    fn=lambda: chat_window.clear(),
+    inputs=None,
+    outputs=gr.outputs.Textbox(label="Chat history cleared."),
+    title="Clear Chat History",
+    description="Click to clear the chat history.",
+    theme="compact",
+    allow_flagging=False,
+    allow_screenshot=False,
+    allow_share=False,
+)
+
+# Combine the chat interface and clear button into a single page
+page = gr.Interface(
+    [chat_interface, clear_button],
+    title="Chatbot",
+    description="Enter your message below to chat with an AI",
+    theme="compact",
+    layout="horizontal",
+)
+
+# Launch the page
+page.launch()