Create app.py

app.py

@@ -0,0 +1,197 @@
+import streamlit as st
+import pandas as pd
+import matplotlib.pyplot as plt
+from dotenv import load_dotenv
+import os
+import google.generativeai as genai
+from io import BytesIO
+
+# Load environment variables
+load_dotenv()
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+def get_gemini_response(prompt):
+    model = genai.GenerativeModel('gemini-pro')
+    response = model.generate_content(prompt)
+    return response.text
+
+#######################
+# REVISED FUNCTIONS
+#######################
+def estimate_cost(project_description):
+    prompt = f"Provide a cost estimation for the following project: {project_description}"
+    return get_gemini_response(prompt)
+
+def assess_sustainability(project_description):
+    prompt = (
+        f"Analyze the sustainability and environmental impact of this project. "
+        f"Suggest eco-friendly materials, energy-saving approaches, and best practices "
+        f"that reduce the carbon footprint and ensure environmental compliance. "
+        f"Project Details: {project_description}"
+    )
+    return get_gemini_response(prompt)
+
+def optimize_resources(project_description):
+    prompt = (
+        f"Suggest an optimal allocation of resources for this project, clearly specifying "
+        f"the types and quantities of personnel (e.g., project managers, engineers, specialists), "
+        f"equipment, software, and budget needed to complete the project effectively. "
+        f"Project Details: {project_description}"
+    )
+    return get_gemini_response(prompt)
+
+def manage_risks(project_description):
+    prompt = (
+        f"Identify potential risks for this project. Provide them in categories such as: "
+        f"Financial, Operational, Compliance, Strategic, etc. For each category, "
+        f"describe the specific risks and suggest corresponding mitigation strategies. "
+        f"Project Details: {project_description}"
+    )
+    return get_gemini_response(prompt)
+
+def prioritize_tasks(project_description):
+    prompt = f"List and prioritize critical tasks for the successful completion of this project: {project_description}"
+    return get_gemini_response(prompt)
+
+def optimize_schedule(project_description):
+    # Renamed to Project Schedule
+    prompt = (
+        f"Create a project schedule (when and how to deliver) for the following project, "
+        f"including timeline and sequencing of tasks to ensure timely completion: {project_description}"
+    )
+    return get_gemini_response(prompt)
+
+def get_project_scope(project_description):
+    prompt = (
+        f"Identify the project scope and deliverables for the following project, specifying "
+        f"goals, tasks, and features that need to be completed: {project_description}"
+    )
+    return get_gemini_response(prompt)
+
+#######################
+# STREAMLIT APP
+#######################
+st.set_page_config(page_title="PlanMaster AI", layout="wide")
+
+# Title & Introduction
+st.markdown("<h1 style='text-align: center; color: #ff6347;'>PlanMaster AI</h1>", unsafe_allow_html=True)
+st.markdown(
+    """
+    <div style='text-align: center; color: #4682b4;'>
+        <h3>Welcome to PlanMaster AI</h3>
+        <p>
+          This application leverages AI to help you plan and manage projects more effectively.
+          Simply provide a project description and select the type of analysis you want to perform.
+        </p>
+    </div>
+    """, 
+    unsafe_allow_html=True
+)
+st.write("---")
+
+# Sidebar
+st.sidebar.markdown("<h2 style='color: #4CAF50;'>Select Analysis Task</h2>", unsafe_allow_html=True)
+
+task_options = {
+    "Cost Estimation": "#ff6347",
+    "Sustainability Assessment": "#4682b4",
+    "Resource Optimization": "#32cd32",
+    "Risk Management": "#ff8c00",
+    "Task Prioritization": "#8a2be2",
+    "Project Schedule": "#ff1493",
+    "Project Scope": "#9400d3"
+}
+
+if "selected_task" not in st.session_state:
+    st.session_state["selected_task"] = None
+
+for option, color in task_options.items():
+    if st.sidebar.button(option):
+        st.session_state["selected_task"] = option
+
+selected_task = st.session_state["selected_task"]
+
+# Project Description
+if 'description_input' not in st.session_state:
+    st.session_state['description_input'] = ""
+
+project_description = st.text_area("Enter Project Description", st.session_state['description_input'])
+
+# Remove "Run All Analyses" button or comment it out
+# if st.button("Run All Analyses"):
+#     st.session_state["selected_task"] = "Run All"
+
+if selected_task:
+    st.write(f"## Current Task: {selected_task}")
+else:
+    st.write("## Please select a task from the sidebar.")
+
+def display_sample_table():
+    data = {
+        'Category': ['Materials', 'Labor', 'Equipment', 'Software', 'Services', 'Travel'],
+        'Cost Estimate ($)': [5000, 10000, 8000, 2000, 3000, 1500]
+    }
+    df = pd.DataFrame(data)
+    st.table(df)
+
+def display_sample_chart():
+    data = {
+        'Risk Factor': ['Financial', 'Operational', 'Compliance', 'Strategic'],
+        'Likelihood (%)': [35, 20, 25, 20],
+    }
+    df = pd.DataFrame(data)
+    fig, ax = plt.subplots()
+    ax.bar(df['Risk Factor'], df['Likelihood (%)'], color='#ff8c00')
+    ax.set_title("Risk Factor Likelihood")
+    st.pyplot(fig)
+
+if project_description.strip():
+    st.write("### Task Result")
+    if selected_task == "Cost Estimation":
+        st.markdown(f"<h4 style='color: {task_options[selected_task]};'>Cost Estimation Result 💸</h4>", unsafe_allow_html=True)
+        st.write(estimate_cost(project_description))
+        display_sample_table()
+    elif selected_task == "Sustainability Assessment":
+        st.markdown(f"<h4 style='color: {task_options[selected_task]};'>Sustainability Assessment Result 🌍</h4>", unsafe_allow_html=True)
+        st.write(assess_sustainability(project_description))
+    elif selected_task == "Resource Optimization":
+        st.markdown(f"<h4 style='color: {task_options[selected_task]};'>Resource Optimization Result 📊</h4>", unsafe_allow_html=True)
+        st.write(optimize_resources(project_description))
+    elif selected_task == "Risk Management":
+        st.markdown(f"<h4 style='color: {task_options[selected_task]};'>Risk Management Result ⚠️</h4>", unsafe_allow_html=True)
+        st.write(manage_risks(project_description))
+        display_sample_chart()
+    elif selected_task == "Task Prioritization":
+        st.markdown(f"<h4 style='color: {task_options[selected_task]};'>Task Prioritization Result 📋</h4>", unsafe_allow_html=True)
+        st.write(prioritize_tasks(project_description))
+    elif selected_task == "Project Schedule":
+        st.markdown(f"<h4 style='color: {task_options[selected_task]};'>Project Schedule Result 📅</h4>", unsafe_allow_html=True)
+        st.write(optimize_schedule(project_description))
+    elif selected_task == "Project Scope":
+        st.markdown(f"<h4 style='color: {task_options[selected_task]};'>Project Scope Result 📝</h4>", unsafe_allow_html=True)
+        st.write(get_project_scope(project_description))
+
+########################
+# CUSTOM PROMPT SECTION
+########################
+st.write("---")
+st.write("### Run a Custom Prompt")
+custom_prompt = st.text_area("Enter any additional prompt you'd like to run")
+if st.button("Run Custom Prompt"):
+    if custom_prompt.strip():
+        custom_response = get_gemini_response(custom_prompt)
+        st.write("#### Custom Prompt Result")
+        st.write(custom_response)
+    else:
+        st.warning("Please enter a prompt before running.")
+
+########################
+# FOOTER
+########################
+st.write("---")
+st.markdown(
+    "<p style='text-align: center; color: #808080;'>"
+    "Created by Nnenna Ginikanwa | Designed by Chidozie"
+    "</p>", 
+    unsafe_allow_html=True
+)