Update src/streamlit_app.py

src/streamlit_app.py

@@ -18,7 +18,6 @@ from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, Tabl
 from reportlab.lib import colors
 import tempfile
 from sendgrid_service import SendGridService
-from whatsapp_service import WhatsAppService
 
 
 # Set page config with dark theme
@@ -381,8 +380,6 @@ if 'pdf_data' not in st.session_state:
     st.session_state.pdf_data = None
 if 'show_email_form' not in st.session_state:
     st.session_state.show_email_form = False
-if 'whatsapp_step' not in st.session_state:
-    st.session_state.whatsapp_step = False
 
 # Helper function to check if explanation has meaningful content
 def has_meaningful_content(text):
@@ -723,8 +720,7 @@ def generate_and_download_report():
     if 'report_step' not in st.session_state:
         st.session_state.report_step = 1
         st.session_state.patient_info = {"name": "", "age": "", "gender": ""}
-        st.session_state.pdf_data = None  # Store PDF data for email/whatsapp
-        st.session_state.whatsapp_step = False  # Flag to track WhatsApp form display
+        st.session_state.pdf_data = None  # Store PDF data for email
 
     # Step 1: Collect patient name
     if st.session_state.report_step == 1:
@@ -768,16 +764,16 @@ def generate_and_download_report():
             
             # Generate PDF
             pdf_bytes = build_medical_report(report_json)
-            st.session_state.pdf_data = pdf_bytes  # Store PDF data
+            st.session_state.pdf_data = pdf_bytes  # Store PDF data for email
             
             # Display success message
             st.success("Your medical report is ready!")
             
-            # Create columns for download, email, and WhatsApp buttons
-            col1, col2, col3 = st.columns(3)
+            # Create two columns for the download and email buttons
+            col1, col2 = st.columns(2)
             
             with col1:
-                # Download button
+                # Offer download
                 st.download_button(
                     label="📥 Download Report",
                     data=pdf_bytes,
@@ -787,17 +783,9 @@ def generate_and_download_report():
                 )
             
             with col2:
-                # Email button
+                # Email button that shows email form when clicked
                 if st.button("📧 Email Report"):
                     st.session_state.show_email_form = True
-                    st.session_state.whatsapp_step = False
-                    st.rerun()
-                    
-            with col3:
-                # WhatsApp button
-                if st.button("📱 WhatsApp Report"):
-                    st.session_state.whatsapp_step = True
-                    st.session_state.show_email_form = False
                     st.rerun()
             
             # Show email form if the button was clicked
@@ -839,62 +827,14 @@ def generate_and_download_report():
                                     st.error(message)
                 
                 # Cancel button to hide email form
-                if st.button("Cancel Email"):
+                if st.button("Cancel"):
                     st.session_state.show_email_form = False
                     st.rerun()
-                    
-            # Show WhatsApp form if that button was clicked
-            if st.session_state.get('whatsapp_step', False):
-                st.subheader("Send Report via WhatsApp")
-                
-                # Check Twilio configuration
-                twilio_sid = os.getenv('TWILIO_ACCOUNT_SID')
-                twilio_number = os.getenv('TWILIO_WHATSAPP_NUMBER')
-                if not all([twilio_sid, twilio_number]):
-                    st.warning("WhatsApp service not configured. Please contact administrator.")
-                
-                # Phone number form with instructions
-                st.markdown("Enter recipient's phone number with country code (e.g., +1234567890)")
-                phone = st.text_input("WhatsApp Number:")
-                
-                # WhatsApp send button
-                if st.button("Send to WhatsApp"):
-                    if not phone:
-                        st.error("Please enter a phone number.")
-                    else:
-                        # Create service and validate/send
-                        whatsapp = WhatsAppService()
-                        is_valid, _ = whatsapp.validate_phone(phone)
-                        
-                        if not is_valid:
-                            st.error("Please enter a valid phone number with country code (e.g., +1234567890)")
-                        else:
-                            # Send via WhatsApp
-                            with st.spinner("Sending to WhatsApp... This may take a few seconds."):
-                                success, message = whatsapp.send_report(
-                                    phone,
-                                    st.session_state.pdf_data,
-                                    st.session_state.patient_info["name"]
-                                )
-                                
-                                if success:
-                                    st.success(message)
-                                    # Hide the form after successful send
-                                    st.session_state.whatsapp_step = False
-                                    st.rerun()
-                                else:
-                                    st.error(message)
-                
-                # Cancel button to hide WhatsApp form
-                if st.button("Cancel WhatsApp"):
-                    st.session_state.whatsapp_step = False
-                    st.rerun()
             
             # Button to start over
             if st.button("Generate Another Report"):
                 st.session_state.report_step = 1
                 st.session_state.show_email_form = False
-                st.session_state.whatsapp_step = False
                 st.rerun()
 
 # Put the toggle in the sidebar - this is the most reliable approach