Update app.py

app.py

@@ -63,24 +63,34 @@ def bot_streaming(message, history, max_new_tokens=250):
     
     
     for x in messages:
+        print("x ,    ", x)
         try:
-            if x['content'][1]['type']=='image':
-                url = x['content'][1]['url']
+            if x['img_url']:
+                url = x['img_url']
                 response = requests.get(url)
                 img = Image.open(BytesIO(response.content)).convert("RGB")
-                images.append(img)
+                images.append(img)   
+
         except Exception as e:
             print(e)
-            try:
-                if x['content'][0]['type']=='image':
-                    url = x['content'][0]['url']
-                    response = requests.get(url)
-                    img = Image.open(BytesIO(response.content)).convert("RGB")
-                    images.append(img)
-
-            except Exception as e:
-                print(e)
-                pass
+        # 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 Exception as e:
+        #     print(e)
+        #     try:
+        #         if x['content'][0]['type']=='image':
+        #             url = x['content'][0]['url']
+        #             response = requests.get(url)
+        #             img = Image.open(BytesIO(response.content)).convert("RGB")
+        #             images.append(img)
+
+        #     except Exception as e:
+        #         print(e)
+        #         pass
 
     print("images   ",images)
 
@@ -90,7 +100,7 @@ def bot_streaming(message, history, max_new_tokens=250):
 
 
     print("\n\ntexts  final chat text         ", texts)
-
+    print('len of images ', len(images))
     if images == []:
         inputs = processor(text=texts, return_tensors="pt").to("cuda")
     else: