Update app.py

app.py

@@ -1,6 +1,8 @@
+# app.py
 import streamlit as st
-import json
+import os
 from typing import List, Dict
+import json
 
 from config import MODEL_CHOICES, DEFAULT_MODEL, DEFAULT_PERSONA_PATH
 from utils import (
@@ -28,7 +30,8 @@ if "api_key" not in st.session_state:
 # Sidebar: API key, model, personas upload
 # -------------------------
 st.sidebar.header("🔑 API Configuration")
-api_key_input = st.sidebar.text_input("OpenAI API Key", type="password", value=st.session_state.api_key)
+api_env = os.getenv("OPENAI_API_KEY", "")
+api_key_input = st.sidebar.text_input("OpenAI API Key (or set OPENAI_API_KEY variable)", type="password", value=st.session_state.api_key or api_env)
 if api_key_input:
     st.session_state.api_key = api_key_input
     import openai
@@ -40,7 +43,7 @@ model_choice = st.sidebar.selectbox("Model", MODEL_CHOICES, index=MODEL_CHOICES.
 
 st.sidebar.markdown("---")
 st.sidebar.header("👥 Personas")
-uploaded = st.sidebar.file_uploader("Upload personas.json", type=["json"])
+uploaded = st.sidebar.file_uploader("Upload personas.json (optional)", type=["json"])
 personas = get_personas(uploaded, path=DEFAULT_PERSONA_PATH)
 st.sidebar.metric("Total Personas", len(personas))
 
@@ -53,13 +56,12 @@ tabs = st.tabs(["Text", "Files"])
 with tabs[0]:
     text_desc = st.text_area("Describe your feature", height=160)
 with tabs[1]:
-    uploaded_files = st.file_uploader("Upload wireframes / mockups", accept_multiple_files=True, type=["png", "jpg", "jpeg", "pdf"])
+    uploaded_files = st.file_uploader("Upload wireframes / mockups", accept_multiple_files=True, type=["png","jpg","jpeg","pdf"])
 
 feature_inputs = {
     "Text": text_desc or "",
     "Files": [f.name for f in (uploaded_files or [])]
 }
-
 st.markdown("---")
 
 # -------------------------
@@ -67,7 +69,7 @@ st.markdown("---")
 # -------------------------
 st.header("👥 Choose Personas")
 if not personas:
-    st.warning("No personas available.")
+    st.warning("No personas available. Create personas in the sidebar or upload a personas.json.")
     selected_personas: List[Dict] = []
 else:
     labels = [f"{p['name']} ({p.get('occupation','')})" for p in personas]
@@ -87,7 +89,7 @@ clear_btn = c3.button("🗑️ Clear")
 
 if ask_btn:
     if not st.session_state.api_key:
-        st.warning("Please set your OpenAI API key in the sidebar.")
+        st.warning("Please set your OpenAI API key in the sidebar or via OPENAI_API_KEY.")
     elif not selected_personas:
         st.warning("Please select at least one persona.")
     elif not (question or feature_inputs["Text"]):
@@ -140,10 +142,9 @@ if st.session_state.conversation_history.strip() and selected_personas:
                 matched = True
                 break
         if not matched:
-            # user or neutral lines
             st.markdown(line)
 
-    st.info("💡 Continue the discussion using the **question field above** to ask a follow-up question.")
+    st.info("💡 Continue the discussion using the **question field above** to ask another question.")
 
     # Build & show heatmap
     df_summary = build_sentiment_summary(lines, selected_personas)
@@ -151,7 +152,7 @@ if st.session_state.conversation_history.strip() and selected_personas:
     st.markdown("## 🔥 Persona Sentiment Heatmap")
     st.altair_chart(chart, use_container_width=True)
 else:
-    st.info("No conversation yet. Ask your personas a question to get started!")
+    st.info("💡 No conversation yet. Ask your personas a question to get started!")
 
 # -------------------------
 # Sidebar persona creation (persist)