Update app.py

app.py

@@ -183,6 +183,17 @@ css = """
 }
 """
 
+# Function to get just the model name from the dropdown selection
+def get_model_name(full_selection):
+    return full_selection.split(" - ")[0]
+
+# Function to handle message submission
+def submit_message(msg, history, prompt, model, tok, temp, k, p, rp):
+    return chat_response(
+        msg, history, prompt,
+        get_model_name(model), tok, temp, k, p, rp
+    ), ""
+
 # ------------------------------
 # Gradio UI
 # ------------------------------
@@ -235,45 +246,26 @@ with gr.Blocks(title="Qwen3 Chat", css=css) as demo:
         <p>Qwen3 models developed by Alibaba Cloud. Interface powered by Gradio and ZeroGPU.</p>
     </div>
     """)
-            
-    # Extract just the model name from the dropdown selection
-    def get_model_name(full_selection):
-        return full_selection.split(" - ")[0]
-            
+    
+    # Event handlers
     clr.click(fn=lambda: ([], ""), outputs=[chat, txt])
     cnl.click(fn=cancel_generation)
     
-    # Clear the text input after submitting the message    
-    def submit_message(msg, history, prompt, model, tok, temp, k, p, rp):
-        return chat_response(
-            msg, history, prompt,
-            get_model_name(model), tok, temp, k, p, rp
-        ), ""
-        
+    # Handle submission from Enter key
     txt.submit(
         fn=submit_message,
-        inputs=[txt, chat, sys_prompt,
-                model_dd, max_tok, temp, k, p, rp],
+        inputs=[txt, chat, sys_prompt, model_dd, max_tok, temp, k, p, rp],
         outputs=[chat, txt],
         show_progress=True
     )
     
-    # Clear the text input after sending the message
-    def send_message(msg, history, prompt, model, tok, temp, k, p, rp):
-        return chat_response(
-            msg, history, prompt,
-            get_model_name(model), tok, temp, k, p, rp
-        ), ""
-    
+    # Handle submission from Send button
     send_btn.click(
-        fn=send_message,
-        inputs=[txt, chat, sys_prompt,
-                model_dd, max_tok, temp, k, p, rp],
+        fn=submit_message,
+        inputs=[txt, chat, sys_prompt, model_dd, max_tok, temp, k, p, rp],
         outputs=[chat, txt],
         show_progress=True
     )
-        show_progress=True
-    )
 
 if __name__ == "__main__":
     demo.launch()