Update app.py

app.py

@@ -60,10 +60,23 @@ def bot_streaming(message, history, max_new_tokens=250):
 
     
     
-    for url in files:
-        response = requests.get(url)
-        img = Image.open(BytesIO(response.content)).convert("RGB")
-        images.append(img)
+    for x in messages:
+        try:
+            if x['content'][1]['type']=='image':
+                url = x['content'][1]['url']
+                response = requests.get(url)
+                img = Image.open(BytesIO(response.content)).convert("RGB")
+                images.append(img)
+        except :
+            try:
+                if x['content'][0]['type']=='image':
+                url = x['content'][1]['url']
+                response = requests.get(url)
+                img = Image.open(BytesIO(response.content)).convert("RGB")
+                images.append(img)
+
+            except:
+                pass
     
 
     print("\n\nfinal messages        ", messages)