Update app.py

app.py

@@ -51,7 +51,12 @@ def respond(message, max_tokens, temperature, top_p):
     response = ""
     for new_text in streamer:
         response += new_text
-        yield response
+        # wrap model name and response in HTML
+        bubble = (
+            f"<div style='background:#eef2f5;padding:4px;border-radius:4px;font-size:small;max-width:fit-content;margin-bottom:4px;'>"
+            f"Model: {current_model_name}</div>" + response
+        )
+        yield bubble
 
 # User input handler
 def user(message, chat_history):
@@ -64,8 +69,7 @@ def bot(chatbot, max_tokens, temperature, top_p):
     response_generator = respond(message, max_tokens, temperature, top_p)
     for response in response_generator:
         chatbot[-1][1] = response
-        # yield model name and updated chat history
-        yield current_model_name, chatbot
+        yield chatbot
 
 # Model selector handler
 def update_model(model_name):
@@ -96,12 +100,9 @@ with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
 
         # Chat column
         with gr.Column(scale=4):
-            # Mini window for model name
-            model_box = gr.Textbox(label="Model", interactive=False)
-            # Main chat window
             chatbot = gr.Chatbot(
                 bubble_full_width=False,
-                height=400
+                height=500
             )
             msg_input = gr.Textbox(
                 placeholder="Type your message and press Enter...",
@@ -120,12 +121,22 @@ with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
     # Event handlers
     model_selector.change(fn=update_model, inputs=[model_selector], outputs=[])
 
-    msg_input.submit(fn=user, inputs=[msg_input, chatbot], outputs=[msg_input, chatbot], queue=False).then(
+    msg_input.submit(
+        fn=user,
+        inputs=[msg_input, chatbot],
+        outputs=[msg_input, chatbot],
+        queue=False
+    ).then(
         fn=bot,
         inputs=[chatbot, max_tokens, temperature, top_p],
-        outputs=[model_box, chatbot]
+        outputs=[chatbot]
+    )
+    clear_button.click(
+        fn=lambda: [],
+        inputs=None,
+        outputs=chatbot,
+        queue=False
     )
-    clear_button.click(fn=lambda: ("", []), inputs=None, outputs=[model_box, chatbot], queue=False)
 
 if __name__ == "__main__":
     demo.queue(default_concurrency_limit=10, max_size=10).launch(ssr_mode=False, max_threads=10)