Update app.py

app.py

@@ -14,6 +14,12 @@ load_dotenv()
 genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
 model = genai.GenerativeModel('gemini-2.0-flash')
 
+# Initialize session state variables if they don't exist
+if 'submitted' not in st.session_state:
+    st.session_state.submitted = False
+if 'offer_result' not in st.session_state:
+    st.session_state.offer_result = ""
+
 # Hide Streamlit menu and footer
 st.markdown("""
 <style>
@@ -47,34 +53,49 @@ with col1:
         temperature = st.slider('🌡️ Creativity Level', min_value=0.0, max_value=2.0, value=0.7,
                             help='Higher values make the output more creative but less focused')
 
-    # Generate button
-    if st.button('Generate Offer 🎉'):
+    # Generate button with callback
+    def generate_offer():
         if not skills or not product_service:
             st.error('Please fill in both Skills and Product/Service fields')
         else:
-            with st.spinner('Creating your perfect offer...'):
-                prompt = f"""Based on the following information, create a compelling offer:
-                Skills: {skills}
-                Product/Service: {product_service}
-                Target Audience: {target_audience if target_audience else 'General audience'}
+            # Set submitted flag to True
+            st.session_state.submitted = True
+            # Store input values in session state
+            st.session_state.skills = skills
+            st.session_state.product_service = product_service
+            st.session_state.target_audience = target_audience
+            st.session_state.temperature = temperature
+            
+    st.button('Generate Offer 🎉', on_click=generate_offer)
+
+# Results column
+with col2:
+    # Check if form has been submitted
+    if st.session_state.submitted:
+        with st.spinner('Creating your perfect offer...'):
+            prompt = f"""Based on the following information, create a compelling offer:
+            Skills: {st.session_state.skills}
+            Product/Service: {st.session_state.product_service}
+            Target Audience: {st.session_state.target_audience if st.session_state.target_audience else 'General audience'}
+            
+            Please create a professional and engaging offer that highlights the value proposition
+            and appeals to the target audience. Include a clear call to action."""
+            
+            try:
+                # Create generation config with temperature
+                generation_config = genai.GenerationConfig(temperature=st.session_state.temperature)
                 
-                Please create a professional and engaging offer that highlights the value proposition
-                and appeals to the target audience. Include a clear call to action."""
+                # Pass the generation config to generate_content
+                response = model.generate_content(prompt, generation_config=generation_config)
+                st.session_state.offer_result = response.text
+                st.success('✨ Your offer is ready!')
                 
-                try:
-                    # Create generation config with temperature
-                    generation_config = genai.GenerationConfig(temperature=temperature)
-                    
-                    # Pass the generation config to generate_content
-                    response = model.generate_content(prompt, generation_config=generation_config)
-                    st.success('✨ Your offer is ready!')
-                    
-                    # Display result in the right column
-                    with col2:
-                        st.markdown('### 📝 Generated Offer')
-                        st.markdown(response.text)
-                except Exception as e:
-                    st.error(f'An error occurred: {str(e)}')
+                # Display result
+                st.markdown('### 📝 Generated Offer')
+                st.markdown(st.session_state.offer_result)
+            except Exception as e:
+                st.error(f'An error occurred: {str(e)}')
+                st.session_state.submitted = False
 
 # Footer
 st.markdown('---')