Update app.py

app.py

@@ -1,4 +1,7 @@
 import streamlit as st
+import pandas as pd
+import smtplib
+from email.mime.text import MIMEText
 
 # Page config
 st.set_page_config(page_title="AI Maturity Model", layout="wide")
@@ -12,7 +15,20 @@ sections = [
     "Finance, Tax & Legal",
     "Data & Technology"
 ]
-selected_section = st.sidebar.radio("Select Section", sections)
+
+# Session state for navigation
+if "section_index" not in st.session_state:
+    st.session_state.section_index = 0
+
+def next_section():
+    if st.session_state.section_index < len(sections) - 1:
+        st.session_state.section_index += 1
+
+def prev_section():
+    if st.session_state.section_index > 0:
+        st.session_state.section_index -= 1
+
+selected_section = sections[st.session_state.section_index]
 
 # Sample questions per section
 questions_dict = {
@@ -64,14 +80,49 @@ st.title(f"{selected_section}")
 
 questions = questions_dict.get(selected_section, [])
 
+responses = []
+
 for idx, q in enumerate(questions):
-    st.subheader(f"Q{idx + 1}: {q['question']}")
-    st.radio("", q["options"], key=f"q{idx}")
+    response = st.radio(f"Q{idx + 1}: {q['question']}", q["options"], key=f"q{idx}")
+    responses.append({"Section": selected_section, "Question": q["question"], "Answer": response})
     st.markdown("---")
 
+# Convert responses to DataFrame
+df = pd.DataFrame(responses)
+
+# Email function
+def send_email(dataframe):
+    sender_email = "your_email@example.com"
+    sender_password = "your_password"
+    recipient_email = "recipient@example.com"
+    subject = "AI Maturity Model Responses"
+    body = dataframe.to_string(index=False)
+    
+    msg = MIMEText(body)
+    msg['Subject'] = subject
+    msg['From'] = sender_email
+    msg['To'] = recipient_email
+    
+    try:
+        with smtplib.SMTP("smtp.example.com", 587) as server:
+            server.starttls()
+            server.login(sender_email, sender_password)
+            server.sendmail(sender_email, recipient_email, msg.as_string())
+        st.success("Responses submitted successfully!")
+    except Exception as e:
+        st.error(f"Error sending responses: {e}")
+
+# Submit button
+if st.button("Submit Questionnaire"):
+    send_email(df)
+
 # Navigation Buttons
 col1, col2 = st.columns([1, 1])
 with col1:
-    st.button("Previous Section")
+    if st.button("Previous Section"):
+        prev_section()
 with col2:
-    st.button("Next Section")
+    if st.button("Next Section"):
+        next_section()
+
+st.experimental_rerun()