Feature: Add Google Gemini Support and a Streamlit Web UI#16
Open
prakhar920 wants to merge 8 commits into
Open
Feature: Add Google Gemini Support and a Streamlit Web UI#16prakhar920 wants to merge 8 commits into
prakhar920 wants to merge 8 commits into
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
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:
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.
Google Gemini Model Support: The agent can now use Google's
gemini-1.5-proandgemini-1.5-flashmodels, giving users more flexibility and choice beyond OpenAI.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:
pip install -r requirements.txt.playwright install..envfile with aGEMINI_API_KEYorOPENAI_API_KEY.streamlit run app.py.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