From aee70ff4cb8d43afabf790fc946ed6d76cbef5ac Mon Sep 17 00:00:00 2001
From: Chester Hu <chengpenghu@meta.com>
Date: Tue, 15 Oct 2024 11:55:00 -0700
Subject: [PATCH] Adding keyboard dismissal when prompt sent

Summary: Dismiss the keyboard when sending the message

Reviewed By: cmodi-meta

Differential Revision: D64413153
---
 .../java/com/example/executorchllamademo/MainActivity.java | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java b/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
index 8bd9d8cb4f1..e9560be2871 100644
--- a/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
+++ b/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
@@ -26,6 +26,7 @@
 import android.system.Os;
 import android.util.Log;
 import android.view.View;
+import android.view.inputmethod.InputMethodManager;
 import android.widget.EditText;
 import android.widget.ImageButton;
 import android.widget.ImageView;
@@ -662,6 +663,12 @@ private void onModelRunStopped() {
     mSendButton.setImageResource(R.drawable.baseline_send_24);
     mSendButton.setOnClickListener(
         view -> {
+          try {
+            InputMethodManager imm = (InputMethodManager) getSystemService(INPUT_METHOD_SERVICE);
+            imm.hideSoftInputFromWindow(getCurrentFocus().getWindowToken(), 0);
+          } catch (Exception e) {
+            ETLogging.getInstance().log("Keyboard dismissal error: " + e.getMessage());
+          }
           addSelectedImagesToChatThread(mSelectedImageUri);
           String finalPrompt;
           String rawPrompt = mEditTextMessage.getText().toString();