Update app.py

app.py

@@ -94,8 +94,8 @@ def fill_up_placeholders(txt):
 
 async def rollback_last(
     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, palm_if
+    global_context, res_temp, res_topk, res_toppo, ctx_num_lconv
+    internet_option, serper_api_key
 ):
     internet_option = True if internet_option == "on" else False
     
@@ -152,8 +152,8 @@ def reset_chat(idx, ld, state, palm_if):
 
 async def chat_stream(
     idx, local_data, instruction_txtbox, chat_state,
-    global_context, res_temp, res_topk, res_topp, 
-    internet_option, serper_api_key, palm_if
+    global_context, res_temp, res_topk, res_topp, ctx_num_lconv
+    internet_option, serper_api_key, 
 ):
     internet_option = True if internet_option == "on" else False
     
@@ -349,7 +349,12 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
                         res_temp = gr.Slider(0.0, 2.0, 1.0, step=0.1, label="temp", interactive=True)
                         res_topk = gr.Slider(20, 1000, 50, step=1, label="top_k", interactive=True)
                         res_topp = gr.Slider(0.0, 1.0, 0.9, step=0.1, label="top_p", interactive=True)
-                
+
+                with gr.Column():
+                    gr.Markdown("#### Context managements")
+                    with gr.Row():
+                        ctx_num_lconv = gr.Slider(2, 10, 3, step=1, label="number of recent talks to keep", interactive=True)        
+        
         gr.Markdown(
             "***NOTE:*** If you are subscribing [PRO](https://huggingface.co/pricing#pro), you can simply duplicate this space and use your "
             "Hugging Face Access Token to run the same application. Just add `HF_TOKEN` secret with the Token value accorindg to [this guide]"
@@ -373,8 +378,8 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
     ).then(
         chat_stream,
         [idx, local_data, instruction_txtbox, chat_state,
-         global_context, res_temp, res_topk, res_topp,
-         internet_option, serper_api_key, palm_if],
+         global_context, res_temp, res_topk, res_topp, ctx_num_lconv
+         internet_option, serper_api_key],
         [instruction_txtbox, context_inspector, chatbot, local_data, regenerate, internet_option]
     ).then(
         None, local_data, None, 
@@ -405,8 +410,8 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
     regen_event = regenerate.click(
         rollback_last,
         [idx, local_data, chat_state,
-         global_context, res_temp, res_topk, res_topp,
-         internet_option, serper_api_key, palm_if],
+         global_context, res_temp, res_topk, res_topp, ctx_num_lconv
+         internet_option, serper_api_key],
         [context_inspector, chatbot, local_data, regenerate, internet_option]
     ).then(
         None, local_data, None,