updates

added requirements.txt
updated image_generator.py

image_generator.py

@@ -1,33 +1,9 @@
-import pollinations
 import constants
 from PIL import Image
 from gradio_client import Client
 
 
-class ImageGenerator:
-    def __init__(self, model=pollinations.Image.flux(), seed="random", width=720, height=1280, enhance=False, nologo=True, private=True, safe=False, referrer="pollinations.py"):
-        # Initialize the image model with provided parameters
-        self.image_model = pollinations.Image(
-            model=model,
-            seed=seed,
-            width=width,
-            height=height,
-            enhance=False,
-            nologo=nologo,
-            private=private,
-            safe=safe,
-            referrer=referrer
-        )
-    
-    def generate_image_with_pollinations_ai(self, prompt):
-        # Generate image using the provided prompt
-        try:
-            image = self.image_model(prompt=prompt)
-            return image  # Return the generated image object
-        except Exception as e:
-            print(f"Error generating image: {e}")
-            return None  # Return None if there's an error
-        
+class ImageGenerator:     
 
     def generate_image(self, prompt, path='test_image.png'):
         try:

youtube_short_generator.py

@@ -59,10 +59,6 @@ class YoutubeShortGenerator:
         # generator.generate_image(self.result.title_image_prompt, path=f"{folder_path}/title.png")
         print("Title Prompt: ",self.result.title_image_prompt)
 
-        # generate images using pollinations_ai
-        # title_image = generator.generate_image(self.result.title_image_prompt)
-        # title_image.save(f"{folder_path}/title.png")
-
         # generate images using stable-diffusion-turbo
         generator.generate_image(self.result.title_image_prompt, path=f"{folder_path}/title.png")
 
@@ -76,10 +72,6 @@ class YoutubeShortGenerator:
         # Generate and save images
         for index, image_prompt in enumerate(image_prompts):
             
-            # generate images using pollinations_ai
-            # image = generator.generate_image(image_prompt)  
-            # image.save(f"{folder_path}/{index}.png")
-            
             # generate images using stable-diffusion-turbo
             generator.generate_image(image_prompt, f"{folder_path}/{index}.png" )  
             print(f"Image {index} saved.")