Update app.py

app.py

@@ -86,7 +86,8 @@ def submit_message(type_select,user_token, prompt, prompt_template, temperature,
                     )
             print("image result ",response)
             image_url = response['data'][0]['url']
-            history.append(prompt_msg)
+
+            history.append({ "role": "image", "content": prompt })
             history.append({ "role": "image", "content": image_url })
 
             state['total_tokens'] += 0
@@ -104,8 +105,8 @@ def submit_message(type_select,user_token, prompt, prompt_template, temperature,
     print(1,chat_messages)
     chat_messages=[]
     for i in range(0, len(history)-1, 2):
-        print(history[i],history[i]['role'])
-        if(history[i]['role']=='image'):
+        print(history[i])
+        if(history[i]['role'] == 'image'):
             chat_messages.append((history[i]['content'], (history[i+1]['content'],)))
         else:
             chat_messages.append((history[i]['content'], history[i+1]['content']))