Update app.py

app.py

@@ -27,22 +27,22 @@ def process_image(data):
 def generate(message, history, model, system_prompt,
              temperature=1.0, top_p=1.0, frequency_penalty=0.0, presence_penalty=0.0):
     # history
-    history_openai_format = [{"role": "system", "content": system_prompt}]
+    history_openai_format=[{"role": "system", "content": system_prompt}]
     for user, assistant in history:
-        if isinstance(user, tuple):  # there were 
+        if isinstance(user, tuple):  # there were files
             content = []
             for filepath in user:
-                mime_type = get_mimetype(filepath)
-                if not mime_type.contains('image'):
-                    raise gr.Error("Momenteel zijn alleen afbeeldingen ondersteund als bijlagen 💥!", duration=5)
-                content.append(
-                    {"type": "image_url",
-                     # for some reason you don't get the same image format in history as in message
-                     "image_url": {"url": process_image({'path': filepath,
-                                                         'mime_type': mime_type})}})
-
-            history_openai_format.append(
-                {"role": "user", "content": content})
+                mime_type = get_mimetype(filepath) or ''
+                if mime_type.startswith("image/"):
+                    content.append(
+                        {"type": "image_url",
+                         # for some reason you don't get the same image format in history as in message
+                         "image_url": {"url": process_image({'path': filepath,
+                                                             'mime_type': get_mimetype(filepath)})}}
+                    )
+            if content:
+                history_openai_format.append(
+                    {"role": "user", "content": content})
         else:  # there was just text
             history_openai_format.append({"role": "user", "content": user})
 
@@ -54,6 +54,9 @@ def generate(message, history, model, system_prompt,
                 "text": message['text']}]
 
     for file in message['files']:
+        mime_type = get_mimetype(file['path']) or ''
+        if not mime_type.startswith('image/'):
+            raise gr.Error("Momenteel zijn alleen afbeeldingen ondersteund als bijlagen 💥!", duration=20)
         content.append({"type": "image_url",
                         "image_url": {"url": process_image(file)}})