Skip to content

Feature: Add Google Gemini Support and a Streamlit Web UI#16

Open
prakhar920 wants to merge 8 commits into
faizann24:mainfrom
prakhar920:main
Open

Feature: Add Google Gemini Support and a Streamlit Web UI#16
prakhar920 wants to merge 8 commits into
faizann24:mainfrom
prakhar920:main

Conversation

@prakhar920
Copy link
Copy Markdown

Hey Faizan!

I've been using Rogue for my university lab project and it's a fantastic tool. I was inspired to build on it and I'm excited to share a couple of major new features that I think will make it even better.

Here’s what I've added:

  1. A Full Web UI with Streamlit: I built a user-friendly interface so that people can configure and run scans directly from their browser. It streams the logs in real-time and makes the tool much more accessible for everyone, not just those who are comfortable with the command line.

  2. Google Gemini Model Support: The agent can now use Google's gemini-1.5-pro and gemini-1.5-flash models, giving users more flexibility and choice beyond OpenAI.

  3. Direct PDF Report Downloads: From the UI, the final report can be downloaded directly as a clean, professional PDF file, which is great for sharing.

My main goal was to lower the barrier to entry and make this powerful tool even easier for people to pick up and use. I also noticed on your README's TODO list that you were looking to add more model support, so I hope adding Gemini is a good step in that direction!

How to Test:

  1. Clone this branch and run pip install -r requirements.txt.
  2. Run playwright install.
  3. Set up a .env file with a GEMINI_API_KEY or OPENAI_API_KEY.
  4. Launch the app with streamlit run app.py.
  5. Run a scan and test the PDF download functionality.

Thanks for creating such an awesome open-source project! I'd love to hear your feedback on these changes and I'm happy to make any adjustments.

Best,
Prakhar

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant