Update app.py

app.py

@@ -87,7 +87,7 @@ 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": "assistant", "content": image_url })
+            history.append({ "role": "image", "content": image_url })
 
             state['total_tokens'] += 0
             
@@ -99,8 +99,16 @@ def submit_message(type_select,user_token, prompt, prompt_template, temperature,
         })
 
     total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
-    chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]
 
+    chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]
+    print(1,chat_messages)
+    chat_messages=[]
+    for i in range(0, len(history)-1, 2):
+        if(history[i]['role']=='image'):
+            chat_messages.append(((history[i]['content'],), None))
+        else:
+            chat_messages.append(history[i]['content'])
+    print(2,chat_messages)
     return '', chat_messages, total_tokens_used_msg, state
 
 def clear_conversation():