Update app.py

app.py

@@ -150,7 +150,7 @@ def _generate_with_openai_provider(client, model_id, prompt, max_tokens, system_
         response = client.chat.completions.create(
             model=model_id,
             messages=messages,
-            temperature=0.6,
+            temperature=0.7, # Slightly higher temp might encourage more diverse options
             max_tokens=max_tokens
         )
         return response.choices[0].message.content
@@ -169,7 +169,7 @@ def _generate_with_gemini_provider(client, model_id, prompt, max_tokens, system_
                 'HARM_CATEGORY_SEXUALLY_EXPLICIT': 'block_none',
                 'HARM_CATEGORY_DANGEROUS_CONTENT': 'block_none',
             },
-            generation_config=client.types.GenerationConfig(temperature=0.7)
+            generation_config=client.types.GenerationConfig(temperature=0.8) # Slightly higher temp
         )
         response = model.generate_content(full_prompt)
 
@@ -206,7 +206,7 @@ def _generate_with_deepseek_provider(api_key, model_id, prompt, max_tokens, syst
     payload = {
         "model": model_id,
         "messages": messages,
-        "temperature": 0.6,
+        "temperature": 0.7, # Slightly higher temp
         "max_tokens": max_tokens
     }
     try:
@@ -235,7 +235,8 @@ def _generate_with_claude_provider(client, model_id, prompt, max_tokens, system_
             system=system_message if system_message else None,
             messages=[
                 {"role": "user", "content": prompt}
-            ]
+            ],
+            temperature=0.8 # Slightly higher temp
         )
         content = ""
         if message.content:
@@ -352,6 +353,8 @@ def render_mermaid_diagram(mermaid_code, key):
     )
 
 # ---------- Initialize Session State (Provibe Workflow) ----------
+# (Session state initialization remains the same as the previous version)
+# ... (Omitted for brevity) ...
 # Core workflow steps
 if 'current_step' not in st.session_state:
     st.session_state.current_step = "input_idea" # input_idea -> refine_idea -> review_idea -> generate_docs -> display_docs
