Update app.py

app.py

@@ -36,24 +36,23 @@ model.generation_config.eos_token_id = 128009
 @spaces.GPU
 def bot_streaming(message, history):
     print(message)
+    image = None  # Initialize image variable
     if message["files"]:
         # message["files"][-1] is a Dict or just a string
         if type(message["files"][-1]) == dict:
-            if image is None:
-                image = "http://files.tatl.tv/ignore.png"
-            else:
-                image = message["files"][-1]["path"]
+            image = message["files"][-1]["path"]
         else:
-            if image is None:
-                image = "http://files.tatl.tv/ignore.png"
-            else:
-                image = message["files"][-1]
+            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]
+                break  # Exit loop after finding the first image
+
+    if image is None:
+        image = "http://files.tatl.tv/ignore.png"
     
     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}")