Update to gr.Chatbot

app.py

@@ -43,7 +43,6 @@ if torch.cuda.is_available():
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],
-    system_prompt: str = "",
     max_new_tokens: int = 1024,
     temperature: float = 0.6,
     top_p: float = 0.9,
@@ -52,9 +51,7 @@ def generate(
 ) -> Iterator[str]:
     print("chat history: ", chat_history)
     conversation = []
-    if not system_prompt:
-        system_prompt = SYSTEM_PROMPT
-    conversation.append({"role": "system", "content": system_prompt})
+    conversation.append({"role": "system", "content": SYSTEM_PROMPT})
     for user, assistant in chat_history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
     conversation.append({"role": "user", "content": message})
@@ -96,35 +93,44 @@ examples = [
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
-
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Enter your message")
-    clear = gr.Button("Clear")
+    msg = gr.Textbox(label="உங்கள் செய்தியை உள்ளிடவும் / Enter your message")
+    submit_btn = gr.Button("சமர்ப்பிக்கவும் / Submit")
+    clear = gr.Button("தெளிவானது / Clear")
 
     def user(user_message, history):
         return "", history + [[user_message, None]]
 
-    def bot(history, system_prompt, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
+    def bot(history, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
         user_message = history[-1][0]
         chat_history = [(msg[0], msg[1]) for msg in history[:-1]]
         bot_message = ""
-        for response in generate(user_message, chat_history, system_prompt, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
+        for response in generate(user_message, chat_history, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
             bot_message = response
             history[-1][1] = bot_message
             yield history
 
+    gr.Examples(examples=examples, inputs=[msg], label="உதாரணங்கள் / Examples")
+
+    with gr.Accordion("மேம்பட்ட விருப்பங்கள் / Advanced Options", open=False):
+        max_new_tokens = gr.Slider(label="Max new tokens", minimum=1, maximum=MAX_MAX_NEW_TOKENS, step=1, value=DEFAULT_MAX_NEW_TOKENS)
+        temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=4.0, step=0.1, value=0.6)
+        top_p = gr.Slider(label="Top-p (nucleus sampling)", minimum=0.05, maximum=1.0, step=0.05, value=0.9)
+        top_k = gr.Slider(label="Top-k", minimum=1, maximum=1000, step=1, value=50)
+        repetition_penalty = gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2)
+
+    submit_btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot,
+        [chatbot, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
+        chatbot,
+    )
     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         bot,
-        [chatbot, gr.Textbox(label="System prompt", lines=6, value=SYSTEM_PROMPT),
-         gr.Slider(label="Max new tokens", minimum=1, maximum=MAX_MAX_NEW_TOKENS, step=1, value=DEFAULT_MAX_NEW_TOKENS),
-         gr.Slider(label="Temperature", minimum=0.1, maximum=4.0, step=0.1, value=0.6),
-         gr.Slider(label="Top-p (nucleus sampling)", minimum=0.05, maximum=1.0, step=0.05, value=0.9),
-         gr.Slider(label="Top-k", minimum=1, maximum=1000, step=1, value=50),
-         gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2)],
+        [chatbot, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
         chatbot,
     )
     clear.click(lambda: None, None, chatbot, queue=False)
-
+    
     gr.Markdown(LICENSE)
 
 if __name__ == "__main__":