Update app.py

app.py

@@ -1,47 +1,38 @@
 import gradio as gr
 import requests
+import random
+import time
 
-# Set up API endpoint
+# Set up API URL
 url = "https://75739ca5-2942-4ef4-b25b-705dae6a0946.id.repl.co/connect_to_websocket"
 
-# Define function for bot response
-def get_bot_response(user_id, message):
-    # Send input message to API
-    response = requests.post(url, data={"inputmsg": message, "userid": user_id})
-
-    # Get the bot's response
+# Define function to send message to API and get response
+def get_bot_response(user_input, user_id):
+    response = requests.post(url, data={"inputmsg": user_input, "userid": user_id})
     bot_response = response.text
-
-    # Return the bot's response
     return bot_response
 
-# Set up Gradio chatbot interface
-title = "Chatbot"
-description = "Enter your message and the bot will respond!"
-chatbot = gr.Chatbot(
-    get_bot_response,
-    username="User",
-    placeholder="Enter your message here...",
-    live=True,
-    button_text="Send",
-    label="Chatbot",
-    xss_safe=True,
-    max_messages=10,
-    min_height="400px",
-    allow_speech=True,
-)
-
-# Set up user ID input
-user_id_input = gr.inputs.Textbox(label="User ID", placeholder="Enter your user ID")
-
-# Set up Gradio interface
-interface = gr.Interface(
-    fn=get_bot_response,
-    inputs=[user_id_input, chatbot],
-    outputs="text",
-    title=title,
-    description=description,
-)
-
-# Launch the interface
-interface.launch()
+with gr.Blocks() as imaginechat:
+    gr.HTML('<h1 style="text-align: center; margin-bottom: 1rem"><a href="https://huggingface.co/spaces/ImagineAI-Real/ImagineChat" target="_blank">ImagineChat</a></h1>')
+    gr.HTML('<p>Imagine a Chatbot</p>')
+    chatbot = gr.Chatbot()
+    msg = gr.inputs.Textbox()
+    user_id_input = gr.inputs.Textbox(label="Enter your user ID:")
+    clear_button = gr.Button("Clear")
+    chat_history = []
+
+    def respond(message, user_id):
+        bot_message = get_bot_response(message, user_id)
+        chat_history.append((message, bot_message))
+        time.sleep(1)
+        return "", chat_history
+
+    def submit(message):
+        user_id = user_id_input.value
+        return respond(message, user_id)
+
+    msg.submit(respond, [msg, user_id_input], [msg, chatbot])
+
+    clear_button.click(lambda: chat_history.clear(), [chatbot])
+
+imaginechat.launch()