Update app.py

app.py

@@ -39,21 +39,22 @@ def bot_streaming(message, history):
     if message["files"]:
         # message["files"][-1] is a Dict or just a string
         if type(message["files"][-1]) == dict:
-            image = message["files"][-1]["path"]
+            if image is None:
+                image = "http://files.tatl.tv/ignore.png"
+            else:
+                image = message["files"][-1]["path"]
         else:
-            image = message["files"][-1]
+            if image is None:
+                image = "http://files.tatl.tv/ignore.png"
+            else:
+                image = message["files"][-1]
     else:
         # if there's no image uploaded for this turn, look for images in the past turns
         # kept inside tuples, take the last one
         for hist in history:
             if type(hist[0]) == tuple:
                 image = hist[0][0]
-    if image is None:
-        # Handle the case where image is None
-        image = "http://files.tatl.tv/ignore.png"
-    else:
-        image = image
-
+    
     prompt = f"<|start_header_id|>user<|end_header_id|>\n\n<image>\n{message['text']}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
     # print(f"prompt: {prompt}")
     image = Image.open(image)