[LlamaDemo] Add a button to toggle thinking mode (#10667)

Author: kirklandsign

examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java

@@ -42,6 +42,7 @@
 import androidx.constraintlayout.widget.ConstraintLayout;
 import androidx.core.app.ActivityCompat;
 import androidx.core.content.ContextCompat;
+import androidx.core.content.res.ResourcesCompat;
 import com.google.gson.Gson;
 import com.google.gson.reflect.TypeToken;
 import java.lang.reflect.Type;
@@ -56,6 +57,7 @@
 
 public class MainActivity extends AppCompatActivity implements Runnable, LlmCallback {
   private EditText mEditTextMessage;
+  private ImageButton mThinkModeButton;
   private ImageButton mSendButton;
   private ImageButton mGalleryButton;
   private ImageButton mCameraButton;
@@ -77,6 +79,7 @@ public class MainActivity extends AppCompatActivity implements Runnable, LlmCall
   private SettingsFields mCurrentSettingsFields;
   private Handler mMemoryUpdateHandler;
   private Runnable memoryUpdater;
+  private boolean mThinkMode = false;
   private int promptID = 0;
   private long startPos = 0;
   private static final int CONVERSATION_HISTORY_MESSAGE_LOOKBACK = 2;
@@ -252,6 +255,7 @@ protected void onCreate(Bundle savedInstanceState) {
       finish();
     }
 
+    mThinkModeButton = requireViewById(R.id.thinkModeButton);
     mEditTextMessage = requireViewById(R.id.editTextMessage);
     mSendButton = requireViewById(R.id.sendButton);
     mSendButton.setEnabled(false);
@@ -271,6 +275,28 @@ protected void onCreate(Bundle savedInstanceState) {
           MainActivity.this.startActivity(myIntent);
         });
 
+    mThinkModeButton.setOnClickListener(
+        view -> {
+          if (mThinkMode) {
+            mThinkMode = false;
+            mThinkModeButton.setImageDrawable(
+                ResourcesCompat.getDrawable(
+                    getResources(), R.drawable.baseline_lightbulb_24, null));
+          } else {
+            mThinkMode = true;
+            mThinkModeButton.setImageDrawable(
+                ResourcesCompat.getDrawable(getResources(), R.drawable.blue_lightbulb_24, null));
+          }
+          runOnUiThread(
+              () -> {
+                String thinkingModeText = mThinkMode ? "on" : "off";
+                mMessageAdapter.add(
+                    new Message(
+                        "Thinking mode is " + thinkingModeText, false, MessageType.SYSTEM, 0));
+                mMessageAdapter.notifyDataSetChanged();
+              });
+        });
+
     mCurrentSettingsFields = new SettingsFields();
     mMemoryUpdateHandler = new Handler(Looper.getMainLooper());
     onModelRunStopped();

examples/demo-apps/android/LlamaDemo/app/src/main/res/drawable/baseline_lightbulb_24.xml

@@ -0,0 +1,5 @@
+<vector xmlns:android="http://schemas.android.com/apk/res/android" android:height="24dp" android:tint="#FFFFFF" android:viewportHeight="24" android:viewportWidth="24" android:width="24dp">
+      
+    <path android:fillColor="@android:color/white" android:pathData="M9,21c0,0.5 0.4,1 1,1h4c0.6,0 1,-0.5 1,-1v-1L9,20v1zM12,2C8.1,2 5,5.1 5,9c0,2.4 1.2,4.5 3,5.7L8,17c0,0.5 0.4,1 1,1h6c0.6,0 1,-0.5 1,-1v-2.3c1.8,-1.3 3,-3.4 3,-5.7 0,-3.9 -3.1,-7 -7,-7z"/>
+    
+</vector>

examples/demo-apps/android/LlamaDemo/app/src/main/res/drawable/blue_lightbulb_24.xml

@@ -0,0 +1,5 @@
+<vector xmlns:android="http://schemas.android.com/apk/res/android" android:height="24dp" android:tint="#6684EC" android:viewportHeight="24" android:viewportWidth="24" android:width="24dp">
+      
+    <path android:fillColor="@android:color/white" android:pathData="M9,21c0,0.5 0.4,1 1,1h4c0.6,0 1,-0.5 1,-1v-1L9,20v1zM12,2C8.1,2 5,5.1 5,9c0,2.4 1.2,4.5 3,5.7L8,17c0,0.5 0.4,1 1,1h6c0.6,0 1,-0.5 1,-1v-2.3c1.8,-1.3 3,-3.4 3,-5.7 0,-3.9 -3.1,-7 -7,-7z"/>
+    
+</vector>

examples/demo-apps/android/LlamaDemo/app/src/main/res/layout/activity_main.xml

@@ -174,6 +174,14 @@
                 android:padding="10dp"
                 android:src="@drawable/baseline_add_24" />
 
+            <ImageButton
+                android:id="@+id/thinkModeButton"
+                android:layout_width="wrap_content"
+                android:layout_height="wrap_content"
+                android:background="@android:color/transparent"
+                android:padding="10dp"
+                android:src="@drawable/baseline_lightbulb_24" />
+
             <EditText
                 android:id="@+id/editTextMessage"
                 android:layout_width="match_parent"