From 81ac77316812ba3e5980d90fdff4c6dbb8ff1d48 Mon Sep 17 00:00:00 2001
From: Auric Vente <mprops@protonmail.com>
Date: Sat, 20 Apr 2024 13:28:39 -0600
Subject: [PATCH] Mods

---
 main.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/main.py b/main.py
index 19c751c..ab97869 100644
--- a/main.py
+++ b/main.py
@@ -247,6 +247,9 @@ async def on_message(ws, message):
         if (not name) or (not ext):
             return
 
+        if ext not in [".jpg", ".jpeg", ".png", ".gif", ".webm", ".mp4"]:
+            return
+
         last_file = f"https://deek.chat/storage/files/{name}"
         last_file_ext = ext
     elif data["type"] in ["message", "messageEnd"]: