Update app_dialogue.py

app_dialogue.py

@@ -293,17 +293,23 @@ def format_user_prompt_with_im_history_and_system_conditioning(
     resulting_list = copy.deepcopy(SYSTEM_PROMPT)
 
     # Format history
-    for turn in history:
-        user_utterance, assistant_utterance = turn
-        splitted_user_utterance = split_str_on_im_markdown(user_utterance)
+    if history:
+        for turn in history:
+            user_utterance, assistant_utterance = turn
+            splitted_user_utterance = split_str_on_im_markdown(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)
+            resulting_list.append(f"<end_of_utterance>\nAssistant: {assistant_utterance}")
+    else:
+        optional_space = " "
 
-        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)
-        resulting_list.append(f"<end_of_utterance>\nAssistant: {assistant_utterance}")
-
+        #resulting_list.extend(splitted_user_utterance)
+        resulting_list.append(f"<end_of_utterance>\nAssistant: {assistant_utterance}")        
     # Format current input
     current_user_prompt_str = remove_spaces_around_token(current_user_prompt_str)
     if current_image is None:
@@ -554,7 +560,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
         #stream = client.generate_stream(prompt=query, **generation_args)
         #resp = client.text_generation(query, **generation_args, stream=False,details=True, return_full_text=False)
         stream = client.text_generation(query, **generation_args, stream=False)
-        
+        print (stream)
         #stream = ""
         #for response in resp:
         #    stream += response.token.text