From 09e811a2b7d86f9675972d6d4acf734903070b3c Mon Sep 17 00:00:00 2001
From: Krzysztof Rudnicki <krzysztofrudnicki0@gmail.com>
Date: Sun, 23 Jun 2024 23:03:20 +0200
Subject: [PATCH] fix: self sending !kot works now

---
 .gitignore |  2 ++
 main.py    | 11 +++++------
 run.sh     |  3 +++
 3 files changed, 10 insertions(+), 6 deletions(-)
 create mode 100644 run.sh

diff --git a/.gitignore b/.gitignore
index 82f9275..4f8d84e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -160,3 +160,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+*.jpg
+*.gif
\ No newline at end of file
diff --git a/main.py b/main.py
index fdc45de..5f9cc56 100644
--- a/main.py
+++ b/main.py
@@ -35,7 +35,7 @@ def fetch_and_download_cat_image():
     with open(image_filename, 'rb') as image_file:
         base64_encoded_data = base64.b64encode(image_file.read()).decode('utf-8')
     
-    print(f"Base64 Encoded Data: {base64_encoded_data}")
+    os.remove(image_filename)
     return base64_encoded_data
 
 def send_cat():
@@ -61,13 +61,12 @@ async def listen_to_server():
         try:
             async for message in websocket:
                 message_json = json.loads(message)
-                #inside_message = message_json.get('envelope', {}).get('syncMessage', {}).get('sentMessage', {})
                 inside_message = message_json.get('envelope', {}).get('dataMessage', {})
+                if inside_message == {}:
+                    inside_message = message_json.get('envelope', {}).get('syncMessage', {}).get('sentMessage', {})
                 print("message", message)
-                if inside_message.get('message') == "!kot":
-                    print("kot message detected")
-                    if inside_message.get('groupInfo', {}).get('groupId', {}) == GROUP_ID:
-                        send_cat()
+                if inside_message.get('message') == "!kot" and inside_message.get('groupInfo', {}).get('groupId', {}) == GROUP_ID:
+                    send_cat()
         except websockets.ConnectionClosed as e:
             print(f"Connection closed: {e}")
 
diff --git a/run.sh b/run.sh
new file mode 100644
index 0000000..fdc41eb
--- /dev/null
+++ b/run.sh
@@ -0,0 +1,3 @@
+#!/bin/sh
+
+sudo apt-get install python3-venv
\ No newline at end of file