Update app.py

app.py

@@ -45,17 +45,41 @@ additional_inputs = [
     gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05, interactive=True, info="Penalize repeated tokens")
 ]
 
-app = gr.Blocks(theme=gr.themes.Soft())
-with app:
-    chatbot = gr.Chatbot()
-    text_input = gr.Textbox(label="Your message")
-    submit_btn = gr.Button("Submit")
+with gr.Blocks(theme=gr.themes.Soft(), css="""
+    .gradio-container {
+        width: 100%;
+        height: 100%;
+    }
+    .gr-button, .gr-text-input, .gr-chatbot {
+        border-radius: 15px;
+    }
+    .gr-button {
+        width: 48px;
+        height: 48px;
+        background-color: #673ab7;
+        color: #fff;
+        font-size: 24px;
+        display: flex;
+        justify-content: center;
+        align-items: center;
+    }
+""") as app:
+    with gr.Row().style(equal_height=True):
+        text_input = gr.Textbox(label="Your message", placeholder="Type your message...", multiline=False, lines=1, max_lines=1, css_classes="flex-grow")
+        submit_btn = gr.Button(
+            "send",
+            variant="primary",
+            css_classes="material-symbols-outlined rounded-btn"
+        ).style(square=True)
+
+    chatbot = gr.Chatbot().style(height="calc(100vh - 100px)")
 
     def process_message(message, history):
         response = generate(message, history, additional_inputs[0].value)
         history.extend(response)
         return history, ""
 
+    text_input.submit(process_message, inputs=[text_input, chatbot], outputs=[chatbot, text_input])
     submit_btn.click(process_message, inputs=[text_input, chatbot], outputs=[chatbot, text_input])
 
     app.launch(show_api=False)