Update app.py

app.py

@@ -119,7 +119,7 @@ async def rollback_last(
         search_prompt = None
         for tmp_prompt, uis in internet_search(ppm, serper_api_key, global_context, ctx_num_lconv):
             search_prompt = tmp_prompt
-            yield "", prompt, uis, str(res), gr.update(interactive=False)    
+            yield "", prompt, uis, str(res), gr.update(interactive=False), "off"
     
     async for result in gen_text(
         search_prompt if internet_option else prompt, 
@@ -134,9 +134,9 @@ async def rollback_last(
         }
     ):
         ppm.append_pong(result)
-        yield prompt, ppm.build_uis(), str(res), gr.update(interactive=False)
+        yield prompt, ppm.build_uis(), str(res), gr.update(interactive=False), "off"
         
-    yield prompt, ppm.build_uis(), str(res), gr.update(interactive=True)
+    yield prompt, ppm.build_uis(), str(res), gr.update(interactive=True), "off"
 
 def reset_chat(idx, ld, state):
     res = [state["ppmanager_type"].from_json(json.dumps(ppm_str)) for ppm_str in ld]
@@ -173,7 +173,7 @@ async def chat_stream(
         search_prompt = None
         for tmp_prompt, uis in internet_search(ppm, serper_api_key, global_context, ctx_num_lconv):
             search_prompt = tmp_prompt
-            yield "", prompt, uis, str(res), gr.update(interactive=False)
+            yield "", prompt, uis, str(res), gr.update(interactive=False), "off"
     
     async for result in gen_text(
         search_prompt if internet_option else prompt, 
@@ -188,9 +188,9 @@ async def chat_stream(
         }
     ):
         ppm.append_pong(result)
-        yield "", prompt, ppm.build_uis(), str(res), gr.update(interactive=False)
+        yield "", prompt, ppm.build_uis(), str(res), gr.update(interactive=False), "off"
 
-    yield "", prompt, ppm.build_uis(), str(res), gr.update(interactive=True)
+    yield "", prompt, ppm.build_uis(), str(res), gr.update(interactive=True), "off"
 
 def channel_num(btn_title):
     choice = 0
@@ -270,6 +270,14 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
                         for channel in channels[1:]:
                             channel_btns.append(gr.Button(channel, elem_classes=["custom-btn"]))
 
+                    internet_option = gr.Radio(choices=["on", "off"], value="off", label="internet mode")
+                    serper_api_key = gr.Textbox(
+                        value= os.getenv("SERPER_API_KEY"),
+                        placeholder="Get one by visiting serper.dev", 
+                        label="Serper api key",
+                        visible=False
+                    )
+            
             with gr.Column(scale=8, elem_id="right-pane"):
                 with gr.Column(
                     elem_id="initial-popup", visible=False
@@ -338,16 +346,6 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
                             elem_id="global-context"
                         )
                     
-                    gr.Markdown("#### Internet search")
-                    with gr.Row():
-                        internet_option = gr.Radio(choices=["on", "off"], value="off", label="mode")
-                        serper_api_key = gr.Textbox(
-                            value= os.getenv("SERPER_API_KEY"),
-                            placeholder="Get one by visiting serper.dev", 
-                            label="Serper api key",
-                            visible=False
-                        )
-                    
                     gr.Markdown("#### GenConfig for **response** text generation")
                     with gr.Row():
                         res_temp = gr.Slider(0.0, 2.0, 1.0, step=0.1, label="temp", interactive=True)
@@ -373,7 +371,7 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
         [idx, local_data, instruction_txtbox, chat_state,
          global_context, res_temp, res_topk, res_rpen, res_mnts, res_sample, ctx_num_lconv,
          internet_option, serper_api_key],
-        [instruction_txtbox, context_inspector, chatbot, local_data, regenerate]
+        [instruction_txtbox, context_inspector, chatbot, local_data, regenerate, internet_option]
     ).then(
         None, local_data, None, 
         _js="(v)=>{ setStorage('local_data',v) }"
@@ -405,7 +403,7 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
         [idx, local_data, chat_state,
          global_context, res_temp, res_topk, res_rpen, res_mnts, res_sample, ctx_num_lconv,
          internet_option, serper_api_key],
-        [context_inspector, chatbot, local_data, regenerate]
+        [context_inspector, chatbot, local_data, regenerate, internet_option]
     ).then(
         None, local_data, None, 
         _js="(v)=>{ setStorage('local_data',v) }"