v1.1.2

Author: DedInc

README.MD

@@ -11,7 +11,8 @@ This module uses web scraping and engineering techniques to interface with Bing'
 ### 🔑 Key Features
 
 - 🖼️ **Generate images** by providing a text prompt
-- 📸 **Get image URLs** up to 4 generated images  
+- 🎥 **Generate videos** by providing a text prompt
+- 📸 **Get image URLs** up to 4 generated images
 - 🔐 **Authentication** via saved Bing cookies or auto-fetched from browsers
 - ⚠️ **Custom exceptions** for common issues
 
@@ -46,8 +47,14 @@ bing_art = BingArt(auto=True)
 
 Call `generate_images()` with your query text:
 
-```python 
-results = bing.generate_images("a cat painting in Picasso style")
+```python
+results = bing_art.generate_images("a cat painting in Picasso style")
+```
+
+And for videos:
+
+```python
+results = bing_art.generate_video("a dancing cat")
 ```
 
 The return value contains image URLs and original prompt: 
@@ -61,6 +68,17 @@ The return value contains image URLs and original prompt:
 }
 ```
 
+For video generation, the output will be:
+
+```json
+{
+  "video": {
+    "video_url": "https://..."
+  },
+  "prompt": "a dancing cat"
+}
+```
+
 ## 🚨 Exceptions
 
 - `AuthCookieError`: Invalid authentication cookie

bingart/bingart.py

@@ -51,33 +51,40 @@ def get_auth_cookies(self):
 
         raise AuthCookieError('Failed to fetch authentication cookies automatically.')
 
-    def _prepare_headers(self):
+    def _prepare_headers(self, content_type='image'):
         cookie_str = ''
         if self.auth_cookie_U:
             cookie_str += f'_U={self.auth_cookie_U};'
         if self.auth_cookie_KievRPSSecAuth:
             cookie_str += f' KievRPSSecAuth={self.auth_cookie_KievRPSSecAuth};'
 
-        return {
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
-            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8',
-            'Referer': self.base_url,
-            'Accept-Language': 'en-US,en;q=0.9',
+        base_headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36',
+            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
+            'Accept-Language': 'ru-RU,ru;q=0.9,en-US;q=0.8,en;q=0.7',
             'Cookie': cookie_str
         }
 
+        if content_type == 'video':
+            base_headers['Referer'] = f'{self.base_url}?ctype=video'
+            base_headers['Content-Type'] = 'application/x-www-form-urlencoded'
+        else:
+            base_headers['Referer'] = self.base_url
+
+        return base_headers
+
     def _get_balance(self):
         response = self.session.get(self.base_url)
         try:
-            coins = int(re.search(r'<div id="reward_c" data-tb="(\d+)"', response.text).group(1))
+            coins = 15 - int(re.search(r'<div id="reward_c" data-tb="(\d+)"', response.text).group(1))
         except AttributeError:
             raise AuthCookieError('Auth cookie failed!')
         return coins
 
     def _fetch_images(self, encoded_query, ID, IG):
         images = []
         while True:
-            response = self.session.get(f'{self.base_url}/async/results/{ID}?{encoded_query}&IG={IG}&IID=images.as'.replace('&amp;nfy=1', ''))
+            response = self.session.get(f'{self.base_url}/async/results/{ID}?{encoded_query}&IG={IG}&IID=images.as'.replace('&nfy=1', ''))
             if 'text/css' in response.text:
                 src_urls = re.findall(r'src="([^"]+)"', response.text)
                 for src_url in src_urls:
@@ -87,22 +94,71 @@ def _fetch_images(self, encoded_query, ID, IG):
                 return images
             time.sleep(5)
 
-    def generate_images(self, query):
+    def _fetch_video(self, encoded_query, ID):
+        while True:
+            response = self.session.get(f'{self.base_url}/async/results/{ID}?{encoded_query}&ctype=video&sm=1&girftp=1')
+            try:
+                result = response.json()
+                if result.get('errorMessage') == 'Pending':
+                    time.sleep(5)
+                    continue
+                elif result.get('showContent'):
+                    return {'video_url': result['showContent']}
+                else:
+                    return None
+            except ValueError:
+                video_url = re.search(r'ourl="([^"]+)"', response.text)
+                if video_url:
+                    return {'video_url': video_url.group(1)}
+                else:
+                    time.sleep(5)
+                    continue
+    
+    def _handle_creation_error(self, response):
+        if 'data-clarity-tag="BlockedByContentPolicy"' in response.text or 'girer_center block_icon' in response.text:
+            raise PromptRejectedError('Error! Your prompt has been rejected for content policy reasons.')
+        else:
+            raise AuthCookieError('Auth cookie failed!')
+
+    def generate(self, query, content_type='image'):
         encoded_query = urlencode({'q': query})
-        self.session.headers.update(self.headers)
-        coins = self._get_balance()
-        rt = '4' if coins > 0 else '3'
-        creation_url = f'{self.base_url}?{encoded_query}&rt={rt}&FORM=GENCRE'
+        
+        if content_type == 'image':
+            headers = self.headers
+            coins = self._get_balance()
+            rt = '4' if coins > 0 else '3'
+            creation_url = f'{self.base_url}?{encoded_query}&rt={rt}&FORM=GENCRE'
+        elif content_type == 'video':
+            headers = self._prepare_headers(content_type='video')
+            creation_url = f'{self.base_url}?{encoded_query}&rt=4&FORM=GENCRE&ctype=video&pt=4&sm=1'
+        else:
+            raise ValueError("content_type must be 'image' or 'video'")
 
+        self.session.headers.update(headers)
         response = self.session.post(creation_url, data={'q': query})
+
         try:
-            ID = re.search(';id=([^"]+)"', response.text).group(1)
-            IG = re.search('IG:"([^"]+)"', response.text).group(1)
+            if content_type == 'image':
+                ID = re.search(';id=([^"]+)"', response.text).group(1)
+                IG = re.search('IG:"([^"]+)"', response.text).group(1)
+            else:
+                ID = re.search(r'id=([^&]+)', response.url).group(1)
+                IG = None
         except AttributeError:
-            raise PromptRejectedError('Error! Your prompt has been rejected for ethical reasons.')
+            self._handle_creation_error(response)
+
+        if content_type == 'image':
+            result = self._fetch_images(encoded_query, ID, IG)
+            return {'images': result, 'prompt': query}
+        else:
+            result = self._fetch_video(encoded_query, ID)
+            return {'video': result, 'prompt': query}
+
+    def generate_images(self, query):
+        return self.generate(query, content_type='image')
 
-        images = self._fetch_images(encoded_query, ID, IG)
-        return {'images': images, 'prompt': query}
+    def generate_video(self, query):
+        return self.generate(query, content_type='video')
 
     def close_session(self):
         self.session.close()

setup.py

@@ -5,10 +5,10 @@
 
 setup(
     name='bingart',
-    version='1.1.1',
+    version='1.1.2',
     author='Maehdakvan',
     author_email='visitanimation@google.com',
-    description='bingart is an unofficial API wrapper for Bing Image Creator (based on DALL-E 3).',
+    description='bingart is an unofficial API wrapper for Bing Image & Video Creator (based on DALL-E 3).',
     long_description=long_description,
     long_description_content_type='text/markdown',
     url='https://github.com/DedInc/bingart',