From 78ddb3fbc68052b17211ffb59dd2e2a366a803de Mon Sep 17 00:00:00 2001
From: Believethehype <1097224+believethehype@users.noreply.github.com>
Date: Tue, 3 Sep 2024 09:46:24 +0200
Subject: [PATCH] fixes nip89

---
 tests/generic_dvm_autotopic_feed.py |  2 +-
 tests/generic_dvm_flux_hf.py        | 18 ++++++++++--------
 2 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/tests/generic_dvm_autotopic_feed.py b/tests/generic_dvm_autotopic_feed.py
index 6bf1e40..052bb24 100644
--- a/tests/generic_dvm_autotopic_feed.py
+++ b/tests/generic_dvm_autotopic_feed.py
@@ -57,7 +57,7 @@ def playground(announce=False):
     }
 
     nip89config = NIP89Config()
-    nip89config.KIND = kind
+    nip89config.KIND = Kind(kind)
     nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89info["image"])
     nip89config.CONTENT = json.dumps(nip89info)
 
diff --git a/tests/generic_dvm_flux_hf.py b/tests/generic_dvm_flux_hf.py
index 388d57e..25e62ff 100644
--- a/tests/generic_dvm_flux_hf.py
+++ b/tests/generic_dvm_flux_hf.py
@@ -42,7 +42,7 @@ def playground(announce=False):
     }
 
     nip89config = NIP89Config()
-    nip89config.KIND = kind
+    nip89config.KIND = Kind(kind)
     nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89info["image"])
     nip89config.CONTENT = json.dumps(nip89info)
 
@@ -65,7 +65,7 @@ def playground(announce=False):
         def query(payload):
             response = requests.post(API_URL, headers=headers, json=payload)
             if response.status_code == 500:
-                raise Exception("Internal server error, try again later.")
+                raise Exception(response.text)
             else:
                 return response.content
 
@@ -76,12 +76,14 @@ def playground(announce=False):
         })
         import io
         from PIL import Image
-
-        image = Image.open(io.BytesIO(image_bytes))
-        image.save("flux.png")
-        url = await upload_media_to_hoster("flux.png")
-        os.remove("flux.png")
-        return url
+        try:
+            image = Image.open(io.BytesIO(image_bytes))
+            image.save("flux.png")
+            url = await upload_media_to_hoster("flux.png")
+            os.remove("flux.png")
+            return url
+        except Exception as e:
+            raise Exception(e)
 
 
     dvm.process = process  # overwrite the process function with the above one