added youtube audio extractor, dice, and direct response for images

bot/plugins/ddg_image_search.py

@@ -1,3 +1,5 @@
+import os
+import random
 from itertools import islice
 from typing import Dict
 
@@ -10,6 +12,9 @@ class DDGImageSearchPlugin(Plugin):
     """
     A plugin to search images and GIFs for a given query, using DuckDuckGo
     """
+    def __init__(self):
+        self.safesearch = os.getenv('DUCKDUCKGO_SAFESEARCH', 'moderate')
+
     def get_source_name(self) -> str:
         return "DuckDuckGo Images"
 
@@ -25,19 +30,45 @@ class DDGImageSearchPlugin(Plugin):
                         "type": "string",
                         "enum": ["photo", "gif"],
                         "description": "The type of image to search for. Default to `photo` if not specified",
+                    },
+                    "region": {
+                        "type": "string",
+                        "enum": ['xa-ar', 'xa-en', 'ar-es', 'au-en', 'at-de', 'be-fr', 'be-nl', 'br-pt', 'bg-bg',
+                                 'ca-en', 'ca-fr', 'ct-ca', 'cl-es', 'cn-zh', 'co-es', 'hr-hr', 'cz-cs', 'dk-da',
+                                 'ee-et', 'fi-fi', 'fr-fr', 'de-de', 'gr-el', 'hk-tzh', 'hu-hu', 'in-en', 'id-id',
+                                 'id-en', 'ie-en', 'il-he', 'it-it', 'jp-jp', 'kr-kr', 'lv-lv', 'lt-lt', 'xl-es',
+                                 'my-ms', 'my-en', 'mx-es', 'nl-nl', 'nz-en', 'no-no', 'pe-es', 'ph-en', 'ph-tl',
+                                 'pl-pl', 'pt-pt', 'ro-ro', 'ru-ru', 'sg-en', 'sk-sk', 'sl-sl', 'za-en', 'es-es',
+                                 'se-sv', 'ch-de', 'ch-fr', 'ch-it', 'tw-tzh', 'th-th', 'tr-tr', 'ua-uk', 'uk-en',
+                                 'us-en', 'ue-es', 've-es', 'vn-vi', 'wt-wt'],
+                        "description": "The region to use for the search. Infer this from the language used for the"
+                                       "query. Default to `wt-wt` if not specified",
                     }
                 },
-                "required": ["query", "type"],
+                "required": ["query", "type", "region"],
             },
         }]
 
     async def execute(self, function_name, **kwargs) -> Dict:
         with DDGS() as ddgs:
+            image_type = kwargs.get('type', 'photo')
             ddgs_images_gen = ddgs.images(
                 kwargs['query'],
-                region="wt-wt",
-                safesearch='off',
-                type_image=kwargs.get('type', 'photo'),
+                region=kwargs.get('region', 'wt-wt'),
+                safesearch=self.safesearch,
+                type_image=image_type,
             )
-            results = list(islice(ddgs_images_gen, 1))
-            return {"result": results[0]["image"]}
+            results = list(islice(ddgs_images_gen, 10))
+            if not results or len(results) == 0:
+                return {"result": "No results found"}
+
+            # Shuffle the results to avoid always returning the same image
+            random.shuffle(results)
+
+            return {
+                'direct_result': {
+                    'kind': image_type,
+                    'format': 'url',
+                    'value': results[0]['image']
+                }
+            }