| import streamlit as st |
| import google.generativeai as genai |
| import os |
|
|
| api_key = os.getenv("GENAI_API_KEY") |
| genai.configure(api_key=api_key) |
|
|
| generation_config = { |
| "temperature": 0.9, |
| "top_p": 1, |
| "top_k": 50, |
| "max_output_tokens": 2048, |
| } |
|
|
| safety_settings = [ |
| {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}, |
| {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}, |
| {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}, |
| {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}, |
| ] |
|
|
| model = genai.GenerativeModel(model_name="gemini-1.0-pro", |
| generation_config=generation_config, |
| safety_settings=safety_settings) |
|
|
| st.title('Project Planner') |
|
|
| project_idea = st.text_area("Describe your coding project idea:", height=150) |
| must_have_features = st.text_input("List must-have features:") |
| timeline = st.text_input("What is your timeline for this project?") |
|
|
| if st.button('Generate Project Charter'): |
| if project_idea and must_have_features and timeline: |
| prompt = ( |
| f"The coding project idea is: {project_idea}\n" |
| f"The must-have features for launch are: {must_have_features}\n" |
| f"The expected timeline for the project is: {timeline}\n" |
|
|
| "Please analyze the project details and generate a draft project charter document covering the following aspects:\n\n" |
|
|
| "- One-paragraph summarizing the project objectives, key requirements, and outcomes\n" |
| "- High-level user flow diagram showing the key screens and user journey\n" |
| "- Breakdown of the high-level technical design and architecture\n" |
| "- Phase-wise plan from design, development, testing to launch\n" |
| "- List of roles required like frontend, backend, devops engineers etc.\n" |
| "- Tools and technologies to be used for the MVP\n" |
| "- Initial effort estimation overview across product, engineering, testing\n" |
| "- Biggest areas of risks and assumptions to validate\n\n" |
|
|
| """For the user flow diagram, please use the mermaid JS syntax, for example:\n\n |
| |
| ```mermaid\n" |
| "flowchart TD\n" |
| " A[Ecommerce] --> B{View product} \n" |
| " B --> C{Add to cart}\n" |
| " C --> D[Payment]\n" |
| " D --> E[Order confirmed]\n" |
| "```\n\n" |
| |
| Provide comprehensive details in each charter section for clarity. Ensure it sets up the key technical guidelines, resource planning, and execution roadmap based on the information provided. |
| """ |
| ) |
|
|
| |
| response = model.generate_content([prompt]) |
| st.write(response.text) |
| else: |
| st.error("Please complete all fields to generate the project charter.") |
|
|
