Update app.py

app.py

@@ -1,6 +1,6 @@
 # app.py
 # =============
-# This is a complete app.py file for deploying the MTSAIR/Cotype-Nano model using Gradio and Hugging Face Transformers with chat and token streaming functionality.
+# This is a complete app.py file for deploying the MTSAIR/Cotype-Nano model using Gradio and Hugging Face Transformers with chat and token streaming functionality, advanced settings, and English interface.
 
 import gradio as gr
 from transformers import pipeline
@@ -10,17 +10,21 @@ model_name = "MTSAIR/Cotype-Nano"
 pipe = pipeline("text-generation", model=model_name, device="cpu")
 
 # Define the system prompt
-system_prompt = {"role": "system", "content": "Ты — ИИ-помощник. Тебе дано задание: необходимо сгенерировать подробный и развернутый ответ."}
+system_prompt = {"role": "system", "content": "You are an AI assistant. Your task is to generate a detailed and comprehensive response."}
 
 # Define the Gradio interface
-def generate_response(history, user_input):
+def generate_response(history, user_input, temperature, max_tokens):
     messages = [system_prompt] + history + [{"role": "user", "content": user_input}]
-    response = pipe(messages, max_length=1024, return_full_text=False)
+    response = pipe(messages, max_length=max_tokens, temperature=temperature, return_full_text=False)
     generated_text = response[0]['generated_text']
     history.append({"role": "user", "content": user_input})
     history.append({"role": "assistant", "content": generated_text})
     return history, ""
 
+# Function to clear chat history
+def clear_chat():
+    return [], ""
+
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("## Cotype-Nano Text Generation Chat")
@@ -30,10 +34,21 @@ with gr.Blocks() as demo:
     with gr.Row():
         txt = gr.Textbox(
             show_label=False,
-            placeholder="Введите ваш запрос здесь...",
+            placeholder="Type your message here...",
         )
 
-    txt.submit(generate_response, [chatbot, txt], [chatbot, txt])
+        send_btn = gr.Button("Send")
+
+    with gr.Row():
+        clear_btn = gr.Button("Clear Chat")
+
+    with gr.Row():
+        temperature_slider = gr.Slider(0, 1, 0.7, step=0.1, label="Temperature")
+        max_tokens_slider = gr.Slider(1, 1000, 100, step=1, label="Max Tokens")
+
+    send_btn.click(generate_response, [chatbot, txt, temperature_slider, max_tokens_slider], [chatbot, txt])
+    txt.submit(generate_response, [chatbot, txt, temperature_slider, max_tokens_slider], [chatbot, txt])
+    clear_btn.click(clear_chat, outputs=[chatbot, txt])
 
 # Launch the interface
 if __name__ == "__main__":