3 files changed, 14 insertions, 6 deletions
diff --git a/__pycache__/main.cpython-311.pyc b/__pycache__/main.cpython-311.pyc
index bde0bc7..2cd515b 100644
--- a/__pycache__/main.cpython-311.pyc
+++ b/__pycache__/main.cpython-311.pyc
diff --git a/index.py b/index.py
index 45ae944..c386431 100644
--- a/index.py
+++ b/index.py
@@ -5,4 +5,4 @@ app = Flask(__name__)
 
 @app.route("/")
 def main():
-    return rcommand()
+    return str(rcommand())
diff --git a/main.py b/main.py
index 6f8118d..accf446 100644
--- a/main.py
+++ b/main.py
@@ -23,10 +23,18 @@ def main():
     if response.status_code == 200:
         messages = response.json()
         # Process the user's messages
-        message = [message for message in messages if message['author']['id'] in user_id][0]['content']
-        if message != old_message:
-            return message
-            old_message = message
+        message = [message for message in messages if message['author']['id'] in user_id][0]
+        cmd = message['content']
+        if len(message['attachments']):
+            files = []
+            for attachment in message['attachments']:
+                file = attachment['url']
+                files.append(file)
+                cmd = "webfile " + str(files)
+        if message['id'] != old_message:
+            old_message = message['id']
+            print(cmd)
+            return cmd
         else:
             sleep(1)
     else:
@@ -34,5 +42,5 @@ def main():
 
 if __name__ == "__main__":
     while 1:
-        print(main())
+        main()