Update app.py

app.py

@@ -28,16 +28,21 @@ def image_to_base64(image_path):
         encoded_string = base64.b64encode(img.read()).decode('utf-8')
     return f"data:image/jpeg;base64,{encoded_string}"
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=60)
 def run_inference(message, history):
     print(message)
     print(history)
 
-    ## to be fixed
+    ## may work
     messages = []
+    images = []
     for couple in history:
-        messages.append(UserMessage(content = [ImageURLChunk(image_url=image_to_base64(file["path"])) for file in couple[0][0]]+[TextChunk(text=couple[0][1])]))
-        messages.append(AssistantMessage(content = couple[1]))
+        if type(couple[0]) is tuple:
+            images += couple[0]
+        elif couple[0][1]:
+            messages.append(UserMessage(content = [ImageURLChunk(image_url=image_to_base64(path)) for path in images]+[TextChunk(text=couple[0][1])]))
+            messages.append(AssistantMessage(content = couple[1]))
+            images = []
     ##
         
     messages.append(UserMessage(content = [ImageURLChunk(image_url=image_to_base64(file["path"])) for file in message["files"]]+[TextChunk(text=message["text"])]))