@@ -555,8 +558,8 @@ doc_options = {
 
 # ---------- UI Layout (Provibe Workflow) ----------
 st.set_page_config(layout="wide", page_title="Provibe Prompt Tester")
-st.title("🧪 Provibe Prompt Tester (with Q&A Refinement)")
-st.caption("Test and refine prompts for the Provibe document generation workflow, including interactive Q&A.")
+st.title("🧪 Provibe Prompt Tester (with Divergent Q&A Refinement)") # Updated Title
+st.caption("Test and refine prompts for the Provibe document generation workflow, including interactive Q&A exploring possibilities.")
 
 # Display API Key Errors
 if secret_errors:
@@ -570,6 +573,8 @@ if not any_secret_loaded or not SUPPORTED_MODELS:
 # --- Workflow Steps ---
 
 # ---------- Step 1: Input Initial Idea ----------
+# (This step remains the same as the previous version)
+# ... (Omitted for brevity) ...
 if st.session_state.current_step == "input_idea":
     st.header("Step 1: Input Product Idea")
     with st.form(key="idea_form"):
@@ -626,14 +631,21 @@ if st.session_state.current_step == "refine_idea":
 
     # --- Sub-Step 2a: Generate Clarifying Questions ---
     if st.session_state.refinement_sub_step == "generate_questions":
-        st.info(f"Using **{st.session_state.model_choice}** to generate clarifying questions. Please wait.")
-        with st.spinner("AI is preparing questions..."):
+        st.info(f"Using **{st.session_state.model_choice}** to generate questions exploring possibilities. Please wait.")
+        with st.spinner("AI is brainstorming questions..."):
 
-            # --- PROMPT: Define the Question Generation Prompt ---
+            # --- PROMPT: Define the Question Generation Prompt (Divergent Thinking) ---
+            # ***** MODIFIED PROMPT *****
             question_gen_prompt = f"""
-# --- PROMPT: Insert your Question Generation prompt here ---
-# Example: Based on the initial product idea and hints below, generate 3-5 specific clarifying questions for the user. These questions should help elicit more detail about key features, target audience, technical constraints, or core functionality needed to write a better product specification. Output *only* the questions, each on a new line, starting with '- '. Do not include numbering or any other text.
-# --- End Question Generation Prompt ---
+# --- PROMPT: Insert your Divergent Question Generation prompt here ---
+# Example: Analyze the initial product idea and hints below. Instead of just asking for details, generate 3-5 questions that present potential *alternative directions, implementation types, or core feature variations* for this idea. Frame the questions to encourage the user to consider different possibilities (e.g., "Would this be more like X or Y?", "Should the focus be A or B?", "Could this also incorporate Z?").
+# The goal is to help the user explore the solution space before refining.
+# Output *only* the questions, each on a new line, starting with '- '. Do not include numbering or any other text.
+# Example for 'water intake tracker':
+# - Should this primarily be a mobile app for manual logging, or involve a physical smart device (like a bottle)?
+# - Would users benefit more from simple reminders, or personalized coaching based on their habits?
+# - Could gamification (streaks, badges) be a core motivator, or should it focus purely on utility?
+# --- End Divergent Question Generation Prompt ---
 
 **Initial Product Idea:**
 ---
@@ -644,8 +656,9 @@ if st.session_state.current_step == "refine_idea":
 """
             # --- End Question Generation Prompt ---
 
-            system_message_qa = "You are an AI assistant helping to clarify a product idea by asking relevant questions."
-            max_tokens_qa = 300
+            # System message can remain similar or be adjusted slightly
+            system_message_qa = "You are an AI assistant skilled at brainstorming alternative product directions and framing them as clarifying questions."
+            max_tokens_qa = 400 # Allow slightly more tokens for potentially more complex questions
 
             questions_raw = generate_with_selected_model(
                 st.session_state.model_choice,
@@ -655,62 +668,55 @@ if st.session_state.current_step == "refine_idea":
             )
 
             if questions_raw and not questions_raw.startswith("Error:"):
-                # Parse the questions (assuming one question per line, maybe starting with '- ')
+                # Parse the questions
                 st.session_state.clarifying_questions = [
                     q.strip('- ') for q in questions_raw.strip().split('\n') if q.strip() and q.strip() != '-'
                 ]
                 if st.session_state.clarifying_questions:
-                    st.session_state.user_answers = {i: "" for i in range(len(st.session_state.clarifying_questions))} # Initialize empty answers
+                    st.session_state.user_answers = {i: "" for i in range(len(st.session_state.clarifying_questions))}
                     st.session_state.refinement_sub_step = "await_answers"
                 else:
                      st.warning("AI generated questions but they seem empty or incorrectly formatted. Proceeding without Q&A.")
-                     # Fallback: Skip Q&A and go directly to final refinement based only on initial idea
                      st.session_state.refinement_sub_step = "generate_final_refinement"
-                     st.session_state.clarifying_questions = [] # Ensure it's empty
+                     st.session_state.clarifying_questions = []
                      st.session_state.user_answers = {}
-
             else:
                 st.error("Failed to generate clarifying questions. Check API errors or model selection.")
                 st.warning("Proceeding to refine based only on the initial idea (skipping Q&A).")
-                st.session_state.refinement_sub_step = "generate_final_refinement" # Skip Q&A on failure
-                st.session_state.clarifying_questions = [] # Ensure it's empty
+                st.session_state.refinement_sub_step = "generate_final_refinement"
+                st.session_state.clarifying_questions = []
                 st.session_state.user_answers = {}
 
-            st.session_state.processing = False # Done generating questions (or failed)
+            st.session_state.processing = False
             safe_rerun()
 
     # --- Sub-Step 2b: Display Questions and Collect Answers ---
     elif st.session_state.refinement_sub_step == "await_answers":
-        st.info("Please answer the following questions to help refine the product idea:")
+        st.info("Consider these possibilities and answer the questions to guide the refinement:") # Updated info text
         with st.form("answers_form"):
             # Display generated questions and input fields for answers
             for i, question in enumerate(st.session_state.clarifying_questions):
                 st.session_state.user_answers[i] = st.text_area(
-                    f"❓ {question}",
+                    f"🤔 {question}", # Changed icon
                     key=f"answer_{i}",
-                    value=st.session_state.user_answers.get(i, ""), # Preserve answers on rerun
+                    value=st.session_state.user_answers.get(i, ""),
                     height=100
                 )
 
             submit_answers_button = st.form_submit_button(
-                "➡️ Submit Answers & Generate Refined Description",
+                "➡️ Submit Choices & Generate Refined Description", # Updated button text
                 use_container_width=True,
                 disabled=st.session_state.processing
             )
 
         if submit_answers_button:
-            # Basic check if answers are provided (optional)
-            # if not all(st.session_state.user_answers.values()):
-            #     st.warning("Please try to answer all questions for the best result.")
-            # else:
             st.session_state.refinement_sub_step = "generate_final_refinement"
-            st.session_state.processing = True # Start final refinement generation
+            st.session_state.processing = True
             safe_rerun()
 
         # Option to go back
         if st.button("⬅️ Back to Idea Input (Discard Q&A)", disabled=st.session_state.processing):
             st.session_state.current_step = "input_idea"
-            # Clear Q&A state
             st.session_state.clarifying_questions = []
             st.session_state.user_answers = {}
             safe_rerun()
@@ -718,20 +724,21 @@ if st.session_state.current_step == "refine_idea":
 
     # --- Sub-Step 2c: Generate Final Refined Description (using Q&A) ---
     elif st.session_state.refinement_sub_step == "generate_final_refinement":
-        st.info(f"Using **{st.session_state.model_choice}** to generate the final refined description based on the idea and your answers. Please wait.")
+        st.info(f"Using **{st.session_state.model_choice}** to generate the final refined description based on the idea and your chosen direction. Please wait.") # Updated info text
         with st.spinner("AI is synthesizing the refined description..."):
 
             # Prepare Q&A string for the prompt
             qa_summary = "\n".join([
                 f"Q: {st.session_state.clarifying_questions[i]}\nA: {answer}"
-                for i, answer in st.session_state.user_answers.items() if answer # Include only answered questions
-            ]) if st.session_state.user_answers else "No questions were answered."
+                for i, answer in st.session_state.user_answers.items() if answer
+            ]) if st.session_state.clarifying_questions and st.session_state.user_answers else "No questions were asked or answered."
 
 
             # --- PROMPT: Define the Final Refinement Prompt (using Q&A) ---
