Update app.py

app.py

@@ -1,38 +1,56 @@
 import streamlit as st
 import openai
 
-# Define your sophisticated prompts and other necessary components from your script
+# Assuming your OpenAI API key is set in an environment variable or entered in the app
+# For the app, we'll ask the user to input it directly for demonstration purposes
 
 def generate_detailed_artifact(api_key, description, artifact_type):
     """
-    Interact with OpenAI using the provided API key to generate the artifact content.
+    Generate a detailed and sophisticated artifact for software development projects.
     """
-    openai.api_key = api_key
-    # Your existing logic to generate artifact
+    sophisticated_prompts = {
+        "Strategic Alignment Report": "Craft a comprehensive report that articulates the alignment of AI initiatives with the organization's long-term strategic goals, demonstrating how AI can be leveraged for strategic advantage.",
+        "Enhanced Process Maps": "Develop enhanced process maps indicating AI integration points. Detail the current versus future state of processes, specifying where AI brings value.",
+        "AI Impact Forecast": "Generate a forecast report that assesses the potential impacts of AI on business processes, scalability of solutions, and a framework for measuring and monitoring these impacts.",
+        "Stakeholder Engagement Plan": "Create a detailed stakeholder engagement plan that lists all stakeholders, their concerns, expectations, and strategies for managing the cultural shift towards AI adoption.",
+        "AI Risk Mitigation Strategy": "Construct a comprehensive risk mitigation strategy document outlining the potential risks of AI integration and detailed plans for addressing these risks.",
+        "Cost-Benefit Analysis Document": "Produce a cost-benefit analysis document for AI projects, detailing investment requirements, financial benefits, total cost of ownership, and prioritization of initiatives based on financial metrics.",
+        "AI Prototyping Strategy": "Outline a strategic plan for prototyping AI initiatives, including the prototyping phases, success metrics, resource needs, and a feedback incorporation mechanism.",
+        "Change Management Framework": "Develop a change management framework that analyzes the impact of AI, provides a detailed change plan, identifies training needs, and establishes clear communication strategies.",
+        "AI Monitoring and Review Framework": "Design a continuous AI monitoring and evaluation framework that details the KPIs, integration with BI systems, and the periodic review process for AI initiatives.",
+        "AI Roadmap and Action Plan": "Synthesize a strategic AI roadmap and action plan document based on workshop findings, assigning responsibilities and deadlines for the completion of action items."
+    }
+
+    prompt = sophisticated_prompts.get(artifact_type, "Generate a detailed document section:") + "\n" + description
+
+    try:
+        openai.api_key = api_key
+        response = openai.Completion.create(
+            model="gpt-4",  # Adjust model as necessary
+            prompt=prompt,
+            temperature=0.5,
+            max_tokens=1024,
+            top_p=1.0,
+            frequency_penalty=0.0,
+            presence_penalty=0.0
+        )
+        return response.choices[0].text.strip()
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
 st.title("AI-Enabled Project Artifact Generator")
 
-# Use Streamlit's session state to remember the API key if it's already been entered
-if 'api_key' not in st.session_state:
-    st.session_state.api_key = ''
+api_key = st.text_input("Enter your OpenAI API key", type="password")
 
-api_key_input = st.text_input("Enter your OpenAI API Key", type="password", value=st.session_state.api_key)
+artifact_type = st.selectbox("Select Artifact Type", list(sophisticated_prompts.keys()))
 
-if api_key_input:
-    st.session_state.api_key = api_key_input  # Update session state with the new key
-    with st.form("artifact_generator"):
-        artifact_types = list(sophisticated_prompts.keys())
-        description = st.text_area("Enter Project Description", "Type your project description here...")
-        selected_artifact_types = st.multiselect("Select Artifact Types", artifact_types, default=artifact_types[0])
-        generate_button = st.form_submit_button("Generate Artifacts")
+description = st.text_area("Project Description", "Enter your project description here.")
 
-    if generate_button and st.session_state.api_key:
-        tab_container = st.tabs([f"{artifact_type}" for artifact_type in selected_artifact_types])
+if st.button("Generate Artifact"):
+    if api_key and description and artifact_type:
+        generated_artifact = generate_detailed_artifact(api_key, description, artifact_type)
+        st.markdown("## Generated Artifact")
+        st.write(generated_artifact)
+    else:
+        st.error("Please fill out all fields.")
 
-        for i, artifact_type in enumerate(selected_artifact_types):
-            artifact_content = generate_detailed_artifact(st.session_state.api_key, description, artifact_type)
-
-            with tab_container[i]:
-                st.markdown(artifact_content, unsafe_allow_html=True)
-else:
-    st.warning("Please enter your OpenAI API key to use the app.")