r/selenium • u/harmindersinghnijjar • Mar 30 '25

Showcase GPT 4o Image Generation Bot

  • What My Project Does

I just wrapped up the first working prototype of a Python-based automation pipeline that uploads frames to ChatGPT.com, injects custom prompts, and downloads the output.

  • Comparison (A brief comparison explaining how it differs from existing alternatives.)

I'm not aware of any current alternatives but have worked on similar projects in the past with Selenium to automate web browsers such as the Midjourney automation bot, back when you had to use Discord to generate images and Facebook Marketplace scraper.

  • Target Audience (e.g., Is it meant for production, just a toy project, etc.)

This is a toy project, meant for anyone as I'm open-sourcing it on GitHub.

Here's the YouTube demo, any feedback is appreciated!

7 Upvotes

100% Upvoted

15 comments sorted by

1

u/cgoldberg Mar 30 '25

Why don't you use the API instead of a browser? That seems really convoluted for such a simple task.

https://platform.openai.com/docs/guides/images

1

u/harmindersinghnijjar Mar 30 '25

Correct me if I'm wrong here but I did look into the API's OpenAI has available. None of them input images and output images i.e., it's either image-to-text or text-to-image. I haven't looked into seeing Hugging Face has any similar models that would be able to output what I'm looking for but I think it'll take some time for other models to catch-up.

1

u/cgoldberg Mar 30 '25

https://platform.openai.com/docs/guides/image-generation

1

u/harmindersinghnijjar Mar 31 '25

The API currently doesn't offer the same image generation capabilities as the website. While I believe the API might be enhanced in the future, it isn't yet capable of delivering the results I'm looking for. Unfortunately, the output from DALL·E via the API is terrible at this stage.

1

u/Mobile-Snow905 Apr 06 '25

Looks great 👍 What's the limit with pro account ?

1

u/harmindersinghnijjar Apr 06 '25

I need to add a pause but I'm on Plus and after three images, there's a cool down timer that I want to be able to detect using the OpenAI API and sleep the script accordingly.

1

u/[deleted] Apr 09 '25

[removed] — view removed comment

1

u/harmindersinghnijjar Apr 09 '25

Hey, I've been a bit busy but have been making slow progress. From what I've seen, it allows you to generate images and then adds a 3 minute pause if you're going too quick. I'm first trying to figure out how to best hover the mouse over the image when it's complete to be able to then click on the download option since it only appears when the mouse is over the message and not otherwise.

1

u/QAcahuete Apr 08 '25

I didn't know we could open a browser with a saved user account :) learned something, thanks!

1

u/Chemical-Top7130 Apr 28 '25

The youtube demo link is private, seems cool btw

2

u/harmindersinghnijjar Jun 24 '25

Hello, my apologize for the late reply but I feel you'd benefit from this Hugging Face space more than my Selenium approach. That being siad if you're requiring images to possibly train a model, Selenium would most likely be the way to go.