Update app_dialogue.py

app_dialogue.py

@@ -314,16 +314,21 @@ def format_user_prompt_with_im_history_and_system_conditioning(
 
     # Format history
     for turn in history:
+        print(f"inside for loop: turn is - {turn}")
         user_utterance, assistant_utterance = turn
+        print(f"inside for loop: user_utterance is - {user_utterance}, assistant_utterance is - {assistant_utterance}")
         assistant_utterance = assistant_utterance.replace(ASSISTANT_PREPEND, "")
         splitted_user_utterance = split_str_on_im_markdown(user_utterance)
+        print(f"inside for loop: splitted_user_utterance after a call to fun split_str_on_im_markdown()  is - {splitted_user_utterance}")
 
         optional_space = ""
         if not is_image(splitted_user_utterance[0]):
             optional_space = " "
         resulting_list.append(f"\nUser:{optional_space}")
         resulting_list.extend(splitted_user_utterance)
+        print(f"inside for loop: resulting_list after extending with 'splitted_user_utterance' is - {resulting_list}")
         resulting_list.append(f"<end_of_utterance>\nAssistant: {assistant_utterance}")
+        print(f"inside for loop: resulting_list after appending is - {resulting_list}")
 
     # Format current input
     current_user_prompt_str = remove_spaces_around_token(current_user_prompt_str)
@@ -343,6 +348,7 @@ def format_user_prompt_with_im_history_and_system_conditioning(
     else:
         # Choosing to put the image first when the image is inputted through the UI, but this is an arbiratrary choice.
         resulting_list.extend(["\nUser:", current_image, f"{current_user_prompt_str}<end_of_utterance>\nAssistant:"])
+        print(f"inside else: current_image is not None; resulting_list after extending is - {resulting_list}")
         current_user_prompt_list = [current_user_prompt_str]
 
     print(f"returns - resulting_list - {resulting_list}, current_user_prompt_list - {current_user_prompt_list}")
@@ -536,6 +542,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
         print(f"params: max_new_tokens - {max_new_tokens}")
         print(f"params: repetition_penalty - {repetition_penalty}")
         print(f"params: top_p - {top_p}")
+        
         if user_prompt_str.strip() == "" and image is None:
             return "", None, chat_history
 
@@ -544,6 +551,8 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
             current_image=image,
             history=chat_history,
         )
+        print(f"formated_prompt_list is - {formated_prompt_list}")
+        print(f"user_prompt_list is - {user_prompt_list}")
 
         client_endpoint = API_PATHS[model_selector]
         client = Client(
@@ -573,6 +582,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
         if image is None:
             # Case where there is no image OR the image is passed as `<fake_token_around_image><image:IMAGE_URL><fake_token_around_image>`
             chat_history.append([prompt_list_to_markdown(user_prompt_list), ASSISTANT_PREPEND])
+            print(f"image is NONE; Chat_history after append is - {chat_history}")
         else:
             # Case where the image is passed through the Image Box.
             # Convert the image into base64 for both passing it through the chat history and
@@ -583,15 +593,18 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
                     ASSISTANT_PREPEND,
                 ]
             )
+            print(f"image is NOT NONE; Chat_history after append is - {chat_history}")
 
         query = prompt_list_to_tgi_input(formated_prompt_list)
+        print(f"query is - {query}")
         stream = client.generate_stream(prompt=query, **generation_args)
 
         print(f"chat_history just before for loop - {chat_history}")
         acc_text = ""
         for idx, response in enumerate(stream):
             text_token = response.token.text
-
+            print(f"text_token is - {text_token}")
+            
             if response.details:
                 # That's the exit condition
                 return
@@ -611,8 +624,8 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
             #print(f"last_turnp[-1] is - {last_turn[-1]}")
             #chat_history.append(last_turn)
             chat_history[-1][-1] = acc_text
-            print(f"yields - chat_history -{chat_history} ")
-            print(f"acc_text -{acc_text} ")
+            #print(f"yields - chat_history -{chat_history} ")
+            #print(f"acc_text -{acc_text} ")
             yield "", None, chat_history
             #acc_text = ""