+            # This prompt might need less adjustment, but ensure it tells the AI to *use* the answers to shape the description
             final_refinement_prompt = f"""
 # --- PROMPT: Insert your Final Refinement prompt (using Q&A) here ---
-# Example: Based on the initial product idea, user preferences, and the following question-answer pairs, generate a concise yet comprehensive 'Refined Product Description'. Synthesize all the information into a well-structured description covering the core value proposition, key features, target audience, and any clarified technical aspects. This description will be the basis for generating all subsequent documents.
+# Example: Based on the initial product idea, user preferences, and the direction indicated by the user's answers to the clarifying questions below, generate a concise yet comprehensive 'Refined Product Description'. Synthesize all the information, *especially incorporating the choices made in the Q&A*, into a well-structured description covering the core value proposition, key features, target audience, and chosen technical/functional approach. This description will be the basis for generating all subsequent documents.
 # --- End Final Refinement Prompt ---
 
 **Initial Product Idea:**
@@ -741,15 +748,15 @@ if st.session_state.current_step == "refine_idea":
 **Optional Preferences/Hints Provided:**
 {st.session_state.tech_stack_hint if st.session_state.tech_stack_hint else "None provided"}
 ---
-**Clarifying Questions & User Answers:**
+**Questions Exploring Possibilities & User's Direction:**
 ---
 {qa_summary}
 ---
 """
             # --- End Final Refinement Prompt ---
 
-            system_message_final_refine = "You are an AI assistant synthesizing information into a final product specification."
-            max_tokens_final_refine = 1500 # Allow slightly more tokens for synthesis
+            system_message_final_refine = "You are an AI assistant synthesizing information and user choices into a focused product specification."
+            max_tokens_final_refine = 1500
 
             final_refined_content = generate_with_selected_model(
                 st.session_state.model_choice,
@@ -760,21 +767,22 @@ if st.session_state.current_step == "refine_idea":
 
             if final_refined_content and not final_refined_content.startswith("Error:"):
                 st.session_state.refined_idea_content = final_refined_content
-                st.session_state.current_step = "review_idea" # Move to the next main step
+                st.session_state.current_step = "review_idea"
             else:
                 st.error("Failed to generate the final refined description.")
-                # Option to retry or go back might be added here
                 st.session_state.current_step = "input_idea" # Go back if failed
 
-            st.session_state.processing = False # End processing
+            st.session_state.processing = False
             safe_rerun()
 
 
 # ---------- Step 3: Review and Confirm Final Idea ----------
+# (This step remains largely the same, but the context expander is more relevant now)
+# ... (Omitted for brevity, assume it's the same as before) ...
 if st.session_state.current_step == "review_idea":
     st.header("Step 3: Review and Confirm Final Refined Idea")
     if st.session_state.refined_idea_content:
-        st.info("Review the AI's final refined description below (generated using your answers). Edit it as needed. This **final text** will be used to generate all documents.")
+        st.info("Review the AI's final refined description below (generated based on your chosen direction). Edit it as needed. This **final text** will be used to generate all documents.")
 
         # Display Q&A for context if available
         if st.session_state.clarifying_questions and st.session_state.user_answers:
@@ -835,6 +843,8 @@ if st.session_state.current_step == "review_idea":
 
 
 # ---------- Step 4: Select and Generate Documents ----------
+# (This step remains the same as the previous version)
+# ... (Omitted for brevity) ...
 if st.session_state.current_step == "generate_docs":
     st.header("Step 4: Generate Product Documents")
     if st.session_state.confirmed_idea_content:
@@ -945,6 +955,8 @@ if st.session_state.current_step == "generate_docs":
 
 
 # ---------- Step 5: Display Generated Documents ----------
+# (This step remains the same as the previous version)
+# ... (Omitted for brevity) ...
 if st.session_state.current_step == "display_docs":
     st.header("Step 5: Generated Documents")
 
@@ -1016,8 +1028,9 @@ if st.session_state.current_step == "display_docs":
 
 
 # ---------- Footer ----------
+# (This section remains the same as the previous version)
+# ... (Omitted for brevity) ...
 st.markdown("---")
 footer_model_choice = st.session_state.get('model_choice', 'N/A')
 st.caption(f"Using model: **{footer_model_choice}** | Workflow Step: **{st.session_state.get('current_step', 'N/A')}**"
            f"{' (Sub-step: ' + st.session_state.get('refinement_sub_step', 'N/A') + ')' if st.session_state.get('current_step') == 'refine_idea' else ''}")
-