Update app.py

app.py

@@ -51,8 +51,7 @@ def respond(message, max_tokens, temperature, top_p):
     response = ""
     for new_text in streamer:
         response += new_text
-        # prepend model name on its own line
-        yield f"**Model: {current_model_name}**\n\n{response}"
+        yield response
 
 # User input handler
 def user(message, chat_history):
@@ -65,7 +64,8 @@ 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 chatbot
+        # yield model name and updated chat history
+        yield current_model_name, chatbot
 
 # Model selector handler
 def update_model(model_name):
@@ -82,6 +82,7 @@ with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
         """ )
 
     with gr.Row():
+        # Options column
         with gr.Column(scale=1, min_width=150):
             model_selector = gr.Dropdown(
                 choices=["LeCarnet-3M", "LeCarnet-8M", "LeCarnet-21M"],
@@ -93,10 +94,14 @@ with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
             top_p = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p Sampling")
             clear_button = gr.Button("Clear Chat")
 
+        # 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=500
+                height=400
             )
             msg_input = gr.Textbox(
                 placeholder="Type your message and press Enter...",
@@ -112,11 +117,15 @@ with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
                 label="Example Prompts"
             )
 
+    # 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(
-        fn=bot, inputs=[chatbot, max_tokens, temperature, top_p], outputs=[chatbot]
+        fn=bot,
+        inputs=[chatbot, max_tokens, temperature, top_p],
+        outputs=[model_box, chatbot]
     )
-    clear_button.click(fn=lambda: None, 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)