Update app.py

app.py

@@ -261,8 +261,7 @@ st.markdown("### 🖼️ Visualizing Your Design Concept")
 if st.button("Generate Concept Image from Design Summary"):
     with st.spinner("Creating a personalized visual..."):
 
-        refined_prompt = f"""
-Based on the following tiny home project details:
+        refined_prompt = f"""Based on the following tiny home project details:
 
 Region: {design_region}
 Style: {design_style}
@@ -285,8 +284,12 @@ Focus on realistic materials, lighting, and color palette. Use a 3D rendering st
             st.image(concept_image_url, caption="Generated Concept Image", use_column_width=True)
             st.markdown(f"[Download Image]({concept_image_url})", unsafe_allow_html=True)
 
-            # Store this if needed
-            st.session_state_
+            # Store this in session state (optional for later use)
+            st.session_state.generated_concept_image_url = concept_image_url
+
+        except Exception as e:
+            st.error(f"Image generation failed: {e}")
+