Update app.py

app.py

@@ -26,32 +26,16 @@ def chat(input_text, history=[]):
 
 
 with gr.Blocks() as app:
-    gr.Markdown("<h1 style='text-align: center;'>دردشة أطلس</h1>")
+    gr.Markdown("<h1 style='text-align: center;'>دردشة أطلس-شات 27B</h1>")
     chatbot = gr.Chatbot(label="المحادثة")
     state = gr.State([])
 
     with gr.Row():
-        txt = gr.Textbox(show_label=False, placeholder="اكتب رسالتك هنا...").style(container=False)
+        txt = gr.Textbox(show_label=False, placeholder="اكتب رسالتك هنا...")
         send_button = gr.Button("إرسال")
 
-    # Define the button click event
-    def user_input(input_text, history):
-        return "", history + [[input_text, None]]
-
-    def bot_response(history):
-        user_message = history[-1][0]
-        inputs = tokenizer(user_message, return_tensors="pt").to(model.device)
-        outputs = model.generate(**inputs, max_new_tokens=150)
-        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        history[-1][1] = response
-        return history
-
-    # Link functions to button clicks
-    txt.submit(user_input, [txt, state], [txt, state]).then(
-        bot_response, state, chatbot
-    )
-    send_button.click(user_input, [txt, state], [txt, state]).then(
-        bot_response, state, chatbot
-    )
+    # Link the chat function to both text submission and button click
+    txt.submit(chat, [txt, state], [chatbot, state])
+    send_button.click(chat, [txt, state], [chatbot, state])
 
 app.launch()