Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import openai
 import requests
 import csv
+import uuid
 
 
 prompt_templates = {"Default ChatGPT": ""}
@@ -72,7 +73,9 @@ def submit_message(type_select,user_token, prompt, prompt_template, temperature,
     
     try:
         if type_select=='TEXT':
-            completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=system_prompt + history[-context_length*2:] + [prompt_msg], temperature=temperature, max_tokens=max_tokens)
+            text_history = [x for x in history if x['role'] != 'image' ]
+            print(text_history)
+            completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=system_prompt + text_history[-context_length*2:] + [prompt_msg], temperature=temperature, max_tokens=max_tokens)
             print(prompt_msg,completion.choices[0].message.to_dict())
             history.append(prompt_msg)
             history.append(completion.choices[0].message.to_dict())
@@ -107,7 +110,15 @@ def submit_message(type_select,user_token, prompt, prompt_template, temperature,
     for i in range(0, len(history)-1, 2):
         print(history[i])
         if(history[i]['role'] == 'image'):
-            chat_messages.append((history[i]['content'], (history[i+1]['content'],)))
+            picture_name = str(uuid.uuid1())+'.png'
+            reponse = requests.get(history[i+1]['content'])
+            with open('/home/user/app/'+picture_name,'wb') as f:
+                f.write(reponse.content)
+            
+            image_his = {'name': '/home/user/app/'+picture_name, 'mime_type': 'image/png', 'alt_text': None, 'data': None, 'is_file': True}
+            
+            chat_messages.append((history[i]['content'],image_his))
+
         else:
             chat_messages.append((history[i]['content'], history[i+1]['content']))
     print(2,chat_messages)
@@ -175,5 +186,5 @@ with gr.Blocks(css=css) as demo:
 
 demo.queue(concurrency_count=10)
 demo.launch(
-    auth=("admin", "IBTGeE3NrPsrViDI"),
+    # auth=("admin", "IBTGeE3NrPsrViDI"),
     height='800px')