Update app.py

app.py

@@ -1,24 +1,4 @@
-def display_solution(solution):
-    """Displays the solution with lead capture form."""
-    if solution:
-        if not solution.startswith("Error") and not solution.startswith("API Error"):
-            st.markdown("## Your Customized Solution Plan")
-            st.markdown(solution)
-            
-            with st.expander("Want expert help implementing this solution?"):
-                st.markdown("""
-                Our growth agency specializes in helping B2B companies implement these solutions and achieve measurable results.
-                
-                **Schedule a free 30-minute consultation** to discuss how we can help you overcome this challenge and accelerate your growth.
-                """)
-                contact_email = st.text_input("Your Email")
-                if st.button("Request Consultation"):
-                    if contact_email:
-                        st.success("Thank you! We'll be in touch shortly to schedule your consultation.")
-                    else:
-                        st.warning("Please enter your email address.")
-        else:
-            st.error(solution)import streamlit as st
+import streamlit as st
 st.set_page_config(page_title="B2B Problem Solver", layout="wide")
 
 import requests
@@ -305,7 +285,25 @@ Company Description: {company_description}
 Company Size: {team_size}
 """
                 solution = generate_problem_solution(business_info, problem_category, problem_description)
-                display_solution(solution)
+                if solution:
+                    if not solution.startswith("Error") and not solution.startswith("API Error"):
+                        st.markdown("## Your Customized Solution Plan")
+                        st.markdown(solution)
+                        
+                        with st.expander("Want expert help implementing this solution?"):
+                            st.markdown("""
+                            Our growth agency specializes in helping B2B companies implement these solutions and achieve measurable results.
+                            
+                            **Schedule a free 30-minute consultation** to discuss how we can help you overcome this challenge and accelerate your growth.
+                            """)
+                            contact_email = st.text_input("Your Email")
+                            if st.button("Request Consultation"):
+                                if contact_email:
+                                    st.success("Thank you! We'll be in touch shortly to schedule your consultation.")
+                                else:
+                                    st.warning("Please enter your email address.")
+                    else:
+                        st.error(solution)
 
 else:
     website_url = st.text_input("Enter your company website URL*", placeholder="e.g., www.example.com")