Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import os
 from huggingface_hub import InferenceClient
-import time
 
 hf_token = os.getenv("hf_token")
 
@@ -25,22 +24,20 @@ def get_response(user_input):
     response = ""
     for chunk in stream:
         response += chunk.choices[0].delta.content
-        yield response  # Yielding progressively as the model generates output
-        time.sleep(0.05)  # Optional: Adjust speed of the stream (in seconds)
+    return response
 
 def chat_interface():
     with gr.Blocks() as demo:
-        with gr.Row():  # No 'min_width' argument here
-            with gr.Column():  # No 'min_width' argument here
+        with gr.Row():
+            with gr.Column(scale=1):
                 chat_output = gr.Chatbot(
                     elem_id="chat-box",
                     label="Xylaria 1.4 Senoa Chatbot",
-                    show_label=False,
-                    type="messages"  # Specify type for correct message format
+                    show_label=False
                 )
 
-        with gr.Row(elem_id="input-row"):
-            with gr.Column():  # No 'min_width' argument here
+        with gr.Row(elem_id="input-row", scale=0.2):
+            with gr.Column(scale=0.8):
                 input_textbox = gr.Textbox(
                     label="Type your message", 
                     placeholder="Ask me anything...",
@@ -50,52 +47,32 @@ def chat_interface():
                     elem_id="user-input",
                     show_label=False
                 )
-            with gr.Column():
+            with gr.Column(scale=0.2):
                 send_button = gr.Button("Send", elem_id="send-btn")
 
         def submit_input(user_input, chat_history):
-            # Initialize the chat history if it's empty
-            if not chat_history:
-                chat_history = []
-            chat_history.append({"role": "user", "content": user_input})
-            return "", chat_history  # Clear input field
+            response = get_response(user_input)
+            chat_history.append((user_input, response))
+            return "", chat_history
 
         input_textbox.submit(submit_input, [input_textbox, chat_output], [input_textbox, chat_output])
         send_button.click(submit_input, [input_textbox, chat_output], [input_textbox, chat_output])
 
-        def handle_response(user_input, chat_history):
-            # Initialize the chat history if it's empty
-            if not chat_history:
-                chat_history = []
-            chat_history.append({"role": "user", "content": user_input})
-            
-            response_stream = get_response(user_input)
-            for partial_response in response_stream:
-                chat_history[-1] = {"role": "user", "content": user_input}  # Update the last user message with content
-                chat_history.append({"role": "assistant", "content": partial_response})  # Add assistant's response
-                yield "", chat_history  # Return the updated chat history progressively
-
-        input_textbox.submit(handle_response, [input_textbox, chat_output], [input_textbox, chat_output])
-        send_button.click(handle_response, [input_textbox, chat_output], [input_textbox, chat_output])
-
     demo.css = """
     #input-row {
         position: absolute;
         bottom: 10px;
         width: 100%;
         padding: 10px;
-        background-color: #333;  /* Dark background */
+        background-color: #f5f5f5;
         border-top: 1px solid #ddd;
     }
     #chat-box {
         height: calc(100vh - 100px); /* Adjust the height of chat history */
         overflow-y: scroll;
     }
-    #user-input, #send-btn {
-        background-color: #333;  /* Keep dark gray background */
-    }
     """
-
+    
     return demo
 
 demo = chat_interface()