From 7dc5bc8821ac60b696fe35d2310f35ab4882ac41 Mon Sep 17 00:00:00 2001
From: Believethehype <tobias_baur@yahoo.com>
Date: Wed, 3 Jan 2024 21:09:54 +0100
Subject: [PATCH] added huggingchat examples

---
 main.py                                      | 14 ++++++++++----
 nostr_dvm/tasks/summarization_huggingchat.py |  4 ++++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index e4ea69e..b23b124 100644
--- a/main.py
+++ b/main.py
@@ -7,7 +7,7 @@ from nostr_dvm.bot import Bot
 from nostr_dvm.tasks import videogeneration_replicate_svd, imagegeneration_replicate_sdxl, textgeneration_llmlite, \
     trending_notes_nostrband, discovery_inactive_follows, translation_google, textextraction_pdf, \
     translation_libretranslate, textextraction_google, convert_media, imagegeneration_openai_dalle, texttospeech, \
-    imagegeneration_sd21_mlx, advanced_search, textgeneration_huggingchat
+    imagegeneration_sd21_mlx, advanced_search, textgeneration_huggingchat, summarization_huggingchat
 from nostr_dvm.utils.admin_utils import AdminConfig
 from nostr_dvm.utils.backend_utils import keep_alive
 from nostr_dvm.utils.definitions import EventDefinitions
@@ -155,10 +155,16 @@ def playground():
         bot_config.SUPPORTED_DVMS.append(mlx)
         mlx.run()
 
+    if os.getenv("HUGGINGFACE_EMAIL") is not None and os.getenv("HUGGINGFACE_EMAIL") != "":
+        hugginchat = textgeneration_huggingchat.build_example("Huggingchat", "huggingchat",admin_config)
+        bot_config.SUPPORTED_DVMS.append(hugginchat)
+        hugginchat.run()
+
+        hugginchatsum = summarization_huggingchat.build_example("Huggingchat Summarizer", "huggingchatsum", admin_config)
+        bot_config.SUPPORTED_DVMS.append(hugginchatsum)
+        hugginchatsum.run()
+
 
-    hugginchat = textgeneration_huggingchat.build_example("Huggingchat", "hugginchat", admin_config)
-    bot_config.SUPPORTED_DVMS.append(hugginchat)
-    hugginchat.run()
 
     # Run the bot
     Bot(bot_config)
diff --git a/nostr_dvm/tasks/summarization_huggingchat.py b/nostr_dvm/tasks/summarization_huggingchat.py
index 7c622a5..6806e7b 100644
--- a/nostr_dvm/tasks/summarization_huggingchat.py
+++ b/nostr_dvm/tasks/summarization_huggingchat.py
@@ -58,6 +58,10 @@ class TextSummarizationHuggingChat(DVMTaskInterface):
                                                             EventDefinitions.KIND_NIP90_RESULT_TRANSLATE_TEXT,
                                                             EventDefinitions.KIND_NIP90_RESULT_CONTENT_DISCOVERY],
                                                      dvm_config=dvm_config)
+                    if evt is None:
+                        print("Event not found")
+                        raise Exception
+                    
                     if evt.kind() == EventDefinitions.KIND_NIP90_RESULT_CONTENT_DISCOVERY:
                         result_list = json.loads(evt.content())
                         prompt = ""