Update app_dialogue.py

app_dialogue.py

@@ -21,9 +21,9 @@ MODELS = [  # TODO uncomment
 ]
 
 API_PATHS = {  # TODO uncomment + correct endpoint
-    #"HuggingFaceM4/idefics-9b-instruct": "https://api-inference.huggingface.co/models/HuggingFaceM4/idefics-9b-instruct", 
-    "HuggingFaceM4/idefics-9b-instruct": os.getenv("TEMP_DEPLOY_ENDPOINT"),
-    "HuggingFaceM4/idefics-80b-instruct": os.getenv("80B"), 
+    "HuggingFaceM4/idefics-9b-instruct": "https://api-inference.huggingface.co/models/HuggingFaceM4/idefics-9b-instruct", 
+    #"HuggingFaceM4/idefics-9b-instruct": os.getenv("TEMP_DEPLOY_ENDPOINT"),
+    #"HuggingFaceM4/idefics-80b-instruct": os.getenv("80B"), 
 }
 
 SYSTEM_PROMPT = [
@@ -621,17 +621,19 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
                 text_token = text_token.lstrip()
 
             acc_text += text_token
-            #print(f"acc_text - first - {acc_text}")
-            #last_turn = chat_history.pop(-1)
-            #print(f"last_turn is - {last_turn}")
-            #last_turn[-1] += acc_text
-            #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 - first - {acc_text}")
+            last_turn = chat_history.pop(-1)
+            print(f"last_turn is - {last_turn}")
+            last_turn[-1] += acc_text
+            print(f"last_turn[-1] is - {last_turn[-1]}")
+            if last_turn[-1].endswith("\nUser"):
+                last_turn[-1] = last_turn[-1][:-5]
+            chat_history.append(last_turn)
+            #chat_history[-1][-1] = acc_text
+            print(f"yields - chat_history -{chat_history} ")
             #print(f"acc_text -{acc_text} ")
             yield "", None, chat_history
-            #acc_text = ""
+            acc_text = ""
 
     def process_example(message, image):
         clear_msg, image_value, chat = model_inference(