Update app.py

app.py

@@ -142,8 +142,7 @@ def generate(
 
     # Use the global variable to store the chat history
     # global global_chat_history
-
-    conversation = []
+    # conversation = []
 
     # Move the condition here after the assignment
     if input_prompt:
@@ -152,15 +151,17 @@ def generate(
     # Convert input prompt to tensor
     input_ids = tokenizer(input_prompt, return_tensors="pt").to(model.device)
 
-
     for user, assistant in chat_history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
+    
     conversation.append({"role": "user", "content": message})
 
     input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
+    
     if input_ids.shape[1] > MAX_INPUT_TOKEN_LENGTH:
         input_ids = input_ids[:, -MAX_INPUT_TOKEN_LENGTH:]
         gr.Warning(f"Trimmed input from conversation as it was longer than {MAX_INPUT_TOKEN_LENGTH} tokens.")
+    
     input_ids = input_ids.to(model.device)
 
     # Set up the TextIteratorStreamer
@@ -226,11 +227,14 @@ examples=[
 
 with gr.Blocks(css="style.css") as demo:
     ## gradio-chatbot-read-query-param   
-    session_index = 'eb3636167d3a63fbeee32934610e5b2f'
-    #url_params = gr.JSON({}, visible=False, label="URL Params")
-    #def get_session_index(url_params):
-    #    session_index = url_params.get('session_index')
-    #    print(session_index)
+    #session_index = 'eb3636167d3a63fbeee32934610e5b2f'
+    url_params = gr.JSON({}, visible=False, label="URL Params")
+    def get_session_index(url_params):
+        session_index = url_params.get('session_index')
+        print(session_index)
+        return session_index
+        
+    session_index = get_session_index(url_params)
     personalized_data = fetch_personalized_data(session_index)
     print(personalized_data)
     
@@ -270,6 +274,16 @@ with gr.Blocks(css="style.css") as demo:
             input_prompt += f"{user} [/INST] {assistant} <s>[INST] "
         input_prompt += f"{message} [/INST] "
         return input_prompt
+
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+    demo.load(
+        fn=lambda x: x,
+        inputs=[url_params],
+        outputs=[url_params],
+        _js=get_window_url_params(),
+        queue=False
+    )
     
     chat_interface.render()
     #gr.Markdown(LICENSE)
@@ -281,13 +295,4 @@ if __name__ == "__main__":
 
 # Register the function to be called when the program exits
 # atexit.register(save_chat_history)
-    
-
-''' demo.load(
-        fn=lambda x: x,
-        inputs=[url_params],
-        outputs=[url_params],
-        _js=get_window_url_params(),
-        queue=False
-    )
-    '''
+