Update app.py

app.py

@@ -52,9 +52,7 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
     print(f'message is - {message}')
     print(f'history is - {history}')
     conversation = []
-    for prompt, answer in history:
-        conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
-    
+   
     if message["files"]:
         image = Image.open(message["files"][-1])
     else:
@@ -63,7 +61,10 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
             image = None
         elif len(history):
             image = Image.open(history[0][0][0])
-    
+            for prompt, answer in history:
+                conversation[0]['content'] = f"<|image_1|>\n{prompt}"
+                conversation.extend([{"role": "assistant", "content": answer}])
+                
     conversation.append({"role": "user", "content": f"<|image_1|>\n{message['text']}"})
     print(f"Conversation is -\n{conversation}")
     inputs = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)